题意:某公司要举办一次晚会,但是为了使得晚会的气氛更加活跃,每个参加晚会的人都不希望在晚会中见到他的直接上司,现在已知每个人的活跃指数和上司关系(当然不可能存在环),求邀请哪些人(多少人)来能使得晚会的总活跃指数最大。
思路:水题一道,树状数组的裸题。
dp[i][0]=dp[v][1] +m[i]
dp[i][1]=max(dp[v][1],dp[v][0])
0表示出场,1表示没出场,v为i的下属
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 6009
using namespace std;
struct node
{
int v,next;
}s[N];
int head[N],m[N],cnt,used[N],pre[N],dp[N][2];
void addedge(int u,int v)
{
s[cnt].v=v;
s[cnt].next=head[u];
head[u]=cnt++;
}
void dfs(int u)
{
used[u]=1;
for(int i=head[u];i!=-1;i=s[i].next)
{
int v=s[i].v;
if(!used[v])
{
dfs(v);
}
}
dp[u][0]=m[u];
dp[u][1]=0;
for(int j=head[u];j!=-1;j=s[j].next)
{
int v=s[j].v;
dp[u][0]+=dp[v][1];
dp[u][1]+=max(dp[v][0],dp[v][1]);
}
}
int main()
{
int n,u,v;
//FILE *fp=fopen("t.txt","r");
while(scanf("%d",&n)!=EOF)
{
cnt=0;
memset(head,-1,sizeof(head));
memset(pre,0,sizeof(pre));
for(int i=1;i<=n;i++)
{
scanf("%d",&m[i]);
}
for(int i=1;i<=n;i++)
{
scanf("%d%d",&v,&u);
if(u==0&&v==0) break;
addedge(u,v);
pre[v]=1;
}
memset(used,0,sizeof(used));
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
{
if(!used[i])
{
dfs(i);
}
}
int ans=0;
for(int i=1;i<=n;i++)
{
if(!pre[i])
{
ans=max(dp[i][0],dp[i][1]);
break;
}
}
printf("%d\n",ans);
}
return 0;
}