jzoj 3508. 【NOIP2013模拟11.5B组】好元素(hash)

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

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

分析：
10% n^4 随便做。
40% n^3 用布尔数组保存存在的元素，用三层枚举a[i],a[j],a[k]，如果a[i]+a[j]+a[k]存在，那么答案加1。注意枚举顺序和判重。
70% n^2 预处理两个数的和，再枚举，因为每一个元素值比较小，开一个数组来存储是否存在。(其实n^3的方法加上剪枝也可以过)
100% n^2 因为元素的值比较大，在70%的思想上用哈希进行判断即可。

代码

#include <cstdio>
#include <string>
#include <cstring>
#define ll long long
#define N 5005
#define mo 14150547
#define inf 2e9
using namespace std;

ll h[mo],a[N],ans,n;

int hash(ll x)
{
    int tmp=x%mo;
    while (h[tmp]!=0&&h[tmp]!=x) tmp=(tmp+1)%mo;
    return tmp;
}

int main()
{
    freopen("good.in","r",stdin);
    freopen("good.out","w",stdout);
    scanf("%d",&n);
    bool fl=false;
    for (int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
        for (int j=1;j<i;j++)
            if (a[i]-a[j]==0&&fl) 
            {
                ans++;
                break;
            }
            else if (a[i]-a[j]!=0)
                if (h[hash(a[i]-a[j]+inf)]==a[i]-a[j]+inf)
                {
                    ans++;
                    break;
                }
        if (a[i]==0) fl=true;
        for (int j=1;j<=i;j++)
        {
            h[hash(a[i]+a[j]+inf)]=a[i]+a[j]+inf;
        }
    }
    printf("%d",ans);
}