计数三角形

链接：https://ac.nowcoder.com/acm/contest/11254/J
来源：牛客网
 

题目描述

古德亚特发现了一个没有方向的完整图形与n个顶点。图形的每个边缘都漆成黑色或白色。他想让你帮他找到三角形的数量（a，b，c）（一个<b<c），这样（a，b），（b，c），（c，a）之间的边缘有相同的颜色。为了避免输入比例过大，我们使用以下代码生成图表中的边缘。

<span style="color:#333333">namespace GenHelper
{
    unsigned z1,z2,z3,z4,b,u;
    unsigned get()
    {
        b=((z1<<6)^z1)>>13;
        z1=((z1&4294967294U)<<18)^b;
        b=((z2<<2)^z2)>>27;
        z2=((z2&4294967288U)<<2)^b;
        b=((z3<<13)^z3)>>21;
        z3=((z3&4294967280U)<<7)^b;
        b=((z4<<3)^z4)>>12;
        z4=((z4&4294967168U)<<13)^b;
        return (z1^z2^z3^z4);
    }
    bool read() {
      while (!u) u = get();
      bool res = u & 1;
      u >>= 1; return res;
    }
    void srand(int x)
    {
        z1=x;
        z2=(~x)^0x233333333U;
        z3=x^0x1234598766U;
        z4=(~x)+51;
      	u = 0;
    }
}
using namespace GenHelper;
bool edge[8005][8005];
int main() {
  int n, seed;
  cin >> n >> seed;
  srand(seed);
  for (int i = 0; i < n; i++)
    	for (int j = i + 1; j < n; j++)
        	edge[j][i] = edge[i][j] = read();
 	return 0;
}</span>

上述代码中的边缘阵列存储图表中边缘的颜色。边缘[i][j]1意味着从i到j的边缘是黑色的，否则它是白色的（∀0≤i≠j≤n−1[所有 0 ] le i \ 内克 j _ 勒 n - 1∀0≤我​=j≤n−1).
确保有一种方法不取决于数据的生成
方式。
 

输入描述:

The first line contains two integers n(n≤8000),seed(seed≤109)n(n \le 8000), seed (seed \le 10^9)n(n≤8000),seed(seed≤109), denote the number of vertices and the seed of random generator. 

输出描述:

Output a line denoting the answer. 

示例1

输入

10 114514

输出

35

说明

There're 35 triangles that all three edges have the same color. 

题意

就是给你n给点的无向连通图

边有两种颜色

求相同颜色的边组成的三角形的个数

题解

这种三角形只有要么三种颜色相同，三个角颜色一样

要么三角形有一个颜色和另外两个相同颜色相同，不同色的角的存在两个

总三角形个数 -不同颜色角/2；

总三角形个数=n*（n-1）*（n-2）*（2*2）

 for(int i=0;i<n;i++)
    {
    	for(int j=0;j<n;j++)
    		if(edge[i][j]) x[i]++;
    		else y[i]++;
    	ans+=x[i]*(y[i]-1);
	}
    ans=1LL*n*(n-1)*(n-2)/6-ans/2;

 

#include <bits/stdc++.h>
using namespace std;

namespace GenHelper
{
    unsigned z1,z2,z3,z4,b,u;
    unsigned get()
    {
        b=((z1<<6)^z1)>>13;
        z1=((z1&4294967294U)<<18)^b;
        b=((z2<<2)^z2)>>27;
        z2=((z2&4294967288U)<<2)^b;
        b=((z3<<13)^z3)>>21;
        z3=((z3&4294967280U)<<7)^b;
        b=((z4<<3)^z4)>>12;
        z4=((z4&4294967168U)<<13)^b;
        return (z1^z2^z3^z4);
    }
    bool read() {
      while (!u) u = get();
      bool res = u & 1;
      u >>= 1; return res;
    }
    void srand(int x)
    {
        z1=x;
        z2=(~x)^0x233333333U;
        z3=x^0x1234598766U;
        z4=(~x)+51;
      	u = 0;
    }
}
using namespace GenHelper;
bool edge[8005][8005];
long long x[8005],y[8005],ans;
int main() {
  int n, seed;
  cin >> n >> seed;
  srand(seed);
  for (int i = 0; i < n; i++)
    	for (int j = i + 1; j < n; j++)
        	edge[j][i] = edge[i][j] = read();
    for(int i=0;i<n;i++)
    {
    	for(int j=0;j<n;j++)
    		if(edge[i][j]) x[i]++;
    		else y[i]++;
    	ans+=x[i]*(y[i]-1);
	}
	ans=1LL*n*(n-1)*(n-2)/6-ans/2;
	cout<<ans<<endl;
 	return 0;
}