Cod sursa(job #543404)

Utilizator Magnuscont cu nume gresit sau fals Magnus Data 27 februarie 2011 23:14:37
Problema Oite Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.16 kb
#include <stdio.h>
#include <vector>

using namespace std;

struct sp
{
    long long s1,s2,s;
};

vector <sp> h[997];
sp aux;
long long n,v[1025],sol;

inline bool comp(long long i,long long j,long long k,long long l)
{
    if (h[i][j].s1==h[k][l].s1) return 0;
    if (h[i][j].s1==h[k][l].s2) return 0;
    if (h[i][j].s2==h[k][l].s1) return 0;
    if (h[i][j].s2==h[k][l].s2) return 0;
    return 1;
}

int main()
{
    long long i,j,k,l,a;
    freopen("oite.in","r",stdin);
    freopen("oite.out","w",stdout);
    scanf("%lld%lld",&n,&l);
    for (i=1;i<=n;++i)
        scanf("%lld",&v[i]);
    for (i=1;i<=n;++i)
        for (j=i+1;j<=n;++j)
            if (i!=j)
            {
                aux.s1=i;
                aux.s2=j;
                aux.s=v[i]+v[j];
                h[aux.s%997].push_back(aux);
            }
    for (i=0;i<=997;++i)
        for (j=0;j<h[i].size();++j)
        {
            a=(l-i)%997;
            if (a<0) a+=997;
            for (k=0;k<h[a].size();++k)
                if ((h[i][j].s+h[a][k].s==l)&&(comp(i,j,a,k)))
                    ++sol;
        }
    printf("%lld",sol/6);
    return 0;
}