Codeforces 765E - Tree Folding 【树,dfs/bfs】

题目:http://codeforces.com/contest/765/problem/E

题意:

给你一棵树,可以在这棵树上进行若干次操作,每次操作可以把两条长度相同的链,根据一个中点合并在一起。
然后问你经过若干次合并之后,最后的最短链长度是多少。
如果不能合并成一条链,输出-1.

分析:

拓扑排序,每次我们从度数为1的点出发bfs。
然后往上合并,如果遇到交叉点,我们就合并链。
我们用一个set来合并就好了,如果两条链长度一样,在set里面自动就合并了。
显然能够bfs的点的set里面只有一个数。
而且只能有一个点的set里面有两个数。
最后扫一遍check一下就好了。
bfs:http://www.cnblogs.com/qscqesze/p/6401615.html
dfs:http://www.cnblogs.com/RUSH-D-CAT/p/6404742.html

代码:

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5+7;
vector<int> g[maxn];
int n,vis[maxn],d[maxn];
set<int> S[maxn];
int main()
{
    scanf("%d",&n);
    for(int i=1; i<n; i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        g[a].push_back(b);
        g[b].push_back(a);
        d[a]++;
        d[b]++;
    }
    queue<int> Q;
    for(int i=1; i<=n; i++)
    {
        if(g[i].size()==1)
        {
            Q.push(i);
            S[i].insert(0);
        }
    }
    while(!Q.empty())
    {
        int now = Q.front();
        vis[now]=1;
        Q.pop();
        for(int i=0; i<g[now].size(); i++)
        {
            int v = g[now][i];
            if(vis[v])continue;
            S[v].insert((*S[now].begin())+1);
            d[v]--;
            if(d[v]==1&&S[v].size()==1)
            {
                Q.push(v);
            }
        }
    }
    int ans = 0,num2 = 0;
    for(int i=1; i<=n; i++)
    {
        if(S[i].size()==0)
        {
            cout<<"-1"<<endl;
            return 0;
        }
        if(S[i].size()>2)
        {
            cout<<"-1"<<endl;
            return 0;
        }
        if(S[i].size()==1)
        {
            ans=max(ans,*S[i].begin());
        }
        if(S[i].size()==2)
        {
            ans=max(ans,*S[i].begin()+*S[i].rbegin());
            num2++;
        }
    }
    if(num2>1)
    {
        cout<<"-1"<<endl;
        return 0;
    }
    while(ans%2==0)
    {
        ans/=2;
    }
    cout<<ans<<endl;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值