763A - Timofey and a tree

codeforces百题计划第一周(5)

（1）一开始树形dp写了半天（真的是半天），没过，看了题解，用了很巧妙的方法。

对于对于每条两端颜色不同的边，都需要切断，才可以满足要求。

因为只能删除一个点，那也就是这些有不同颜色端点的边都连在一个点上，那么才能通过删掉这个点把树分成几个纯色子树。

每找到一条这样的边，异色边E++，两个端点num[u]++,num[v]++。最后找最大num[i]，如果等于E，一定可以通过删除i满足要求。

 

（2）感觉树形dp也行的，但是没写好，过了一半测试点，改一点，多过几个，实在不想改了。

对于当前节点v，记录v上面点的颜色数，再统计v节点的每一颗子树的颜色，如果，每颗子树的颜色数是1，v节点上方的颜色数也是1，那么删除v节点可以满足要求。

两次dfs，一次向上，一次回溯。思路应该没有问题，但是写炸了。

#include <bits/stdc++.h>
using namespace std;

const int maxn=2e5+5;


int num[maxn];
int c[maxn];
struct node{
    int u;int v;
}e[maxn];


int main()
{
    int n;scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int u,v;
        scanf("%d%d",&e[i].u,&e[i].v);
    }
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&c[i]);
    }
    int m=0;
    for(int i=1;i<n;i++)
    {
        //cout <<e[i].u<<" "<<e[i].v<<endl;
        if(c[e[i].u]!=c[e[i].v]) m++,num[e[i].u]++,num[e[i].v]++;
    }
    int maa=num[1];int ans=1;
    for(int i=2;i<=n;i++)
    {
        if(num[i]>maa) maa=num[i],ans=i; 
    }
    //cout <<m<<" "<<maa<<endl;
    if(maa==m) printf("YES\n%d\n",ans);
        else printf("NO\n");
    return 0;
}