CodeChef - TAPAIR Counting The Important Pairs

题面

题意

给出一张无向连通图,问去掉两条边后能将其分成多个连通块共有几种方法。

做法

建出dfs树后,发现一共有三种方法使它被分为多个连通块:
1.这两条边中有至少一条边是桥。
2.这两条边分别是仅被一条返祖边覆盖的树边和覆盖这条树边的返祖边。
3.被相同返祖边覆盖的两条树边。
对于第一种方法,只要统计出树边的数量即可,要注意和第三种方法的重合部分。
对于第二种方法,可以通过差分统计出所有树边被几条返祖边覆盖。
对于第三种方法,相对麻烦一点,可以给每条返祖边rand一个编号(尽量大),然后每个树边的编号为所有覆盖它的返祖边的Xor和(也同样可以用差分处理),若两条树边的编号相同则可以看作它们被同一组返祖边覆盖。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#define ll long long
#define N 100100
#define M 300100
using namespace std;

ll n,m,bb=1,first[N],deep[N],a[M],b[M],xo[N],cnt[N],ans;
bool vis[N],tree[M];
struct Bn
{
    ll to,next;
} bn[M<<1];

inline ll ra()
{
    return ((ll)rand()<<48)|((ll)rand()<<32)|((ll)rand()<<16)|rand();
}

inline void add(ll u,ll v)
{
    bb++;
    bn[bb].to=v;
    bn[bb].next=first[u];
    first[u]=bb;
}

void dfs(ll now)
{
    ll p,q;
    for(p=first[now]; p!=-1; p=bn[p].next)
    {
        if(vis[bn[p].to]) continue;
        vis[bn[p].to]=1;
        deep[bn[p].to]=deep[now]+1;
        tree[p>>1]=1;
        dfs(bn[p].to);
    }
}

void Dfs(ll now,ll last)
{
    ll p,q;
    for(p=first[now]; p!=-1; p=bn[p].next)
    {
        if(!tree[p>>1]||bn[p].to==last) continue;
        Dfs(bn[p].to,now);
        cnt[now]+=cnt[bn[p].to];
        xo[now]^=xo[bn[p].to];
    }
}

int main()
{
    srand(517);
    memset(first,-1,sizeof(first));
    ll i,j,p,q;
    cin>>n>>m;
    for(i=1; i<=m; i++)
    {
        scanf("%lld%lld",&p,&q);
        a[i]=p,b[i]=q;
        add(p,q),add(q,p);
    }
    vis[1]=deep[1]=1;
    dfs(1);
    for(i=1; i<=m; i++)
    {
        if(tree[i]) continue;
        if(deep[a[i]]<deep[b[i]]) swap(a[i],b[i]);
        p=ra();
        cnt[a[i]]++,cnt[b[i]]--;
        xo[a[i]]^=p,xo[b[i]]^=p;
    }
    Dfs(1,-1);
    sort(xo+2,xo+n+1);
    for(i=2; i<=n; i++)
    {
        for(j=i+1; j<=n&&xo[j]==xo[i]; j++);
        ans+=(j-i)*(j-i-1)/2;
        i=j-1;
    }
    p=0;
    for(i=2; i<=n; i++)
    {
        if(!cnt[i]) ans+=m-1,p++;
        else if(cnt[i]==1) ans++;
    }
    cout<<ans-p*(p-1);
}

猜你喜欢

转载自blog.csdn.net/yzyyylx/article/details/81002169