nyoj 376 小强的Linux

来源：互联网发布：医疗教育 gis 数据编辑：程序博客网时间：2024/04/28 11:32

点击打开题目链接

简单树形dp，建立一个总源点就行了。

#include<iostream>#include<cstdio>#include<cstring>#include<vector>#define Max(a,b) a>b?a:busing namespace std;const int INF=1005;int dp[INF],visit[INF],v[INF];vector<int> vec[INF];void dfs(int cur){    if(visit[cur]) return;    int size=vec[cur].size();    if(size==0)    {        dp[cur]=v[cur];        return;    }    for(int i=0;i<size;i++)    {        int child=vec[cur][i];        dfs(child);        visit[child]=1;        dp[cur]=Max(dp[cur],dp[child]+v[cur]);    }}int main(){    int n,m;    while(~scanf("%d",&n))    {        for(int i=1;i<=n;i++)            scanf("%d",&v[i]);        scanf("%d",&m);        int a,b;        for(int i=0;i<m;i++)        {            scanf("%d%d",&a,&b);            vec[a].push_back(b);        }        for(int i=1;i<=n;i++)        {            vec[0].push_back(i);        }        memset(dp,0,sizeof(dp));        memset(visit,0,sizeof(visit));        dfs(0);        printf("%d\n",dp[0]);        for(int i=0;i<=n;i++)            vec[i].clear();    }    return 0;}