SGU 195. New Year Bonus Grant

时间:2021-05-16 22:24:13

时间限制:0.75s

空间限制:4M

题意:

在一颗树(最多500000个节点)中,可以对节点染色,但是一个节点染了色后,它的父节点和兄弟节点都不能再染了,求最大的染色节点数,并输出所有染色节点。

Solution:

一开始写了一个树形DP

后来看到有人说题目的输入是从上到下的。。。直接倒过来贪心就好了。ORZ

代码:

#include <iostream>
#include <cstdio>
using namespace std;
const int INF = ;
int f[INF], vis[INF], ans[INF], tol, n;
int main() {
scanf ("%d", &n);
for (int i = ; i < n; i++)
scanf ("%d", &f[i + ]);
for (int i = n; i > ; i--)
if (!vis[i] && !vis[f[i]]) {
vis[f[i]] = vis[i] = ;
ans[++tol] = i;
}
printf ("%d\n", tol * );
for (int i = tol; i > ; i--)
printf ("%d ", ans[i]);
return ;
}