Cod sursa(job #209434)

Utilizator anna_bozianuBozianu Ana anna_bozianu Data 22 septembrie 2008 14:23:29
Problema Puteri Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.07 kb
#include<stdio.h>
#define P 128
#define N 100002
int n,i,x[N],y[N],z[N],s[P],j,k,a[P][P][P],op[P],sol;
void readd(),solve(),semne();
int comb(int nr);
int main()
{       readd();
	solve();
	return 0;
}
void readd()
{
	freopen("puteri.in","r",stdin);
	freopen("puteri.out","w",stdout);
	scanf("%d",&n);
	for(i=1;i<=n;i++)scanf("%d%d%d",&x[i],&y[i],&z[i]);
}
void semne()
{
 s[2]=1;s[3]=1;s[5]=1;s[6]=-1;s[7]=1;s[10]=-1;s[11]=1;s[13]=1;s[14]=-1;
 s[15]=-1;s[17]=1;s[19]=1;s[21]=-1;s[22]=-1;s[23]=1;s[26]=-1;s[29]=1;
 s[30]=1;s[31]=-1;s[33]=-1;s[34]=-1;s[35]=-1;s[37]=1;s[38]=-1;s[39]=-1;
 s[41]=1;s[42]=1;s[43]=1;s[46]=-1;s[47]=1;s[51]=-1;s[53]=1;s[55]=-1;
 s[57]=-1;s[58]=-1;s[59]=1;s[61]=1;s[62]=-1;s[65]=-1;s[66]=1;s[67]=1;
 s[69]=-1;s[70]=1;s[71]=1;s[73]=1;s[74]=-1;s[77]=-1;s[78]=1;s[79]=1;
 s[82]=-1;s[83]=1;s[85]=-1;s[86]=-1;s[87]=-1;s[89]=1;s[91]=-1;s[93]=-1;
 s[94]=-1;s[95]=-1;s[97]=1;s[101]=1;s[102]=1;s[103]=1;s[105]=1;
 s[106]=-1;s[107]=1;s[109]=1;s[110]=1;s[111]=-1;s[113]=1;s[114]=1;s[115]=-1;
 s[118]=-1;s[119]=-1;s[122]=-1;s[123]=-1;s[127]=-1;

}
void solve()
{       int val,r1,r2,r3,q1,q2,q3,k1,gata;
	semne();
	for(k=2;k<=127;k++)
	{ if(!s[k])continue;
	  val=0;gata=0;
	  for(i=1;i<k;i++)op[i]=k-i;
	  for(i=1;i<=n;i++)a[x[i]%k][y[i]%k][z[i]%k]++;
	  for(r1=0;r1<k;r1++)
	   { q1=op[r1];
	     for(r2=0;r2<k;r2++)
	     { q2=op[r2];
	       for(r3=0;r3<k;r3++)
	       { q3=op[r3];
		 if(r1<q1){val+=a[r1][r2][r3]*a[q1][q2][q3];continue;}
		 if(r1==q1&&r2<q2){val+=a[r1][r2][r3]*a[q1][q2][q3];continue;}
		 if(r1==q1&&r2==q2&&r3<q3){val+=a[r1][r2][r3]*a[q1][q2][q3];continue;}
		 if(r1==q1&&r2==q2&&r3==q3){val+=comb(a[r1][r2][r3]);continue;}
		 if(2*r1>k){gata=1;break;}
		 if(2*r1==k&&2*r2>k){gata=1;break;}
		 if(2*r1==k&&2*r2==k&&2*r3>k){gata=1;break;}
	       }
	       if(gata)break;
	     }
	     if(gata)break;
	   }
	  sol+=s[k]*val;
	  k1=(k<64)?k:64;
	  for(r1=0;r1<k1;r1++)
	   for(r2=0;r2<k1;r2++)
	    for(r3=0;r3<k1;r3++)
	     a[r1][r2][r3]=0;
	}
	printf("%d\n",sol);
}
int comb(int nr)
{
	return (nr*(nr-1))/2;
}