bzoj 4346: [POI2016]Nadajniki 树形动规

       首先一个点最多放两个wifi；不妨考虑用树形动规得到以某个点i的子树中wifi覆盖的最优解，那么转移的时候只需要让这个点到其儿子的路径覆盖wifi即可。有多种情况：

       1.在这个点放wifi。那么可以用p[i][1],p[i][2]表示放1、2个wifi的最优解，如果i放了wifi，那么它到儿子的路径显然已经被覆盖，用儿子的对应状态转移一下即可。

       2.不放wifi。那么i到儿子的路径，要么被i的儿子覆盖，要么被i的儿子的儿子和i的父亲覆盖。令f[i][x][y]表示i不放的时候，父亲节点放x个，儿子节点放y个转移即可。

       第一种情况的转移是很方便的，第二种转移就要麻烦的多；然后根据转移可以把f[i][][]的9个状态缩一下，然后就得到了Claris的博客里的题解。

AC代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#define inf 1000000000
#define N 200005
using namespace std;

int n,tot,fst[N],pnt[N<<1],nxt[N<<1],f[N][3][3],g[N][3],p[N][3],q[N][3],c[3][3];
int read(){
	int x=0; char ch=getchar();
	while (ch<'0' || ch>'9') ch=getchar();
	while (ch>='0' && ch<='9'){ x=x*10+ch-'0'; ch=getchar(); }
	return x;
}
void dn(int &x,int y){ if (y<x) x=y; }
void add(int x,int y){
	pnt[++tot]=y; nxt[tot]=fst[x]; fst[x]=tot;
}
void dfs(int x,int fa){
	int edg,i,j,k,tmp;
	p[x][1]=1; p[x][2]=2;
	memset(g[x],0x3f,sizeof(g[x]));
	memset(f[x],0x3f,sizeof(f[x])); f[x][0][0]=0;
	for (edg=fst[x]; edg; edg=nxt[edg]){
		int y=pnt[edg];
		if (y!=fa){
			dfs(y,x); int tmp=min(min(p[y][1],p[y][2]),q[y][1]);
			for (i=0; i<3; i++) dn(tmp,g[y][i]);
			p[x][1]+=tmp; p[x][2]+=min(tmp,q[y][2]);
			memset(c,0x3f,sizeof(c));
			for (i=0; i<3; i++)
				for (j=0; j<3; j++) if (f[x][i][j]<inf){
					tmp=f[x][i][j];
					for (k=1; k<3; k++) dn(c[min(2,i+k)][j],tmp+p[y][k]);
					for (k=0; k<3; k++) dn(c[i][max(j,2-k)],tmp+g[y][k]);
				}
			memcpy(f[x],c,sizeof(c));
		}
	}
	for (i=0; i<3; i++)
		for (j=0; j<=i; j++) dn(g[x][i],f[x][i][j]);
	q[x][1]=min(f[x][0][1],f[x][1][2]);
	q[x][2]=f[x][0][2];
}
int main(){
	n=read(); int i,x,y;
	for (i=1; i<n; i++){
		x=read(); y=read();
		add(x,y); add(y,x);
	}
	dfs(1,0); int ans=min(p[1][1],p[1][2]);
	for (i=0; i<3; i++) dn(ans,g[1][i]);
	printf("%d\n",ans);
	return 0;
}

by lych

2016.3.31