洛谷1352没有上司的舞会——树型dp

最新推荐文章于 2024-10-04 13:02:05 发布

weixin_33713350

最新推荐文章于 2024-10-04 13:02:05 发布

阅读量111

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Narh/p/8443245.html

本文提供了一种解决洛谷P1352问题的算法实现方案，通过深度优先搜索（DFS）来计算最大收益。代码采用C++编写，展示了如何构造图结构并进行遍历求解。

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

题目：https://www.luogu.org/problemnew/show/P1352

#include<iostream>
#include<cstdio>
using namespace std;
int n,a[6005],l[6005],cnt,x,y,f[6005][2],sum;
struct Node{
    int to,next;
}edge[6005];
bool b[6005],rd[6005];
void add(int x,int y)
{
    cnt++;
    edge[cnt].next=l[x];
    edge[cnt].to=y;
    l[x]=cnt;
    rd[y]=1;
}
void dfs(int cur)
{
    if(b[cur])return;
    for(int i=l[cur];i;i=edge[i].next)
    {
        int v=edge[i].to;
        dfs(v);
        f[cur][0]+=max(f[v][0],f[v][1]);
        f[cur][1]+=f[v][0];
    }
    f[cur][1]+=a[cur];
    b[cur]=1;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&x,&y);
        add(y,x);
    }
    for(int i=1;i<=n;i++)
    {
        dfs(i);
        if(!rd[i])sum=max(sum,sum+max(f[i][0],f[i][1]));
    }
    printf("%d",sum);
    return 0;
}