Cod sursa(job #1152327)

Utilizator alevasluialeHuhurez Marius alevasluiale Data 24 martie 2014 17:37:07
Problema Oite Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.27 kb
#include <cstdio>
#include <vector>
#include <algorithm>
#define mod 7919
using namespace std;
vector <int> x[mod];
int v[1030];
vector <int> :: iterator it;
int main()
{
    freopen("oite.in","r",stdin);
    freopen("oite.out","w",stdout);
    int n,c,k,i,j,sol=0;
    scanf("%d%d",&n,&c);

    for(i=1;i<=n;i++) scanf("%d",&v[i]);
    sort(v+1,v+n+1);
    for(i=3;i<=n-1;i++)
    {
        for(j=i+1;j<=n;j++)
        {
            k=v[i]+v[j];

                x[k%mod].push_back(k);

        }
    }
    for(i=2;i<=n-2;i++)
    {
        for(j=1;j<i;j++)
        {   k=c-v[i]-v[j];
            if(k>=0)
            {
                if(x[k%mod].size()!=0)
                {
            for(it=x[k%mod].begin();it!=x[k%mod].end();it++)
                    {
                if(*it==k) sol++;
                    }
                }
            }
        }
        for(j=i+2;j<=n;j++)
        {
        k=v[i+1]+v[j];
        if(x[k%mod].size()!=0)
        {
        for(it=x[k%mod].begin();it!=x[k%mod].end();it++)
        {
            if(*it==k)
            {
                *it=x[k%mod].back();
                x[k%mod].pop_back();
                break;
            }
        }
        }
        }
    }
    printf("%d",sol);
}