poj3659 Cell Phone Network（最小支配集-树形dp）

题目链接：点击打开链接

题目描述：给定一棵树，从中选取尽量少的点使每个点要么被选中，要么和被选中的点直接相连？

解题思路：树上的最小支配集，树形dp

dp[i][0]:选中i作为支配集

dp[i][1]:不选i作为支配集，但其子节点覆盖了i

dp[i][2]:不选i作为支配集，而且其子节点没有覆盖i

代码：

#pragma comment(linker,"/STACK:1024000000,1024000000")
#include <cstdio>
#include <cstring>
#include <iostream>
#define MAXN 10010
#define INF 1e9+7
using namespace std;
int head[MAXN],tol;
struct Edge{
int v,next;
}edge[MAXN*2];
void addEdge(int u,int v){
edge[tol].v=v;edge[tol].next=head[u];head[u]=tol++;
edge[tol].v=u;edge[tol].next=head[v];head[v]=tol++;
}
int min(int a,int b){
return a<b?a:b;
}
int dp[MAXN][3];
int n;
void DP(int u,int p){
dp[u][2]=0;
dp[u][0]=1;
int sum=0,inc=INF;
int k,to;
bool s=false;
for(k=head[u];k!=-1;k=edge[k].next){
to=edge[k].v;
if(to==p) continue;
DP(to,u);
dp[u][0]+=min(dp[to][0],min(dp[to][1],dp[to][2]));
if(dp[to][0]<=dp[to][1]){
sum+=dp[to][0];
s=true;
}
else
{
sum+=dp[to][1];
inc=min(inc,dp[to][0]-dp[to][1]);
}
if(dp[to][1]!=INF&&dp[u][2]!=INF) dp[u][2]+=dp[to][1];
else dp[u][2]=INF;
}
if(inc==INF&&!s) dp[u][1]=INF;
else{
dp[u][1]=sum;
if(!s) dp[u][1]+=inc;
}
}
int main(){
while(scanf("%d",&n)!=EOF){
tol=0;memset(head,-1,sizeof(head));
int u,v;
for(int i=0;i<n-1;++i){scanf("%d%d",&u,&v);addEdge(u,v);}
DP(1,1);
printf("%d\n",min(min(dp[1][0],dp[1][1]),dp[1][2]+1));
}
return 0;
}