题目大意:对于一个员工关系图(树),一个员工只有其或者其下属中的唯一一个可以获得奖金,根节点没有奖金,要求使得所有人获得的奖金数是最多的!
解题思路:可以使用贪心,从叶子节点开始,总是选择最低一层可以获得奖金的人去获得奖金,假设一个树有n层,那么获得奖金的人可能是隔层分布的,而且一个父节点只有一个人获得奖金,所以假设一个树从根节点到叶子节点是一个直线,如果假设叶子节点获得奖金的话,那么获得奖金的总数就是n/2,如果不是叶子节点的人获得奖金,那么总数就是(n-1)/2所以由此可以得出这样的结论的。
#include<iostream>
#include<cmath>
#include<cstdio>
#include<queue>
#include<cstring>
#include<stack>
using namespace std;
#define rep(i,n) for(int i=0; i<n; ++i)
#define repf(i,n,m) for(int i=(n) ; i<=(m);++i)
#define repd(i,n,m) for(int i=(n); i>=(m); --i)
#define ll long long
#define inf 1000000
#define N 500100
int vis[N];
int pre[N];
int main()
{
int test;
int n;
scanf("%d",&test);
while(test--)
{
scanf("%d",&n);
repf(i,2,n)
scanf("%d",&pre[i]);
memset(vis,0,sizeof(vis));//0代表对状态不确定的
int sum=0;
repd(i,n,2)
{
int x=pre[i];
if(vis[x]==0 && vis[i]==0)
{
sum++;
vis[x]=-1;//代表自己已经给下级进行奖金评估了
vis[i]=1;//代表获得了奖金
}
}
cout<<sum*1000<<endl;
int m=0;
repf(i,2,n)
{
if(m!=0) cout<<" ";
if(vis[i]==1)
{
cout<<i;
m++;
}
}
cout<<endl;
// cout<<test<<endl;
if(test!=0)
cout<<endl;
}
// cout<<"sdfs"<<endl;
// getchar();getchar();
return 0;
}