题意:微软公司要给员工发钱。公司员工是严格的树形结构。规定一个人要么接收上级的钱,要么给下级发钱,两者不能都有。问所有员工最多可以拿到多少钱
思路:我的方法是树形DP。dp[p][0]表示p个人接受上级的钱,dp[p][1]表示p个人给下级发钱。dp[p][0]=sigma{dp[sub][1]}+1000.就是说一个人接受上级的钱的话,他这棵子树最多能拿到的钱应当是下级都往下发的和再加上自己拿的上级的1000块。dp[p][1]是p的一个下属员工接受p的钱,其他的下发的所有情况的最大值。
错误:最开始用深度优先写的,就re了。深度太深了。。。
代码如下:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
const int mmm=500000+100;
struct Edge
{
int u,v,next;
}edge[mmm];
int tot;
int first[mmm];
void add(int uu,int vv)
{
edge[tot].u=uu;
edge[tot].v=vv;
edge[tot].next=first[uu];
first[uu]=tot++;
}
int dp[mmm][2];
int next[mmm];
int record[mmm];
int ts[mmm];
int tcnt=0;
void toposort()
{
queue<int> q;
q.push(1);
tcnt=0;
ts[tcnt++]=1;
while(!q.empty()){
int t=q.front();
q.pop();
for(int i=first[t];i!=-1;i=edge[i].next){
q.push(edge[i].v);
ts[tcnt++]=edge[i].v;
}
}
reverse(ts,ts+tcnt);
}
int solve()
{
for(int i=0;i<tcnt;++i){
int p=ts[i];
dp[p][0]=0;
dp[p][1]=0;
int sav=-1;
int mm=0;
for(int i=first[p];i!=-1;i=edge[i].next){
dp[p][0]+=dp[edge[i].v][1];
int tmp=dp[edge[i].v][0]-dp[edge[i].v][1];
if(tmp>mm){
mm=tmp;
sav=edge[i].v;
}
}
dp[p][1]=dp[p][0]+mm;
if(mm>0)next[p]=sav;
dp[p][0]+=1;
}
return dp[1][1];
}
struct sta
{
int p,f;
sta(int p,int f):p(p),f(f){}
};
int tt=0;
void print(int p,int f)
{
tt=0;
queue<sta> q;
q.push(sta(p,f));
while(!q.empty()){
sta tmp=q.front();
q.pop();
p=tmp.p;
f=tmp.f;
if(f==1){
if(next[p]!=-1)
record[tt++]=next[p];
}
for(int i=first[p];i!=-1;i=edge[i].next){
if(edge[i].v==next[p])
q.push(sta(edge[i].v,0));
else q.push(sta(edge[i].v,1));
}
}
}
int main()
{
// freopen("data.txt","r",stdin);
int n;
scanf("%d",&n);
while(scanf("%d",&n)!=EOF)
{
for(int i=1;i<=n;++i){
first[i]=-1;
}
dp[1][0]=dp[1][1]=-1;
tot=0;
next[1]=-1;
tt=0;
for(int i=2;i<=n;++i){
int a;
scanf("%d",&a);
add(a,i);
dp[i][0]=-1;
dp[i][1]=-1;
next[i]=-1;
}
toposort();
int ans=solve();
printf("%d\n",ans*1000);
if(ans!=0)
{
print(1,1);
sort(record,record+tt);
printf("%d",record[0]);
for(int i=1;i<tt;++i){
printf(" %d",record[i]);
}
printf("\n");
}
}
return 0;
}