2021牛客暑期多校训练营3 J题

J题: Counting Triangles

原题链接:https://ac.nowcoder.com/acm/contest/11254/J

题目大意

有一个 $n(n\le 8000)$ 个点的完全无向图，每条边被染为黑色或白色。求三角形 $(a,b,c)(a<b<c)$ 的个数，使得$(a,b),(b,c),(a,c)$ 三边均为同一种颜色。
为避免输入过大，采用以下方式随机生成边:

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;
}

题解

数据范围 $n$ 最大为8000，若暴力枚举三个点复杂度必然TLE。
显然，直接寻找合法的三角形不行，那么我们考虑采用排除非法的三角形。
显然，对于每个三角形的染色情况只会有 $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];
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();
    if(n<3){//不清楚是否保证存在三角形，先写个特判xd
        cout<<0;
        return 0;
    }
    long long sum=0,s1,s2;//记录每个点的异色边总情况数
    for(int i=0;i<n;i++){
        s1=0,s2=0;
        for(int j=0;j<n;j++){
            if(i==j)continue;
            if(edge[i][j])s1++;//统计该点连接的黑色边数
            else s2++;//统计该点连接的白色边数
        }
        sum+=s1*s2;//根据乘法原理得到两边一黑一白的情况数
    }
    cout<<1ll*n*(n-1)*(n-2)/6-sum/2;//通过C(n,3)算出总三角形数，再减去非法三角形数，注意给出的代码中n为int类型，若直接计算在n较大的情况下会溢出，注意转long long
 	return 0;
}