Cod sursa(job #1043541)

Utilizator Kira96Denis Mita Kira96 Data 28 noiembrie 2013 18:36:53
Problema Oite Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.8 kb
#include<fstream>
#define N 2000100
#include<vector>
#define mod 666013
#define nm 1080
using namespace std;
ifstream f("oite.in");
ofstream g("oite.out");
int v[nm],n,sum,i,S,k,mo,j;
long long sol;
vector<pair<int,int> >H[mod];
int main()
{
    f>>n>>sum;
    for(i=1;i<=n;++i)
        f>>v[i];
    for(i=1;i<=n;++i)
	{
		for(j=i+1;j<=n;++j)
		{
			S=sum-v[i]-v[j];
			if(S<0)
				continue;
			mo=S%mod;
			for(k=0;k<H[mo].size();++k)
				if(H[mo][k].first==S)
					sol+=H[mo][k].second;
		}
		for(j=1;j<i;++j)
		{
			S=v[i]+v[j];
			if(S>sum)
				continue;
			mo=S%mod;
			for(k=0;k<H[mo].size();++k)
				if(H[mo][k].first==S)
				{
					H[mo][k].second++;
					break;
				}
			if(k==H[mo].size())
				H[mo].push_back(make_pair(S,1));
		}
	}
    g<<sol;
    return 0;
}