差(二分查找)
题目描述
楠楠在网上刷题,感觉第一题:求两数的和(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]。
这题一开始看到1e9真没想到二分能过(因为我想它再怎么分那1e9的for循环还是要走的,可能是数据水了),最基础的二分跑一遍就行了。
#include<iostream>
using namespace std;
int sum=0;
int num[10000005];
int main(int argc, char const *argv[])
{
int n,m;
while(~scanf("%d%d",&n,&m))
{
sum=0;
for(int i=0;i<n;i++)
{
scanf("%d",&num[i]);
}
for(int k=0;k<n;k++)
{
int c=num[k]+m;
int i=0,j=n-1;
while(i<=j)
{
int mid=(i+j)/2;
if(num[mid]==c)
{
sum++;
int p=mid-1;
while(num[p]==c&&p>=0)
{
p--;
sum++;
}
p=mid+1;
while(num[p]==c&&p<n)
{
p++;
sum++;
}
break;
}
else if(num[mid]>c)
j=mid-1;
else
i=mid+1;
}
}
printf("%d\n",sum);
}
return 0;
}
//最后附上一个最最基础的二分模板。
int dls(int num[],int n,int want_to_find)
{
int i=0,j=n-1;
while(i<=j)
{
int mid=(i+j)/2;
if(num[mid]==want_to_find)
{
return mid; //返回下标
}
else if(num[mid]>want_to_find)
{
j=mid-1;
}
else
{
i=mid+1;
}
}
return -1; //如果找不到就返回-1
}