[agc006f]Blackout

题目大意

给你一个人有向图。
如果x->y,y->z,加上z->x。
重复该过程直到不能再添加。
求最终图中有多少边。

做法

每一个弱联通块独立。
尝试对一个弱联通块三染色，然后分为三种情况。
1、染色失败。
一定是出现了二元环或自环，此时所有边均会出现。
2、染色不完全（没有用完三种颜色）。
此时不会出现新的边。
3、染色成功。
所有染色为0的向1连，1向2，2向0。

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=100000+10;
int h[maxn],go[maxn*2],dis[maxn*2],nxt[maxn*2],fa[maxn],co[maxn];
bool bz[maxn],pd[maxn];
int i,j,k,l,t,n,m,tot,top,cnt,zero,one,two,mx;
bool czy;
ll ans;
void add(int x,int y,int z){
    go[++tot]=y;
    dis[tot]=z;
    nxt[tot]=h[x];
    h[x]=tot;
}
void dfs(int x){
    bz[x]=1;
    top++;
    if (!co[x]) zero++;
    else if (co[x]==1) one++;
    else two++;
    mx=max(mx,co[x]);
    int t=h[x];
    while (t){
        if (!pd[(t+1)/2]){
            pd[(t+1)/2]=1;
            cnt++;
            if (!bz[go[t]]){
                co[go[t]]=(co[x]+dis[t]+3)%3;
                dfs(go[t]);
            }
            else{
                if (co[go[t]]!=(co[x]+dis[t]+3)%3) czy=1;
            }
        }
        t=nxt[t];
    }
}
int main(){
    scanf("%d%d",&n,&m);
    fo(i,1,m){
        scanf("%d%d",&j,&k);
        add(j,k,1);add(k,j,-1);
    }
    fo(i,1,n)
        if (!bz[i]){
            czy=top=zero=one=two=cnt=mx=0;
            dfs(i);
            if (czy){
                ans+=(ll)top*top;
                continue;
            }
            if (!zero||!one||!two){
                ans+=(ll)cnt;
                continue;
            }
            ans+=(ll)zero*one;
            ans+=(ll)one*two;
            ans+=(ll)two*zero;
        }
    printf("%lld\n",ans);
}