JZOJ 3508. 【NOIP2013模拟11.5B组】好元素

3508. 【NOIP2013模拟11.5B组】好元素(good) 
(File IO): input:good.in output:good.out

Time Limits: 2000 ms  Memory Limits: 262144 KB  Detailed Limits  

Goto ProblemSet

Description

小A一直认为，如果在一个由N个整数组成的数列An中，存在Am + An + Ap = Ai（1 <= m, n, p < i）（m, n, p可以相同）的话，Ai就是一个“好元素”。现在，小A有一个数列，他想知道这个数列中有多少个“好元素”，请你帮帮他。

 

Input

第一行只有一个正整数N，意义如上。

第二行包含N个整数，表示数列An。

Output

输出一个整数，表示这个数列中“好元素”的个数。

 

Sample Input

输入1：
2
1 3
输入2：
6
1 2 3 5 7 10
输入3：
3
-1 2 0

Sample Output

输出1：
1
输出2：
4
输出3：
1

 

Data Constraint

对于10%的数据  1<=N<=10

对于40%的数据  1<=N<=500    -10^5<=Ai<=10^5

对于70%的数据  1<=N<=5000   -10^6<=Ai<=10^6

对于100%的数据 1<=N<=5000   -10^9<=Ai<=10^9

 

做法：预处理ai + aj可以达到的值，用hash存起来，然后枚举l, k就好了

 

代码如下：

 #include <cstdio>
 #include <cstring>
 #include <string>
 #include <iostream>
 #include <algorithm>
 #define N 5007
 #define mo 14150547
 #define z 2000000000
 #define LL long long
 using namespace std;
 LL n, a[N], h[mo + ];
 int ans;
 bool f;

 LL hash(LL x)
 {
     LL p = x % mo;
     while (h[p] !=  && h[p] != x - z && p <= mo)    p = (p + ) % mo;
     return p;
 }

 void work()
 {
     for (int i = ; i <= n; i++)
     {
         for (int j = ; j < i; j++)
             if (h[hash(a[i] - a[j] + z)] == a[i] - a[j] && a[i] - a[j] != )
             {
                 ans++;
                 break;
             }
             else if (a[i] - a[j] == )
             {
                 if (f)
                 {
                     ans++;
                     break;
                 }
             }
         for (int j = ; j <= i; j++)
         {
             LL p = (a[i] + a[j] + z) % mo;
             while (h[p] != a[i] + a[j] && h[p] !=  && p <= mo)    p = (p + ) % mo;
             if (a[i] + a[j] != )    h[p] = a[i] + a[j];
             else f = true;
         }
     }
 }

 int main()
 {
     freopen("good.in", "r", stdin);
     freopen("good.out", "w", stdout);
     scanf("%lld", &n);
     for (int i = ; i <= n; i++)
         scanf("%lld", &a[i]);
     work();
     cout << ans;
 }