LightOJ1201 A Perfect Murder（树形DP）

一道经典的树型DP入门题。dp[u][0/1]表示u点不选或选时以u为根的子树最多能选择的点数。

题目给的有向有环图可以看作森林，注意不是树，因为题目没有说图是连通的！

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAXN 1111
#define MAXM 1111*1111
struct Edge{
    int v,next;
}edge[MAXM];
int NE,head[MAXN];
void addEdge(int u,int v){
    edge[NE].v=v; edge[NE].next=head[u]; head[u]=NE++;
}
int d[MAXN][2];
int dfs(int u,int k,int fa){
    if(d[u][k]!=-1) return d[u][k];
    int res=k;
    for(int i=head[u]; i!=-1; i=edge[i].next){
        int v=edge[i].v;
        if(v==fa) continue;
        if(k) res+=dfs(v,0,u);
        else res+=max(dfs(v,0,u),dfs(v,1,u));
    }
    return d[u][k]=res;
}
int par[MAXN];
int Find(int a){
    while(a!=par[a]){
        par[a]=par[par[a]];
        a=par[a];
    }
    return a;
}
void Union(int a,int b){
    int pa=Find(a),pb=Find(b);
    if(pa==pb) return;
    par[pb]=pa;
}
int main(){
    int t,n,m,a,b;
    scanf("%d",&t);
    for(int cse=1; cse<=t; ++cse){
        NE=0;
        memset(head,-1,sizeof(head));
        scanf("%d%d",&n,&m);
        for(int i=1; i<=n; ++i) par[i]=i;
        while(m--){
            scanf("%d%d",&a,&b);
            addEdge(a,b); addEdge(b,a);
            Union(a,b);
        }
        memset(d,-1,sizeof(d));
        int res=0;
        for(int i=1; i<=n; ++i){
            if(par[i]==i) res+=max(dfs(i,0,0),dfs(i,1,0));
        }
        printf("Case %d: %d
",cse,res);
    }
    return 0;
}