Colorful Graph CodeForces - 246D

最新推荐文章于 2022-04-25 15:14:18 发布

原创最新推荐文章于 2022-04-25 15:14:18 发布 · 267 阅读

0 ·

CC 4.0 BY-SA版权

cf 同时被 2 个专栏收录

10 篇文章

订阅专栏

STL

2 篇文章

订阅专栏

本文介绍了一种算法问题，即在给定的一棵树及其节点颜色中寻找周围不同颜色种类的最大值。通过使用 set 结构来记录每种颜色出现的情况，并通过一次遍历找出最优解。

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

题意就是给你一棵树，和树上每个节点的颜色，求节点周围不同颜色的种类的最大值，优先输出序号小的节点；
思路就是用set的大小表示每个节点周围和自己的颜色；枚举一遍就行

#include <bits/stdc++.h>
using namespace std;
#define maxn 100100
set<int> st[maxn];
int s[maxn];
int main()
{
    int n,m,minn=100100;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&s[i]);
        st[s[i]].insert(s[i]);
        if(s[i]<minn)
            minn=s[i];
    }
    for(int i=1;i<=m;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        st[s[a]].insert(s[b]);
        st[s[b]].insert(s[a]);
    }
    int ans=0,maxx=0;
    for(int i=minn;i<maxn;i++)
    {
        if(st[i].size()>maxx)
        {
            maxx=st[i].size();
            ans=i;
        }
    }
    printf("%d\n",ans);
}