树形dp解决一类树的问题

最小支配集：选择尽量少的点覆盖整张图的所有点

USACO 2008 January Gold  Cell Phone Network手机网络

最小支配集树dp模板。详见代码。

#include<bits/stdc++.h>
using namespace std;
#define Inc(i,L,r) for(register int i=(L);i<=(r);++i)
const int N = 1e4+10,inf = 1<<30;
struct Edge{
	int cnt,h[N],to[N<<1],next[N<<1];
	inline void add(int x,int y){
		next[++cnt]=h[x];
		to[cnt]=y;
		h[x]=cnt;
	}
}e;
int n,f[N][3];
/*
以i为根的子树中（包含i） 
f[i][0]：i属于支配集，覆盖整棵子树
f[i][1]：i不属于支配集，i被至少一个儿子覆盖
f[i][2]：i不属于支配集，i不被子树覆盖，被父亲覆盖
*/
void init(){
	scanf("%d",&n);
	Inc(i,1,n-1){
		int x,y;scanf("%d%d",&x,&y);
		e.add(x,y),e.add(y,x);
	}
}
void trdp(int x,int fa){
	f[x][0]=1;
	int sum=0,inc=inf;
	for(int p=e.h[x];p;p=e.next[p]){
		int to=e.to[p];
		if(to==fa)continue;
		trdp(to,x);
		f[x][0]+=min(f[to][0],min(f[to][1],f[to][2]));//x在支配集中，子树随便怎样 
		
		if(f[to][0]<=f[to][1]){//如果有儿子在支配集就不管 
			sum+=f[to][0];
			inc=0;
		}else {//如果儿子没有一个在支配集中，强制补一个儿子进入支配集 
			sum+=f[to][1];
			inc=min(inc,f[to][0]-f[to][1]);//之所以inc不直接补1是因为根据定义 
		}
		
		if(f[to][1]!=inf&&f[x][2]!=inf)f[x][2]+=f[to][1];//儿子没有儿子 
        else f[x][2]=inf;
	}
	f[x][1]=sum+inc;
}
int main(){
	init();
	trdp(1,0);
	cout<<min(f[1][0],f[1][1])<<"\n";
	return 0;
}

最小点覆盖：选择尽量少的点覆盖整张图的所有边

hdu1054 Strategic Game

#include<bits/stdc++.h>
using namespace std;
#define Inc(i,L,r) for(register int i=(L);i<=(r);++i)
const int N = 2333;
struct Edge{
	int cnt,h[N],to[N<<1],next[N<<1];
	inline void add(int x,int y){
		next[++cnt]=h[x];
		to[cnt]=y;
		h[x]=cnt;
	}
}e;
int n,f[N][2];
/*
f[i][0]i不属于最小点覆盖，以i为根的子树被完全覆盖的最小点数 
f[i][1]i属于最小点覆盖，以i为根的子树被完全覆盖的最小点数 
*/
void trdp(int x,int fa){
	f[x][0]=0;
	f[x][1]=1;
	for(int p=e.h[x];p;p=e.next[p]){
		int to=e.to[p];
		if(to==fa)continue;
		trdp(to,x);
		f[x][0]+=f[to][1];
		f[x][1]+=min(f[to][0],f[to][1]);
	}
}
void addedge(int x){
	char c=getchar();while(c!='(')c=getchar();
	int y,num;scanf("%d%c",&num,&c);
	Inc(i,1,num){
		scanf("%d",&y);
		e.add(x+1,y+1),e.add(y+1,x+1);
	}
}
void init(){
	e.cnt=0;
	memset(f,0,sizeof(f));
	memset(e.h,0,sizeof(e.h));
	Inc(i,1,n){
		int u;scanf("%d",&u);
		addedge(u);
	}
}
int main(){
	while(~scanf("%d",&n)){
		init();
		trdp(1,0);
		cout<<min(f[1][0],f[1][1])<<"\n";
	}
	return 0;
}

最大独立集：选择尽量多的点，使它们之间没有公共边

TYVJ1052 没有上司的舞会

Description

Ural大学有N个职员，编号为1~N。他们有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。每个职员有一个快乐指数。现在有个周年庆宴会，要求与会职员的快乐指数最大。但是，没有职员愿和直接上司一起与会。

Input

第一行一个整数N。(1<=N<=6000)
接下来N行，第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
接下来N-1行，每行输入一对整数L,K。表示K是L的直接上司。
最后一行输入0,0。

Output

输出最大的快乐指数。

Sample Input

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

Sample Output

5

类似最大独立集（将权值改为1即可）。

#include<bits/stdc++.h>
using namespace std;
#define Inc(i,L,r) for(register int i=(L);i<=(r);++i)
const int N = 1e4+10;
struct Edge{
	int cnt,h[N],to[N<<1],next[N<<1];
	inline void add(int x,int y){
		next[++cnt]=h[x];
		to[cnt]=y;
		h[x]=cnt;
	}
}e;
int n,rt,p[N],a[N];
void init(){
	scanf("%d",&n);
	Inc(i,1,n)scanf("%d",&a[i]);
	Inc(i,1,n){
		int x,y;scanf("%d%d",&x,&y);
		e.add(x,y),e.add(y,x);
		p[x]=y;
	}
	Inc(i,1,n)if(!p[i])rt=i;
}
int f[N][2];
void trdp(int x,int fa){
	f[x][0]=0;
	f[x][1]=a[x];
	for(int p=e.h[x];p;p=e.next[p]){
		int to=e.to[p];
		if(to==fa)continue;
		trdp(to,x);
		f[x][0]+=max(f[to][0],f[to][1]);
		f[x][1]+=f[to][0];
	}
}
int main(){
	init();
	trdp(rt,0);
	cout<<max(f[rt][0],f[rt][1])<<"\n";
	return 0;
}