计蒜客NOIP2017提高组模拟赛（三）day2-数三角形

传送门

这题有点坑啊

设A为两边颜色不同的角，B为两边颜色相同的角

那么考虑三种三角形：异色，同色，其他

对于任何一个异色三角形，一定会有三个颜色不同的角，

对于任何一个同色三角形，一定会有零个颜色不同的角，

对于任何一个其他三角形，一个会有两个颜色不同的角，

那么A一定等于异色三角形数目*3+其他三角形数目*2

对于任何一个异色三角形，一定会有零个颜色相同的角，

对于任何一个同色三角形，一定会有三个颜色相同的角，

对于任何一个其他三角形，一个会有一个颜色相同的角，

那么B一定等于同色三角形数目*3+其他三角形数目*1

设异色为x种，同色为y中，其他为z种

则A=3x+2z B=3y+z

题目要求的就是3x-6y

发现就是A-2B

然后就水过了~

PS这数据范围10^5也不对啊。。。难道是考虑高精度么。。。

 #include<cstdio>
 #include<cstdlib>
 #include<algorithm>
 #include<cstring>
 #include<cmath>
 #include<map>
 #include<set>
 #include<queue>
 #include<vector>
 #define INF 0x7f7f7f7f
 #define pii pair<int,int>
 #define ll long long
 using namespace std;
 int n,m;
 namespace solve2
 {
     int G[][];
     int n,m;
     void solve(){
         n=::n,m=::m;
         for(int i=;i<=m;i++){
             int x,y,c;
             scanf("%d%d%d",&x,&y,&c);
             G[x][y]=G[y][x]=c;
         }
         ll ans=;
         for(int i=;i<=n;i++){
             for(int j=i+;j<=n;j++){
                 for(int k=j+;k<=n;k++){
                     if(G[i][j]==G[j][k]&&G[j][k]==G[i][k]){
                         ans-=;
                     }
                     else if(G[i][j]!=G[j][k]&&G[j][k]!=G[i][k]&&G[i][j]!=G[i][k]){
                         ans+=;
                     }
                 }
             }
         }
         printf("%lld\n",ans);
     }
 }
 namespace solve1
 {
     ll A,B;
     int a[][];
     int n,m;
     void solve(){
         n=::n,m=::m;
         for(int i=;i<=m;i++){
             int x,y,c;
             scanf("%d%d%d",&x,&y,&c);c--;
             a[x][c]++,a[y][c]++;
         }
         for(int i=;i<=n;i++){
             ll s1=n--a[i][]-a[i][];
             ll s2=a[i][];
             ll s3=a[i][];
             A+=s1*s2+s1*s3+s2*s3;
             B+=s1*(s1-)/+s2*(s2-)/+s3*(s3-)/;
         }
         printf("%lld",A-*B);
     }
 }
 int read(){
     int x=,f=;char ch=getchar();
     while(ch<''||ch>''){if('-'==ch)f=-;ch=getchar();}
     while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
     return x*f;
 }
 int main()
 {
 //    freopen("data.in","r",stdin);
     n=read(),m=read();
     if(n<=){
         solve2::solve();
     }
     else{
         solve1::solve();
     }
     return ;
 }