题意:给出了,参加party的人的关系,但是如果一个人的直接上司出席,那么他就不会出席,每个人有自己的rating。现在想要rating最大。
DFS搜索的时候,回溯更新。
1.dp[u][0]:表示u这个人不去,那么他的手下v可以选择去或者不去。所以 dp[u][0]+=max(dp[v][0],dp[v][1]);
2.dp[i][1]:表示u这个人去, 那么由题意他的手下肯定不会去,所以dp[u][1]+=dp[v][0];
状态表示,以及状态的转移方程有了,下面就是DFS这颗关系树了。
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<set>
#include<map>
#include<string>
#include<cstring>
#include<stack>
#include<queue>
#include<vector>
#include<cstdlib>
#define lson (rt<<1),L,M
#define rson (rt<<1|1),M+1,R
#define M ((L+R)>>1)
#define cl(a,b) memset(a,b,sizeof(a));
#define LL long long
#define P pair<int,int>
#define X first
#define Y second
#define pb push_back
#define fread(zcc) freopen(zcc,"r",stdin)
#define fwrite(zcc) freopen(zcc,"w",stdout)
using namespace std;
const int maxn=6005;
const int inf=999999;
int a[maxn];
int in[maxn];
vector<int> G[maxn];
int dp[maxn][2];
void Tree_dfs(int u){
dp[u][0]=0;//u不去
dp[u][1]=a[u];//u去的话,首先会有自己的rating
int N=G[u].size();
for(int i=0;i<N;i++){//遍历所有子节点
int v=G[u][i];
Tree_dfs(v);
dp[u][0]+=max(dp[v][0],dp[v][1]);//回溯的时候,从叶子节点更新到父亲节点
dp[u][1]+=dp[v][0];//也是回溯的更新
}
}
int main(){
int n;
while(~scanf("%d",&n)){
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
G[i].clear();
in[i]=0;
dp[i][0]=dp[i][1]=0;
}
int x,y;
while(scanf("%d%d",&x,&y)&&(x+y)){
G[y].pb(x);//y是x的上司,不要写反了
in[x]++;//顶点的入度
}
int root=1;
for(int i=1;i<=n;i++)if(in[i]==0){//入度是0的点是根节点
root=i;break;
}
Tree_dfs(root);//从根节点出发搜索
printf("%d\n",max(dp[root][0],dp[root][1]));//答案
}
return 0;
}