基环树 Codeforces Round #686 (Div. 3) E题

最新推荐文章于 2023-12-06 14:43:55 发布

倔强的blackatao

最新推荐文章于 2023-12-06 14:43:55 发布

阅读量134

点赞数

分类专栏： codeforces 总结

本文链接：https://blog.csdn.net/weixin_44745441/article/details/115561213

版权

codeforces 总结专栏收录该内容

29 篇文章 0 订阅

订阅专栏

基环树 Codeforces Round #686 (Div. 3) E题

题意：
给定一棵n个结点的基环树，求所有长度大于等于1的路径的个数

解析：
在这里插入图片描述

如果没有环的话，本来可以这么求，
因为有环，如果经过环的话，环里可以沿顺时针或逆时针，所以还要多加上 cnt[i] * (n-cnt[i])
cnt[i]代表环上挂着的子树的结点数，n-cnt[i]代表另外的那些结点，cnt[i]这些结点到另外的结点都有两路可以走。

在这里插入图片描述

如上图a到b 这样。

int main(){
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        vector<set<ll>> g(n+1);
        for(int i=0;i<n;i++){
            ll x,y;
            scanf("%lld%lld",&x,&y);
            g[x].insert(y);
            g[y].insert(x);
        }
        vector<ll> a(n+1,1);
        queue<ll> leaf;
        for(int i=1;i<=n;i++){
            if(g[i].size() == 1){
                leaf.push(i);
            }
        }
        while(leaf.size()){
            ll v = leaf.front();
            leaf.pop();
            ll u = *g[v].begin();
            a[u] += a[v];
            a[v] = 0;
            g[v].clear();
            g[u].erase(v);
            if(g[u].size() == 1) leaf.push(u);
        }
        ll res = 0;
        for(int i=1;i<=n;i++){
            res += a[i] * (a[i] - 1) / 2 + a[i] * (n - a[i]);
        }
        printf("%lld\n", res);
    }
    return 0;
}