Cod sursa(job #190738)

Utilizator hadesgamesTache Alexandru hadesgames Data 23 mai 2008 23:52:22
Problema Triplete Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.95 kb
#include <stdio.h>
#include <vector>
#include <set>
using namespace std;
#define fori(it,v) for (typeof ((v).begin()) it=(v).begin();it!=(v).end();++it)
#define pb push_back
#define all(c) c.begin(),c.end()
set<int> b;
vector<set<int> >c(2100);
vector<vector<int> > a(4100);
int main()
{
	FILE *in,*out;
	int i,m,n,x,y,nr=0;
	in=fopen("triplete.in","r");
	out=fopen("triplete.out","w");
	fscanf(in,"%d%d",&n,&m);
	for (i=1;i<=m;++i)
	{
		fscanf(in,"%d%d",&x,&y);
		if (x>=2099)
			c[x-2099].insert(y);
		if (y>=2099)
			c[y-2099].insert(x);
		if (x>=2099)
		{
			fori(j,a[y])
				if (c[x-2099].find(*j)!=c[x-2099].end())
					nr++;
		}
		else
			if (y>=2099)
			{
				fori(j,a[x])
					if (c[y-2099].find(*j)!=c[y-2099].end())
						nr++;
			}
			else
			{
				fori (j,a[x])
					b.insert(*j);
				fori (j,a[y])
					if (b.find(*j)!=b.end())
					nr++;
				b.erase(all(b));
			}
		a[x].pb(y);
		a[y].pb(x);
	}
	fprintf(out,"%d\n",nr);
	fclose(in);
	fclose(out);
	return 0;
}