树形dp 入门

以前看过点树形dp,不过全忘了。今天做了一道入门级的简单题,回忆了一下。所谓树形dp就是在一棵树上进行状态转移。有时候他的状态转移方程比普通dp还简单,只不过建图比较麻烦,顺便复习了一下邻接表的用法,真是好久不练就都忘了。

下面是一道最基础的题,要求父节点和儿子节点不能同时选,求最大权值,很明显状态转移方程就是dp[i][1]+=dp[j][0]; dp[i][0]+=max(dp[j][0],dp[j][1])

其中0表示不选,1表示选。进行dp其实就是一个递归的过程,先求出儿子节点的结果,再去更新父节点。

#include
#include
int dp[100010][2];
int head[100010];
struct s{
    int to,next;
};
s edg[200010];
int cnt;
void add(int u,int v){
    edg[cnt].to=v;
    edg[cnt].next=head[u];
    head[u]=cnt++;
    edg[cnt].to=u;
    edg[cnt].next=head[v];
    head[v]=cnt++;
}
void dfs(int x,int y){
    for(int i=head[x];i!=-1;i=edg[i].next){
        int v=edg[i].to;
        if(v==y) continue;
        dfs(v,x);
        dp[x][1]+=dp[v][0];
        dp[x][0]+=dp[v][0]>dp[v][1]?dp[v][0]:dp[v][1];
    }
}
int main(){
    int n;
    cnt=0;
    scanf("%d",&n);
    int u,v;
    memset(dp,0,sizeof(dp));
    memset(head,-1,sizeof(head));
    for(int i=1;i<=n;i++)scanf("%d",&dp[i][1]);
    for(int i=1;idp[1][1]? dp[1][0]:dp[1][1]);
    return 0;
}


你可能感兴趣的:(dp)