#2021牛客暑假多校3_J.Counting Triangles_正难则反

J.Counting Triangles

题目传送门：

J

题面：

一句话：

给一个 n(n<=8000) 个顶点的图，每两个顶点边颜色为黑或白，求三个顶点组成的三角形边颜色相同的个数。
正难则反！！！！！！！！
不要直接求个数，用总数减去不符合条件的个数。

代码：

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;

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, ans = 0;
    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)cout << 0 << endl;
    //无法组成三角形

    else {
        ll ans = 1ll * n * (n - 1) * (n - 2) / 6;
        //ans为总数
        ll a, b;
        ll num = 0;
        for (int i = 0; i < n; i++) {
            a = b = 0;
            for (int j = 0; j < n; j++) {
                if (i == j)continue;
                // 图中不存在自环，枚举到相同的点要skip。
                if (edge[i][j])b++;
                else a++;
            }
            num += a * b;
            //注意到一个神奇的性质:每个三⻆形要么同色，要么有两边同色另一边异色。
            // 对于有两边同色另一边异色的三角形，恰有两个异色⻆，而同色没有异色⻆。
            // 对于每一个点，枚举一下所有连出去的黑边和白边的数目
            // 其中任意一条黑边搭配上任意一条白边都一定是不满足要求的三角形上的一对边。
        }
        cout << ans - num / 2 << endl;
        //因此异色⻆数/2 即为不符合条件的三⻆个数。
    }
    return 0;
}