C++ / ACM 自学笔记 · v5.0

P1102 A-B 数对

原题入口

对应章节:三、解题方法论:从题意到代码 → (十二) 一题多建模与最优方法的选择过程;三、解题方法论:从题意到代码 → (十七) 解题复盘与模型迁移;五、数据结构与算法 → (一) 时间复杂度与空间复杂度;五、数据结构与算法 → (九) 双指针与滑动窗口(尺取法);五、数据结构与算法 → (十) 二分与三分搜索。

(一) 任务与输出目标

统计差为 C 的位置对,相同值的不同位置分别计数。

(二) 建模与推导

固定 A,寻找 B=A-C。预先保存每个值的出现次数,一个 A 的位置就贡献 count[A-C] 对。3 出现两次、1 出现三次且 C=2 时,贡献为 2×3=6。A、B 由数值关系确定,不要求 A 的输入位置在 B 前。

(三) 手算与过程

1,1,3,3,3,C=2:三个 3 各与两个 1 配对,共 6 对。

按具体数据展开的处理与返回过程

(四) 带注释实现

完整源文件

#include <bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;

void solve() {
    int n;
    int c;
    cin >> n >> c;
    vector<int> a(n);
    map<int, int> count;
    for (auto& x : a) {
        cin >> x;
        ++count[x];
    }
    int answer = 0;
    for (int x : a) {
        auto it = count.find(x - c); // 查出每个 A 能配上的 B 的位置数。
        if (it != count.end())
            answer += it->second;
    }
    cout << answer << '\n';
}

signed main() {
    ios::sync_with_stdio(false); // 关闭同步,使用 cin 与 cout 完成输入输出。
    cin.tie(0), cout.tie(0);

    int T = 1; // 本题读入一组数据。
    // cin >> T; // 题目给出测试组数时开启,并在 solve 中处理一组。
    while (T--)
        solve();
}

(五) 复杂度

时间 \(O(n\log n)\),空间 \(O(n)\)。

返回本章题解索引 · 返回全部题号索引

查阅笔记正文与全部题解 · Ctrl / ⌘ K