小美的树上染色

美团2024届秋招笔试第一场编程真题

先提一个小知识:题目中凡是提到树结构都要使用图的存储方式,只有二叉树例外。

分析:在树结构中,孩子和父节点是相邻节点,而父节点可能有多个孩子节点。在染色的过程中,本质是父子节点满足条件就染成红色。具体染色策略以下图为例

图中节点权值均为1,也就是任意相邻节点都可以满足染色条件,显然我们不能先将a和b染色,那样就只有2个点能被染色,而是应该先染色<f,b>或 <e,b>,然后再<a,c> 或<a,d>,这样可以染4个点。下图又该先考虑染色那些节点呢?

这样总结出贪心策略:先处理叶子结点的染色,然后处理内部节点。比如上图,先处理<g,f>。如果能染色,哪么将g,f标记红色,同时<f,b>就不可能染色了如果不能染色,未来还可以继续试探<f,b>。

实现方法:先处理最外层节点(叶子),外层处理完了,这些叶子就可以不要了,往内层推进。

使用拓扑排序的思想。当然不是入度0节点入队,这可不是有向图。而是度为1(叶子)节点入队,从树叶向树根进行拓扑处理。

图问题的复杂度正常都是遍历所有的结点,也就是访问所有的点和边,此图n-1条边,复杂度为O(n)。

cpp 复制代码
#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
ll n,a[100005],d[100005],red[100005],v[100005],ans=0;
vector <int> e[100005];
bool isP(ll x)/**< 平方数判定 */
{
    ll y=(int)sqrt(x);
    return y*y==x;
}
int main()
{
    ios::sync_with_stdio(0),cin.tie(0);
    int i,j,x,y;
    cin>>n;
    for(i=1;i<=n;i++)
        cin>>a[i];
    for(i=1;i<n;i++)
    {
        cin>>x>>y;
        e[x].push_back(y),e[y].push_back(x);
        d[x]++,d[y]++;/**< d数组统计度 */
    }
    queue<int>q;
    for(i=1;i<=n;i++)
        if(d[i]==1)/**< 度为1的入队*/
        q.push(i);
    while(q.size())/**< 模拟拓扑排序处理方法 */
    {
        x=q.front();
        v[x]=1;/**< 标记这个节点,避免重新被入队处理 */
        q.pop();
        for(i=0;i<e[x].size();i++)/**< 找到x所有邻接点 */
        {
            y=e[x][i];
            if(v[y]) /**< y已经访问过,其实y一定是x的子节点 */
                continue;
            if(red[x]==0&&red[y]==0&&isP(a[x]*a[y]))/**< x和它父节点y满足条件 */
                red[x]=red[y]=1,ans+=2;/**< 标记颜色,计数器+2 */
            d[y]--;
            if(d[y]==1)/**< 度为1说明y子节点都处理完了,此时可以入队 */
              q.push(y);
        }
    }
    cout<<ans;
    return 0;
}
相关推荐
没书读了6 小时前
考研复习-数据结构-第六章-图
数据结构
future14128 小时前
C#进阶学习日记
数据结构·学习
ChoSeitaku11 小时前
NO.4数据结构数组和矩阵|一维数组|二维数组|对称矩阵|三角矩阵|三对角矩阵|稀疏矩阵
数据结构·人工智能·矩阵
Paper Clouds12 小时前
代码随想录|图论|15并查集理论基础
数据结构·算法·leetcode·深度优先·图论
Paper Clouds12 小时前
代码随想录|图论|14有向图的完全可达性
数据结构·算法·深度优先·图论·宽度优先
闻缺陷则喜何志丹14 小时前
【并集查找 虚拟节点】P1783 海滩防御|省选-
数据结构·c++·洛谷·并集查找·虚拟节点
7 9715 小时前
C语言基础知识--柔性数组
数据结构·算法
双叶83615 小时前
(C++)STL标准库(vector动态数组)(list列表)(set集合)(map键值对)相关对比,基础教程
c语言·开发语言·数据结构·c++·list
lifallen17 小时前
Flink Exactly Once 和 幂等
java·大数据·数据结构·数据库·分布式·flink
不二青衣1 天前
牛客网50题
数据结构·c++·算法