Cod sursa(job #665358)

Utilizator nautilusCohal Alexandru nautilus Data 21 ianuarie 2012 23:15:33
Problema Oite Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.95 kb
#include<fstream>
#include<vector>
using namespace std;
#define NMAX 1030
#define PRIM 666013

typedef struct pereche
{
 int elemmax,suma;
} pereche;

int n,l;
int a[NMAX];
vector<pereche> t[PRIM + 5];
long long moduri;

void read()
{
 int i;
 ifstream fin("oite.in");
 fin>>n>>l;
 for (i=1; i<=n; ++i)
	 fin>>a[i];
 fin.close();
}

void solve()
{
 int i,j,k,suma1,suma2;
 pereche aux;
 for (i=1; i<n; ++i)
	 for (j=i+1; j<=n; ++j)
		{
		 suma1 = a[i] + a[j];
		 suma2 = l - suma1;
		 if ( ! t[suma2 % PRIM].empty() )
			for (k=0; k<(int)t[suma2 % PRIM].size(); ++k)
				if (t[suma2 % PRIM][k].suma == suma2)
					if (t[suma2 % PRIM][k].elemmax < i && t[suma2 % PRIM][k].elemmax < j )
						moduri++;
		 aux.elemmax = j; aux.suma = suma1;
		 t[suma1 % PRIM].push_back(aux);
		}
}

void write()
{
 ofstream fout("oite.out");
 fout<<moduri<<'\n';
 fout.close();
}

int main()
{
 read();
 solve();
 write();
 return 0;
}