Cod sursa(job #1770018)

Utilizator ionanghelinaIonut Anghelina ionanghelina Data 3 octombrie 2016 17:47:15
Problema Oite Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.36 kb
#include<bits/stdc++.h>
#define cheie 666013
#define maxC 1035
using namespace std;
typedef struct tip
{
  int sum,a,b;
};
vector<tip> h[cheie+5];
vector<tip>::iterator it;
void hadd(tip x)
{
    int y=x.sum%cheie;
    h[y].push_back(x);
}
long long hsearch(int x,int a,int b)
{
    int y=x%cheie,suma,v1,v2;
    long long sol=0;
    for(it=h[y].begin();it!=h[y].end();it++)
    {
        suma=(*it).sum;
        v1=(*it).a;
        v2=(*it).b;
        if(suma==x && v1!=a && v1!=b && v2!=a && v2!=b)
        {
            sol=sol+1LL;
        }
    }
    return sol;
}
int n,l,v[maxC];
long long posibilitati;
int main()
{
    freopen("oite.in","r",stdin);
    freopen("oite.out","w",stdout);
    scanf("%d%d",&n,&l);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&v[i]);
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=(i+1);j<=n;j++)
        {
            int sum=v[i]+v[j];
            tip x;
            x.sum=sum;
            x.a=i;
            x.b=j;
            hadd(x);
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=(i+1);j<=n;j++)
        {
            if((v[i]+v[j])<=l)
            {
                int dif=l-v[i]-v[j];
                posibilitati=posibilitati+hsearch(dif,i,j);
            }
        }
    }
    posibilitati/=6;
    printf("%lld\n",posibilitati);
    return 0;
}