P1352 没有上司的舞会 树形dp

　　

题目描述

某大学有N个职员，编号为1~N。他们之间有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。现在有个周年庆宴会，宴会每邀请来一个职员都会增加一定的快乐指数Ri，但是呢，如果某个职员的上司来参加舞会了，那么这个职员就无论如何也不肯来参加舞会了。所以，请你编程计算，邀请哪些职员可以使快乐指数最大，求最大的快乐指数。

输入输出格式

输入格式：

第一行一个整数N。(1<=N<=6000)

接下来N行，第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)

接下来N-1行，每行输入一对整数L,K。表示K是L的直接上司。

最后一行输入0 0

输出格式：

输出最大的快乐指数。

输入输出样例

输入样例#1： 复制

7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0

输出样例#1： 

5


入门树形dp

一开始90分   少考虑了一种情况   

#include<bits/stdc++.h>
using namespace std;
//input by bxd
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i>=(b);--i)
#define RI(n) scanf("%d",&(n))
#define RII(n,m) scanf("%d%d",&n,&m)
#define RIII(n,m,k) scanf("%d%d%d",&n,&m,&k)
#define RS(s) scanf("%s",s);
#define ll long long
#define pb push_back
#define REP(i,N)  for(int i=0;i<(N);i++)
#define CLR(A,v)  memset(A,v,sizeof A)
//////////////////////////////////
#define inf 0x3f3f3f3f
const int N=6000+5;
const int M=50005;
int head[M],pos;
struct Edge
{
    int nex,to,v;
}edge[M];
void add(int a,int b)
{
    edge[++pos].nex=head[a];
    head[a]=pos;
    edge[pos].to=b;
}
int n,m;
int dp[N][N];
int siz[N];
int node[N];

int vis[N];

void dfs(int u,int fa)
{
    for(int i=head[u];i;i=edge[i].nex)
    {
        int v=edge[i].to;
        if(v==fa)continue;
        dfs(v,u);
        dp[u][1]=max(dp[u][1],dp[u][1]+dp[v][0]);
        dp[u][0]=max(dp[u][0],dp[u][0]+dp[v][1]);
    }
}

int main()
{
    RI(n);
    rep(i,1,n)
    RI(node[i]),dp[i][1]=node[i];
    
    int a,b;
    while(RII(a,b))
    {
        if(a==0&&b==0)break;
        add(a,b);add(b,a);
        vis[a]=1;
    }
   
    int root;
    rep(i,1,n)
    if(!vis[i])root=i;

    dfs(root,0);
    
    cout<<max( dp[root][0],dp[root][1] );

    return 0;
}

修改后：

#include<bits/stdc++.h>
using namespace std;
//input by bxd
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i>=(b);--i)
#define RI(n) scanf("%d",&(n))
#define RII(n,m) scanf("%d%d",&n,&m)
#define RIII(n,m,k) scanf("%d%d%d",&n,&m,&k)
#define RS(s) scanf("%s",s);
#define ll long long
#define pb push_back
#define REP(i,N)  for(int i=0;i<(N);i++)
#define CLR(A,v)  memset(A,v,sizeof A)
//////////////////////////////////
#define inf 0x3f3f3f3f
const int N=6000+5;
const int M=50005;
int head[M],pos;
struct Edge
{
    int nex,to,v;
}edge[M];
void add(int a,int b)
{
    edge[++pos].nex=head[a];
    head[a]=pos;
    edge[pos].to=b;
}
int n,m;
int dp[N][N];
int siz[N];
int node[N];

int vis[N];

void dfs(int u,int fa)
{
    for(int i=head[u];i;i=edge[i].nex)
    {
        int v=edge[i].to;
        if(v==fa)continue;
        dfs(v,u);
        dp[u][1]=max(dp[u][1],dp[u][1]+dp[v][0]);
        dp[u][0]=max(dp[u][0],max(dp[v][0]+dp[u][0],dp[v][1]+dp[u][0]) );
    }
}

int main()
{
    RI(n);

    rep(i,1,n)
    {
        RI(node[i]);

    if(node[i]>0)dp[i][1]=node[i];
    }
    int a,b;
    while(RII(a,b))
    {
        if(a==0&&b==0)break;
        add(a,b);add(b,a);
        vis[a]=1;
    }

    int root;
    rep(i,1,n)
    if(!vis[i])root=i;

    dfs(root,0);

    cout<<max( dp[root][0],dp[root][1] );

    return 0;
}

大佬的做法：

#include<bits/stdc++.h>
using namespace std;
//input by bxd
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i>=(b);--i)
#define RI(n) scanf("%d",&(n))
#define RII(n,m) scanf("%d%d",&n,&m)
#define RIII(n,m,k) scanf("%d%d%d",&n,&m,&k)
#define RS(s) scanf("%s",s);
#define ll long long
#define pb push_back
#define REP(i,N)  for(int i=0;i<(N);i++)
#define CLR(A,v)  memset(A,v,sizeof A)
//////////////////////////////////
#define inf 0x3f3f3f3f
const int N=6000+5;
const int M=50005;
int head[M],pos;
struct Edge
{
    int nex,to,v;
}edge[M];
void add(int a,int b)
{
    edge[++pos].nex=head[a];
    head[a]=pos;
    edge[pos].to=b;
}
int n,m;
int dp[N][2];
int node[N];

int vis[N];

void dfs(int u,int fa)
{
    for(int i=head[u];i;i=edge[i].nex)
    {
        int v=edge[i].to;
        if(v==fa)continue;
        dfs(v,u);

        dp[u][0]+=max( dp[v][1],dp[v][0] );
        dp[u][1]+=dp[v][0];
    }
}

int main()
{
    RI(n);

    rep(i,1,n)
    {
        RI(node[i]);
        dp[i][1]=node[i];
    }

    int a,b;
    while(RII(a,b))
    {
        if(a==0&&b==0)break;
        add(a,b);add(b,a);
        vis[a]=1;
    }

    int root;
    rep(i,1,n)
    if(!vis[i])root=i;

    dfs(root,0);

    cout<<max( dp[root][0],dp[root][1] );

    return 0;
}