差（C++）

问题 G: 差

时间限制: 1.000 Sec 内存限制: 128 MB

题目描述

楠楠在网上刷题，感觉第一题：求两数的和(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 <iostream>
using namespace std;
long long a[100010];
int main()
{
    long long n,c,sum=0,ans=0;
    cin>>n>>c;
    for(int i=0;i<n;i++)
    {
        cin>>a[i];
    }
    for(int i=n-1;i>=1;i--)
    {
         if(a[i]-a[0]<c) break;//优化 
         if(a[i]==a[i+1]&&i<=n-2)//优化，重复数据算一次 
         {
         ans+=sum;
         continue;
         }
         sum=0;
         for(int j=i-1;j>=0;j--)
         {
         int x=a[i]-a[j];
         if(x==c) sum++;
         else if(x>c) break;//优化 
         }
         ans=ans+sum;
    }
    cout<<ans;
    return 0;
}