题目描述
楠楠在网上刷题,感觉第一题:求两数的和(A+B Problem)太无聊了,于是增加了一题:A-B Problem,难倒了一群小朋友,哈哈。
题目是这样的:给出N个从小到大排好序的整数,一个差值C,要求在这N个整数中找两个数A和B,使得A-B=C,问这样的方案有多少种?
例如:N=5,C=2,5个整数是:2 2 4 8 10。答案是3。具体方案:第3个数减第1个数;第3个数减第2个数;第5个数减第4个数。
输入
第一行2个正整数:N,C。
第二行N个整数:已经有序。注意:可能有相同的。
输出
一个整数,表示该串数中包含的所有满足A-B=C的数对的方案数。
样例输入
4 1
1 1 2 2
样例输出
4
提示
5个数据:N的范围是[1…1,000]。
5个数据:N的范围是[1…100,000]。
所有数据:
C的范围是[1…1,000,000,000]。
N个整数中每个数的范围是:[0…1,000,000,000]。
#include <bits/stdc++.h> //注:这次我是贴的一个大佬的代码……未解emmmmmmm
using namespace std;
map<long long,long long> bk;
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
long long n,c,sum=0;
cin>>n>>c;
for(int i=0;i<n;i++)
{
int t;
cin>>t;
if(t-c>=0)
sum+=bk[t-c];
bk[t]++;
}
cout<<sum;
}
大佬的思路
又是一个朴素O(N*N)超时的题,还得寻找别的方法
实际上这里去记录每个差值的数,然后加上这个差值所包含的数,因为数据给出就是升序的,所以不用排序,然后把每次数的数量记录一下
注意如果差值是一个负数那么可以不用记录,因为那样没有什么意义
实际上这个东西就是把A-B=C移项变成了A-C=B然后记录B有多少
例如样例
4 1
1 1 2 2
1-1=0
sum+=bk[0]=0
bk[1]=1
表明1已经出现了1次了
1-1=0
sum+=bk[0]=0
bk[1]=2
表明1已经出现了2次了
2-1=1
sum+=bk[1]=2
bk[2]=1
2-1=1
sum+=bk[1]=4
bk[2]=2
所以答案是4