D. The Number of Imposters

显然我们对于每一个关系就相当于连一个无向边
我们显然对于每一个连通块来讲 我们确定其中一个也就确定了这个连通块里的所有
就相当于二分图染色了 直接做即可 到时候我们看白色黑色谁多就加到答案里去
判断无解 也和二分图染色一样

#include <bits/stdc++.h>
using namespace std;
const int N = 2e5+10;
const int M = 998244353;
const int mod = 998244353;
#define int long long
int up(int a,int b){return a<0?a/b:(a+b-1)/b;}
#define endl '\n'
#define all(x) (x).begin(),(x).end()
#define YES cout<<"YES"<<endl;
#define NO cout<<"NO"<<endl;
#define _ 0
#define pi acos(-1)
#define INF 0x3f3f3f3f3f3f3f3f
#define fast ios::sync_with_stdio(false);cin.tie(nullptr);
int sum[N];
void solve() {
    int n;cin>>n;
    vector<pair<int,int>>p(n+1);
    vector<int>b(n+10),a(n+10);
    for(int i=1;i<=n;i++){
        cin>>p[i].first>>p[i].second;
        a[p[i].first]++;
        b[p[i].second]++;
    }
    sort(all(p));
    int ans=0;
    for(int i=1;i<=n;i++){
        if(p[i].first==p[i-1].first||p[i].first==p[i+1].first){
            ans+=(b[p[i].second]-1)*(a[p[i].first]-1);
        }
    }
    cout<<sum[n]-ans<<endl;
}
signed main(){
    fast
    int t;t=1;cin>>t;
    for(int i=3;i<N;i++)sum[i]=i*(i-1)*(i-2)/6;
    while(t--) {
        solve();
    }
    return ~~(0^_^0);
}

D. Training Session

考虑反向 总的-不合法
显然对于每一个同组的 我们之间绝对不会出现同组别的
我们不合法的解 就只有
同 同 同 同
1 2 3 4
找下面的cnt1 cnt2 cnt3 的个数
就是不合法的个数

#include <bits/stdc++.h>
using namespace std;
const int N = 2e5+10;
const int M = 998244353;
const int mod = 998244353;
#define int long long
int up(int a,int b){return a<0?a/b:(a+b-1)/b;}
#define endl '\n'
#define all(x) (x).begin(),(x).end()
#define YES cout<<"YES"<<endl;
#define NO cout<<"NO"<<endl;
#define _ 0
#define pi acos(-1)
#define INF 0x3f3f3f3f3f3f3f3f
#define fast ios::sync_with_stdio(false);cin.tie(nullptr);
int sum[N];
void solve() {
    int n;cin>>n;
    vector<pair<int,int>>p(n+1);
    vector<int>b(n+10),a(n+10);
    for(int i=1;i<=n;i++){
        cin>>p[i].first>>p[i].second;
        a[p[i].first]++;
        b[p[i].second]++;
    }
    sort(all(p));
    int ans=0;
    for(int i=1;i<=n;i++){
        if(p[i].first==p[i-1].first||p[i].first==p[i+1].first){
            ans+=(b[p[i].second]-1)*(a[p[i].first]-1);
        }
    }
    cout<<sum[n]-ans<<endl;
}
signed main(){
    fast
    int t;t=1;cin>>t;
    for(int i=3;i<N;i++)sum[i]=i*(i-1)*(i-2)/6;
    while(t--) {
        solve();
    }
    return ~~(0^_^0);
}