hihocoder 1505 小Hi和小Ho的礼物

#include<cstdio>
#include<cstring>
#include<map>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<map>
#include<queue>
#include<vector>
using namespace std;
#define ll long long


#define MAXN  2000010

int sum[MAXN];
int cnt[MAXN];
int z[1010];

int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&z[i]);
        cnt[z[i]]++;
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=i+1;j<=n;j++)
            sum[z[i]+z[j]]++;
    }
    ll ans=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=i+1;j<=n;j++)
        {
            ans=ans+sum[z[i]+z[j]];
            if(z[i]==z[j])
            {
                ans=ans-(cnt[z[i]]-1)-(cnt[z[j]]-1)+1;
            }
            else
            {
                ans=ans-cnt[z[i]]-cnt[z[j]]+1;
            }
        }
    }
    cout<<ans<<endl;
    return 0;
}