Cod sursa(job #1152281)

Utilizator alevasluialeHuhurez Marius alevasluiale Data 24 martie 2014 17:16:51
Problema Oite Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.38 kb
#include <cstdio>
#include <vector>
#include <algorithm>
#define mod 3033
using namespace std;
vector <int> x[3033];
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);
    int v[n];
    for(i=0;i<n;i++) scanf("%d",&v[i]);
    sort(v,v+n);
    for(i=2;i<n-1;i++)
    {
        for(j=i+1;j<n;j++)
        {
            k=v[i]+v[j];

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

        }
    }
    for(i=1;i<n-2;i++)
    {
        for(j=0;j<i;j++)
        {
            if(x[(c-v[i]-v[j])%mod].size()!=0)
            {
                for(it=x[(c-v[i]-v[j])%mod].begin();it!=x[(c-v[i]-v[j])%mod].end();it++)
                {

                if(*it==c-v[i]-v[j]) sol++;


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