【codeforces 765E】Tree Folding

【题目链接】:http://codeforces.com/problemset/problem/765/E

【题意】

给你一棵树;
可以把一个节点的两条相同长度的链合并成一条链;
且这两条相同长度的链上的点不能有“分叉”;
问你最后是否能形成一条链;
然后让你求链的最短值;

【题解】

dfs;
类似树形DP的东西;
对于每个节点x;
看看它下面的链的长度有多少种->把链的长度存在set里面最后看看set的大小,来确定链的种类;
如果链的种类为0,则这个节点为叶子节点,直接返回0
如果链的种类为1,则这个节点以下的节点都能合并成同一条链,输出这个链的长度就好;
如果链的种类为2,则如果这个节点是根节点的话(没有父亲节点),则这两种链合成一种链;
如果链的种类为2,如果这个节点不是根节点的话,就接近无解了,但我们可以再尝试一下,重新dfs一遍,让这个节点作为根节点,然后再dfs一次;看看这次能不能符合.
如果有3种以上的链,则直接输出无解.

【Number Of WA】

0

【完整代码】

#include <bits/stdc++.h>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define LL long long
#define rep1(i,a,b) for (int i = a;i <= b;i++)
#define rep2(i,a,b) for (int i = a;i >= b;i--)
#define mp make_pair
#define ps push_back
#define fi first
#define se second
#define rei(x) scanf("%d",&x)
#define rel(x) scanf("%lld",&x)
#define ref(x) scanf("%lf",&x)
#define ms(x,y) memset(x,y,sizeof x)

typedef pair<int,int> pii;
typedef pair<LL,LL> pll;

const int dx[9] = {0,1,-1,0,0,-1,-1,1,1};
const int dy[9] = {0,0,0,-1,1,-1,1,-1,1};
const double pi = acos(-1.0);
const int N = 2e5+100;

vector <int> G[N];
int n,root;

int dfs(int x,int fa)
{
    set <int> box;
    for (int y:G[x])
    {
        if (y==fa) continue;
        int d = dfs(y,x);
        if (d==-1) return -1;
        box.insert(d+1);
    }
    if (box.empty()) return 0;
    if (int(box.size())==1) return *box.begin();
    if (int(box.size())==2 && !fa) return *box.begin()+*box.rbegin();
    if (int(box.size())==2 && fa)
    {
        root = x;
        return -1;
    }
    if (int(box.size())>=3) return -1;
}

int main()
{
    //freopen("F:\rush.txt","r",stdin);
    rei(n);
    rep1(i,1,n-1)
    {
        int x,y;
        rei(x),rei(y);
        G[x].ps(y),G[y].ps(x);
    }
    int temp = dfs(1,0);
    if (temp==-1 && root) temp = dfs(root,0);
    while (temp%2==0) temp/=2;
    printf("%d
",temp);
    //printf("
%.2lf sec 
", (double)clock() / CLOCKS_PER_SEC);
    return 0;
}