P1352-没有上司的舞会【树形dp】

正题

评测记录:https://www.luogu.org/recordnew/lists?uid=52918&pid=P1352


题目大意

一棵树,如果选择了子节点那么就不能选择父节点,如果选择了父节点那么就不能选择子节点。
求选择的点的最大权值和。


解题思路

和最大利润类似。


code

#include
#include
#include
using namespace std;
struct tree{
    int x,y,next;
}a[200001];
bool ok[100001];
int n,money[100001],w,ls[100001],f[100001],g[100001],dx,dy;
bool dp(int x)
{
    if (ok[x]) return false;
    ok[x]=true;//标记
    int q=ls[x];
    while (q!=0)
    {
        if (dp(a[q].y))
        {
            f[x]+=g[a[q].y];
            g[x]+=max(f[a[q].y],g[a[q].y]);//动态转移
        }
        q=a[q].next;//下一条边
    }
    f[x]+=money[x];//价值
    return true;
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",&money[i]);
    for (int i=1;iscanf("%d%d",&dx,&dy);
        a[++w].x=dx;a[w].y=dy;
        a[w].next=ls[dx];ls[dx]=w;//邻接表
        a[++w].x=dy;a[w].y=dx;
        a[w].next=ls[dy];ls[dy]=w;
    }
    memset(ok,false,sizeof(ok));//有些莫名其妙的bug所以...
    dp(1);
    printf("%d",max(f[1],g[1]));//输出
}

你可能感兴趣的:(dp)