Codeforces 1225 F Tree Factory —— 贪心

This way

题意:

现在你有一根竹子(除叶子结点外其它节点都只有一个儿子的树),你每次可以选择一个点,使得他的父亲节点变成它父亲的父节点。现在给你变换完成的树,让你回推这根竹子,并且输出每次变换的节点编号,并且变换次数要最少。

题解:

其实也是一道规律,,像下面这个图:
在这里插入图片描述
最优的变换策略应该是:
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
也就是将深度最大的放在最后,这样子就可以少移一点。

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
vector<int>son[N];
int dep[N],fa[N];
vector<int>ans;
void dfs(int x){
    int mx=N-1;
    printf("%d ",x);
    if(!son[x].size())
        return ;
    for(int i:son[x])
        if(dep[i]>=dep[mx])
            mx=i;
    int pre=0;
    for(int i:son[x]){
        if(i!=mx){
            dfs(i);
            while(pre--)
                ans.push_back(i);
            pre=dep[i];
        }
    }
    while(pre--)
        ans.push_back(mx);
    dfs(mx);
}
int main()
{
    int n,x;
    scanf("%d",&n);
    for(int i=1;i<n;i++)
        scanf("%d",&x),son[x].push_back(i),fa[i]=x,dep[i]=1;
    for(int i=n-1;~i;i--)
        dep[fa[i]]=max(dep[fa[i]],dep[i]+1);
    dfs(0);
    printf("\n");
    printf("%d\n",ans.size());
    for(int i:ans)
        printf("%d ",i);
    printf("\n");
    return 0;
}

发布了554 篇原创文章 · 获赞 31 · 访问量 5万+

猜你喜欢

转载自blog.csdn.net/tianyizhicheng/article/details/104293527