Cod sursa(job #730521)

Utilizator SebiSebiPirtoaca George Sebastian SebiSebi Data 6 aprilie 2012 13:39:43
Problema Oite Scor 70
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.93 kb
#include<iostream>
#include<fstream>
#include<vector>
using namespace std;
struct oita {
	int val;
	short a,b;
};
vector <oita> v[3033];
int a[1025];
inline int cauta(oita x)
{
	int i,k,n,s;
	k=x.val%3033;
	n=v[k].size()-1;
	s=0;
	for(i=0;i<=n;i++) {
		if((v[k][i].val==x.val)&&(v[k][i].a!=x.a)&&(v[k][i].a!=x.b)&&(v[k][i].b!=x.a)&&(v[k][i].b!=x.b)) 
			s++;
	}
	return s;
}
inline void adauga(oita x)
{
	v[x.val%3033].push_back(x);
}
int main ()
{
	int n,s,i,j;
	long long nr;
	oita x;
	ifstream f("oite.in");
	ofstream g("oite.out");
	f>>n>>s;
	for(i=1;i<=n;i++)
		f>>a[i];
	f.close();
	for(i=1;i<=n-1;i++)
		for(j=i+1;j<=n;j++) 
			if((a[i]+a[j])<=s) {
				x.val=a[i]+a[j];
				x.a=i;
				x.b=j;
				adauga(x);
			}
	nr=0;
	for(i=1;i<=n-1;i++)
		for(j=i+1;j<=n;j++) {
			x.val=s-(a[i]+a[j]);
			if(x.val>=0) {
				x.a=i;
				x.b=j;
				nr=nr+cauta(x);
			}
		}
	g<<nr/6;
	g.close();
	return 0;
}