题目背景
woshiren在洛谷刷题,感觉第一题:求两数的和(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 的数对的方案数。
输入输出样例
输入样例#1:
4 1
1 1 2 2
输出样例#1:
4
说明
对于50% 的数据:N 的范围是[1…1,000]。
对于另外50% 的数据:N 的范围是[1…100,000]。
对于100% 的数据:C 的范围是[1…1,000,000,000],N 个整数中每个数的范围是:[0…1,000,000,000]。
思路:二分答案
#include<map> #include<set> #include<cstdio> #include<cstring> #include<iostream> #include<algorithm> using namespace std; set<int>se; map<int,int>ma; int n,c,num,ans; int l,r,mid; int val[100000100]; int main(){ scanf("%d%d",&n,&c); for(int i=1;i<=n;i++){ scanf("%d",&r); ma[r]++; if(se.find(r)==se.end()){ val[++num]=r; se.insert(r); } } for(int i=num;i>=1;i--){ int z=val[i]-c; l=0;r=i;; bool bns=0; for(int j=1;j<=50;j++){ mid=(l+r)/2; if(z>val[mid]) l=mid+1; else if(z<val[mid]) r=mid-1; else if(z==val[mid]){ bns=1; break; } } if(bns) ans+=ma[val[i]]*ma[val[mid]]; } cout<<ans; }