树状DP完整版完整版.doc

树状DP：树+DP

Poj1947

题意：FarmerJohn农场里的粮仓之间通过一条双向路径连接，若两个粮仓之间的道路组成一个无向树。现给定粮仓之间道路的连接方式，问至少摧毁几条路，使得剩下的粮仓组成的子树恰好有P个粮仓。

完整代码：

#includestdio.h

#includestring.h

#includealgorithm

#includevector

#includeiostream

usingnamespacestd;

#defineM160

#defineINF130

vectorintG[M];

intn,p;

intdp[M][M];//dp[i][j]表示以i为根节点保留j个节点需删除的最小边数

intf[M][M];//f[i][j]表示前i个子节点

voidUpdate(introot)

{

inti,j,k;

ints=G[root].size();

dp[root][1]=s;//如果保留的节点为1，那只需剩下root节点，需s次把root的儿子全部删去

if(s==0) return;//找到叶子节点了

for(i=0;is;i++)

for(j=1;j=p;j++)

f[i][j]=INF;

f[0][1]=1;//对于第0号儿子节点，要保留1个节点，只需把根节点与0号节点之间的边删除，即保留根节点

for(i=2;i=p;i++)

f[0][i]=dp[G[root][0]][i-1];//因为有root节点，所以是i-1，即f[0][i]等于root的0号节点保留i-1个节点最少所需删除的边数

for(i=1;is;i++)

{

for(j=1;j=p;j++)

{

for(k=1;k=j-1;k++)//用到第i号节点。前i-1号儿子节点保留k个节点，第i号儿子保留j-k个节点

{

if(f[i-1][k]!=INFdp[G[root][i]][j-k]!=INF)

{

if(f[i][j](f[i-1][k]+dp[G[root][i]][j-k]))

f[i][j]=f[i-1][k]+dp[G[root][i]][j-k];

}

}

if(f[i-1][j]!=INF)//不用i号儿子节点。花1次代价将之删除

{

if(f[i][j]f[i-1][j]+1)

f[i][j]=f[i-1][j]+1;

}

}

}

for(i=1;i=p;i++)

{

dp[root][i]=f[s-1][i];//把f数组的值转移给dp

//printf(***%d%d\n,i,f[s-1][i]);

}

}

voiddfs(introot)

{

inti;

for(i=0;iG[root].size();i++)

dfs(G[root][i]);

Update(root);

return;

}

intmain()

{

inti,j;

while(scanf(%d%d,n,p)!=EOF)

{

for(i=1;i=n;i++)

G[i].clear();

for(i=1;in;i++)

{

inta,b;

scanf(%d%d,a,b);

G[a].push_back(b);

}

for(i=1;i=n;i++)

for(j=1;j=n;j++)

dp[i][j]=INF;

dfs(1);

intans=dp[1][p];

for(i=2;i=n;i++)

if(dp[i][p]+1ans)

{

ans=dp[i][p]+1;

}

printf(%d\n,ans);

}

return0;

}