【BZOJ 2916】 2916: [Poi1997]Monochromatic Triangles (容斥)
2916: [Poi1997]Monochromatic Triangles
Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 310 Solved: 150Description
空间中有n个点,任意3个点不共线。每两个点用红线或者蓝线连接,如果一个三角形的三边颜色相同,那么称为同色三角形。给你一组数据,计算同色三角形的总数。Input
第一行是整数n, 3 <= n <= 1000,点的个数。
第二行是整数m, 0 <= m <= 250000,红线数目。接下来的m行,每行两个数p和k,1 <= p < k <= n。表示一条红线的两个端点。Output
一个整数,单色三角形的数目。
Sample Input
6
9
1 2
2 3
2 5
1 4
1 6
3 4
4 5
5 6
3 6
Sample Output
2HINT
数据已加强
Source
【分析】
三角形只有4种情况。
同色:红红红、蓝蓝蓝
非同色:红红蓝、红蓝蓝。
发现如果找到红-蓝,那么肯定是非同色,不用管第三条边,然后会重复两次算到它。
于是答案=全集-非同色。直接枚举连接点就好了。
1 #include<cstdio> 2 #include<cstdlib> 3 #include<cstring> 4 #include<iostream> 5 #include<algorithm> 6 using namespace std; 7 #define LL long long 8 #define Maxn 1010 9 10 int d[1010]; 11 12 int main() 13 { 14 int n,m,ans=0; 15 scanf("%d%d",&n,&m); 16 memset(d,0,sizeof(d)); 17 for(int i=1;i<=m;i++) 18 { 19 int x,y; 20 scanf("%d%d",&x,&y); 21 d[x]++;d[y]++; 22 } 23 for(int i=1;i<=n;i++) 24 { 25 int x=n-1-d[i]; 26 ans-=1LL*x*d[i]; 27 } 28 ans/=2; 29 ans+=1LL*n*(n-1)*(n-2)/6; 30 printf("%d\n",ans); 31 return 0; 32 }
2017-04-20 15:57:10