URAL 1039 Anniversary Party 树形dp

原创于 2014-01-12 18:37:33 发布 · 655 阅读

0 ·

CC 4.0 BY-SA版权

解题报告同时被 2 个专栏收录

260 篇文章

订阅专栏

88 篇文章

订阅专栏

本文介绍了一种结合图遍历与动态规划算法求解特定问题的方法。通过定义图结构并实现深度优先搜索（DFS）算法，该方法能够有效地解决带权有向图中的最优化问题。文章详细展示了如何初始化图、添加边以及使用递归方式实现DFS遍历，并通过动态规划技术避免重复计算。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

-----------

const int maxn=10000;
const int maxm=110000;
int n;

struct EdgeNode{
    int to,next;
}edges[maxm];
int head[maxn],edge;
void init(){
    memset(head,-1,sizeof(head));
    edge=0;
}
void addedge(int u,int v){
    edges[edge].to=v,edges[edge].next=head[u],head[u]=edge++;
}
int a[maxn];
int f[maxn][2];
int id[maxn];
int dfs(int u,int flag){
    if (f[u][flag]!=-1) return f[u][flag];
    int res;
    if (flag){
        res=0;
        for (int i=head[u];i!=-1;i=edges[i].next){
            int v=edges[i].to;
            res+=dfs(v,0);
        }
    }
    else{
        int res1=0,res2=0;
        for (int i=head[u];i!=-1;i=edges[i].next){
            int v=edges[i].to;
            res1+=dfs(v,1);
        }
        for (int i=head[u];i!=-1;i=edges[i].next){
            int v=edges[i].to;
            res2+=dfs(v,0);
        }
        res=max(res1+a[u],res2);
    }
    return f[u][flag]=res;
}
int main(){
    while (~scanf("%d",&n)){
        init();
        memset(f,-1,sizeof(f));
        memset(id,0,sizeof(id));
        for (int i=1;i<=n;i++) scanf("%d",&a[i]);
        int x,y;
        while (~scanf("%d%d",&x,&y)){
            if (x==0&&y==0) break;
            addedge(y,x);
            id[x]++;
        }
        int rt=1;
        for (int i=1;i<=n;i++){
            if (id[i]==0) rt=i;
        }
        printf("%d\n",dfs(rt,0));
    }
	return 0;
}

-----------