洛谷P1102 A-B 数对(A+B Problem)
作者:互联网
题目描述
出题是一件痛苦的事情!
相同的题目看多了也会有审美疲劳,于是我舍弃了大家所熟悉的 A+B Problem,改用 A-B 了哈哈!
好吧,题目是这样的:给出一串数以及一个数字 CC,要求计算出所有 A - B = CA−B=C 的数对的个数(不同位置的数字一样的数对算不同的数对)。
输入格式
输入共两行。
第一行,两个整数 N, CN,C。
第二行,NN 个整数,作为要求处理的那串数。
输出格式
一行,表示该串数中包含的满足 A - B = CA−B=C 的数对的个数。
输入输出样例
输入
4 1 1 1 2 3
输出
3
说明/提示
对于 75% 的数据,。
对于 100% 的数据,。
保证所有输入数据绝对值小于 。
思路
以该题目为例,找有多少个数对能满足A-B=C(其中A、B未知,C已知),对于一个式子来说,要找两个都未知的数字组合起来等于一个常数,这听起来就比较困难,所以我们对式子做个变形A-C=B,也就是说在数组中去找那些数字A减去一个常量等于另外一个数。(想法尝试)
关于此题的纯暴力方法,就是O(n2)两个for循环遍历,这肯定会超时,我们就想能不能优化只用一个for循环,遍历一次得到结果。
那我们就尝试遍历A,在遍历A的同时去查找B,这里就有一个使用map的方法能打到此效果
map用来装数组中每个数出现的次数,即map<int int> = map<数组元素,出现次数>
最后我们就遍历整个数组,累加每个数据(A)+C的结果的map映射值(B),如果存在B就有具体的值,没有B就是0,所以最后的和就是我们想要的答案
AC代码
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
map<ll,ll> ma;
int main()
{
int n;
ll c;
cin>>n>>c;
ll a[n+5];
for(int i=0; i<n; i++)
{
cin>>a[i];
ma[a[i]]++;
}
ll cnt=0;
for(int i=0; i<n; i++)
{
cnt += ma[a[i]+c];
}
cout<<cnt;
return 0;
}
标签:map,遍历,洛谷,int,ll,数对,题目,Problem,输入 来源: https://blog.csdn.net/m0_56722204/article/details/121056255