Cod sursa(job #1152291)

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

    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++)
        {
            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();

                        break;
                    }
                }
            }
        }
    }
    printf("%d",sol);
}