Codeforces 764C Timofey and a tree（思路）

最新推荐文章于 2019-11-23 21:22:00 发布

wang2147483647

最新推荐文章于 2019-11-23 21:22:00 发布

阅读量436

点赞数

CC 4.0 BY-SA版权

分类专栏：思路 OJ_Codeforces ACM

本文链接：https://blog.youkuaiyun.com/wang2147483647/article/details/54846993

ACM 同时被 3 个专栏收录

232 篇文章

订阅专栏

OJ_Codeforces

44 篇文章

订阅专栏

思路

37 篇文章

订阅专栏

本文提供了一种解决CodeForces平台C题的有效算法。通过分析不同颜色节点间的连接特性，确定了根节点的位置，实现了对特定图结构的快速识别。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目地址：http://codeforces.com/contest/764/problem/C

思路：对于所有两点颜色不同的边，一定为其中一点与根节点相连（若不相连则子树颜色不同），所有两点颜色不同的边组成的图，其边总数为tot，则根节点即为度为tot的点（与所有颜色不同的点相连）。

#include<set>
#include<cstdio>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=1e5+50;
struct Node
{
    int u,v;
};
int n;
int d[maxn];
int col[maxn];
Node e[maxn];
int main()
{
    scanf("%d",&n);
    for(int i=1; i<n; i++)
        scanf("%d%d",&e[i].u,&e[i].v);
    for(int i=1; i<=n; i++)
        scanf("%d",&col[i]);
    int tot=0;
    for(int i=1; i<n; i++)
    {
        if(col[e[i].u]!=col[e[i].v])
        {
           tot++;
           d[e[i].u]++,d[e[i].v]++;
        }
    }
    for(int i=1;i<=n;i++)
    {
        if(d[i]==tot)
        {
            printf("YES\n%d\n",i);
            return 0;
        }
    }
    printf("NO\n");
   return 0;
}