这题有点坑啊
设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 ;
}