赶着学的树形DP,就先从最最最最最裸的树上DP题开始做起了,
前六道: ye-leng大佬的六道起步题。
一号板子题:
Anniversary party [POJ-2342]
题意:
给出每个点的快乐值当做点权,员工快乐时,他的直接领导不能在场,求最大快乐值。
题解:
f
[
i
]
[
j
]
:
f[i][j]:
f[i][j]:表示第i个人状态为j(表示选或不选)的最大快乐值
状态转移方程式:
f
[
f
a
]
[
1
]
+
=
f
[
i
]
[
0
]
f[fa][1]+=f[i][0]
f[fa][1]+=f[i][0]//父亲的快乐+=儿子不快乐
f
[
f
a
]
[
0
]
+
=
m
a
x
(
f
[
i
]
[
0
]
,
f
[
i
]
[
1
]
)
f[fa][0]+=max(f[i][0],f[i][1])
f[fa][0]+=max(f[i][0],f[i][1])父亲不快乐+=儿子不快乐/快乐
代码:
#include<iostream>
#include<stdio.h>
using namespace std;
const int sea=6060;
inline int read()
{
int s=0,w=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
while(ch<='9'&&ch>='0')s=s*10+ch-'0',ch=getchar();
return s*w;
}
int n,a[sea],x,y,root,fa[sea],v[sea],f[sea][2];//f[i][j]:表示第i个人状态为j(表示选或不选)的最大快乐值
void dp(int pa)
{
v[pa]=1;
for(int i=1;i<=n;i++)
{
if(v[i]||fa[i]!=pa) continue;
dp(i);
f[pa][1]+=f[i][0];
f[pa][0]+=max(f[i][0],f[i][1]);
}
}
int main()
{
n=read();
for(int i=1;i<=n;i++) f[i][1]=read();
for(int i=1;i<=n;i++)
{
x=read(); y=read();
if(x==0&&y==0) break;
fa[x]=y;
}
for(int i=1;i<=n;i++)
{
if(fa[i]!=0) continue;
else root=i;
}
dp(root);
int ans=max(f[root][1],f[root][0]);
printf("%d",ans);
return 0;
}
二号板子题
Strategic game [POJ-1463]
题意:
给一棵树放士兵,士兵可以管辖此点及此点的儿子,求最少的士兵放置数。
题解:
f
[
i
]
[
j
]
f[i][j]
f[i][j]:表示第i个节点的状态为j(有无人看管)
状态转移方程:
f
[
f
a
]
[
0
]
+
=
f
[
i
]
[
1
]
f[fa][0]+=f[i][1]
f[fa][0]+=f[i][1]//父亲有没人管,儿子要有人管
f
[
f
a
]
[
1
]
+
=
m
i
n
(
f
[
i
]
[
0
]
,
f
[
i
]
[
1
]
)
f[fa][1]+=min(f[i][0],f[i][1])
f[fa][1]+=min(f[i][0],f[i][1])//父亲有人管,儿子可管可不管
代码:
#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
const int sea=3030;
bool fa[sea];
int n,root,f[sea][2],brother[sea],son[sea];
void dp(int pa)
{
f[pa][0]=0; f[pa][1]=1;
int ff=son[pa];
while(ff!=-1)
{
dp(ff);
f[pa][1]+=min(f[ff][0],f[ff][1]);
f[pa][0]+=f[ff][1];
ff=brother[ff];
}
}
int main()
{
// freopen("a.txt","r",stdin);
while(scanf("%d",&n)!=EOF)
{
memset(son,-1,sizeof(son));
memset(fa,0,sizeof(fa));
for(int i=0;i<n;i++)
{
int m,x,y;
scanf("%d: (%d)",&y,&m);
for(int i=0;i<m;i++)
{
scanf("%d",&x);
brother[x]=son[y];//这个儿子的兄弟是y的上一个儿子
son[y]=x; //这个儿子也是y的儿子
fa[x]=1;//这个儿子有父亲
}
}
for(int i=0;i<n;i++) if(!fa[i]){root=i; break;}
dp(root);
int ans=min(f[root][1],f[root][0]);
printf("%d\n",ans);
}
return 0;
}
三号板子题:
题目链接:Tree Cutting [POJ 2378]
题意:一棵树删点,使删过几个点之后的联通块总点数<=n / 2,求点
题解:
这个题其实跟DP没有什么关系,,,,
枚举每个点,判断如果它的子树大小 <= n/2&&n – 它的子树大小<=n/2
就可删。
代码:
#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
inline int read()
{
int s=0,w=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
while(ch<='9'&&ch>='0')s=s*10+ch-'0',ch=getchar();
return s*w;
}
const int sea=10100l;
struct see{int x,ver,next;}e[sea<<1];
int f[sea],n,v[sea],last[sea],tot,cnt=0;
void add(int x,int y){e[++tot].x=x;e[tot].ver=y;e[tot].next=last[x];last[x]=tot;}
void dfs(int x,int fa)
{
bool flag=1;
for(int i=last[x];i!=-1;i=e[i].next)
{
int y=e[i].ver; if(y==fa) continue;
dfs(y,x);
f[x]+=f[y];
if(f[y]>n/2) flag=0;
}
if(n-f[x]>n/2) flag=0;
if(flag) v[x]=1;
}
int main()
{
while(scanf("%d",&n)!=EOF)//记得是多组数据哦
{
memset(last,-1,sizeof(last));
for(int i=1;i<=n;i++) f[i]=1;
for(int i=1;i<n;i++)
{
int x,y;
x=read(); y=read();
add(x,y); add(y,x);
}
dfs(1,-1); int flag=0;
for(int i=1;i<=n;i++) if(v[i]) printf("%d\n",i),flag=1;
if(!flag) puts("NONE");
}
return 0;
}
四号板子题:
题目链接:Tree of Tree [ZOJ-3201]
题意:
求一棵树大小为k的子树最大的权值和。
题解:
f [ i ] [ j ] f[i][j] f[i][j]:表示的是第i个节点体积为j时的最大子树的总权重(有点类似于01背包)
状态转移方程:
f
[
x
]
[
j
]
=
m
a
x
(
f
[
x
]
[
j
]
,
f
[
x
]
[
j
−
k
]
+
f
[
y
]
[
k
]
)
f[x][j]=max(f[x][j],f[x][j-k]+f[y][k])
f[x][j]=max(f[x][j],f[x][j−k]+f[y][k])
代码:
#include<bits/stdc++.h>
using namespace std;
inline int read()
{
int s=0,w=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
while(ch<='9'&&ch>='0')s=s*10+ch-'0',ch=getchar();
return s*w;
}
const int sea=110;
struct see{int x,ver,next;}e[sea<<1];
int n,m,ans,a[sea],last[sea],tot;
int f[sea][sea];//f[i][j]表示的是第i个节点体积为j时的最大子树的总权重,类似于01背包
void add(int x,int y){e[++tot].x=x;e[tot].ver=y;e[tot].next=last[x];last[x]=tot;}
void dfs(int x,int fa)
{
f[x][1]=a[x];
for(int i=last[x];i!=-1;i=e[i].next)
{
int y=e[i].ver;if(y==fa) continue;
dfs(y,x);
for(int j=m;j>0;j--) for(int k=0;k<j;k++)
f[x][j]=max(f[x][j],f[x][j-k]+f[y][k]);
}
ans=max(ans,f[x][m]);
return ;
}
int main()
{
int x,y;
while(scanf("%d%d",&n,&m)!=EOF)
{
memset(f,0,sizeof(f));
memset(last,-1,sizeof(last));
for(int i=0;i<n;i++) a[i]=read();
for(int i=1;i<n;i++)
{
x=read(); y=read();
add(x,y); add(y,x);
}
ans=0; dfs(0,-1);
printf("%d\n",ans);
}
return 0;
}
五号板子题:
Cell Phone Network [POJ 3659]
题意:求一棵树的最小支配集
题解:
f
[
i
]
[
2
]
f[i][2]
f[i][2]在此节点的子节点建塔
f
[
i
]
[
0
]
f[i][0]
f[i][0]在此节点的父亲点建塔
f
[
i
]
[
1
]
f[i][1]
f[i][1]在此点自身建塔
状态转移方程:
f
[
i
]
[
2
]
=
m
i
n
(
f
[
i
]
[
0
]
,
f
[
i
]
[
1
]
,
f
[
i
]
[
2
]
)
f[i][2]=min(f[i][0],f[i][1],f[i][2])
f[i][2]=min(f[i][0],f[i][1],f[i][2])
f
[
i
]
[
1
]
=
m
i
n
(
f
[
i
]
[
0
]
,
f
[
i
]
[
1
]
)
f[i][1]=min(f[i][0],f[i][1])
f[i][1]=min(f[i][0],f[i][1])
f
[
i
]
[
0
]
=
m
i
n
(
f
[
i
]
[
0
]
,
f
[
i
]
[
1
]
)
f[i][0]=min(f[i][0],f[i][1])
f[i][0]=min(f[i][0],f[i][1])
对于f[i][1]的情况还是需要特别处理一下的,对于遇到像下面这样的点(当我们在g位置放过了时,就会发现s1或s2点不能被覆盖)时,我们就特别取一下,(就如下面代码)
......
if(f[y][2]<f[y][1])
f[x][2]+=f[y][2],ccc=min(ccc,f[y][1]-f[y][2]);//直接计算差值,就是特例的个数
else f[x][2]+=f[y][1],flag=1;//打个标记记录一下这个
......
if(!flag) f[x][2]+=ccc;//判断标记,加上差值
代码:
#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
inline int read()
{
int s=0,w=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
while(ch<='9'&&ch>='0')s=s*10+ch-'0',ch=getchar();
return s*w;
}
const int sea=10100l;
const int inf=999999;
struct see{int x,ver,next;}e[sea<<1];
int f[sea][3],n,last[sea],tot,ans;
void add(int x,int y){e[++tot].x=x;e[tot].ver=y;e[tot].next=last[x];last[x]=tot;}
void dfs(int x,int fa)
{
int ccc=inf;
bool flag=0;
for(int i=last[x];i;i=e[i].next)
{
int y=e[i].ver;if(y==fa) continue;
dfs(y,x);
f[x][1]+=min(f[y][1],min(f[y][0],f[y][2]));
f[x][0]+=min(f[y][1],f[y][2]);
if(f[y][2]<f[y][1])
f[x][2]+=f[y][2],ccc=min(ccc,f[y][1]-f[y][2]);
else f[x][2]+=f[y][1],flag=1;
}
if(!flag) f[x][2]+=ccc;
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
for(int i=1;i<=n;i++) f[i][1]=1,f[1][0]=0,f[2][0];
for(int i=1;i<n;i++)
{
int x=read(),y=read();
add(x,y); add(y,x);
}
dfs(1,0);
printf("%d\n",min(f[1][1],f[1][2]));
}
return 0;
}
六号板子题:
题目链接:Computer [HDU-2196]
题意:求树上的每个点到达整棵树对于这个顶点的最远距离。
题解:
(这个题啊,告诉我们不要在不常用的OJ上交题,否则就光CE就能CE好长时间,,o(╥﹏╥)o)
(正经的题解:)
对于一个节点,要想找他的最远距离,就只能从两个方向,一个是从他的父亲节点的方向,第二个就是从他的儿子节点的方向找,但是,但是,当你在向他的父亲方向找距离的时候,又时候会找到一条路径,这条路径是最大的,而且这条路径包括了此点,这样子就多找了一个点,所以,就可以在维护最大值得时候,顺便再维护一个次大值,这样的话就解决了这类问题,最重要的部分就是这样了,其他的就在注释里吧。
代码:
#include<iostream>
#include<cstdio>
#include<string.h>
#include<algorithm>
#define inf 1000000000
using namespace std;
inline int read()
{
int s=0,w=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
while(ch<='9'&&ch>='0')s=s*10+ch-'0',ch=getchar();
return s*w;
}
const int sea=10005;
struct see{int ver,next,edge;}e[20005];
int f[sea],g[sea],n,last[20005],tot,ans[sea];
//f[i]存的是第i个点父亲节点的最大距离
//g[i]存的是第i个点儿子节点的最大距离
void add(int x,int y,int z){e[++tot].edge=z;e[tot].ver=y;e[tot].next=last[x];last[x]=tot;}
void dp(int x,int fa)
{
for(int i=last[x];i;i=e[i].next)
{
int y=e[i].ver,z=e[i].edge;
if(y!=fa) dp(y,x),f[x]=max(f[x],f[y]+z);
}
}
void so(int x,int fa)
{
int p=0,q=0,p1=0,q1=0;
for(int i=last[x];i;i=e[i].next)//最大值
{
int y=e[i].ver,z=e[i].edge;
if(y!=fa&&f[y]+z>=f[p]+p1) p=y,p1=z;
}
for(int i=last[x];i;i=e[i].next)//次大值
{
int y=e[i].ver,z=e[i].edge;
if(y!=fa&&y!=p)
{
g[y]=ans[y]=f[p]+z+p1;
if(f[y]+z>=f[q]+q1) q=y,q1=z;
}
}
g[p]=ans[p]=f[q]+p1+q1;
for(int i=last[x];i;i=e[i].next)//更新子节点
{
int y=e[i].ver,z=e[i].edge;
if(y!=fa) g[y]=max(g[y],g[x]+z);
}
for(int i=last[x];i;i=e[i].next)//更新答案
{
int y=e[i].ver,z=e[i].edge;
if(y!=fa) ans[y]=max(ans[y],g[x]+z),so(y,x);
}
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
tot=0;
memset(last,0,sizeof(last));
memset(f,0,sizeof(f));
memset(g,0,sizeof(g));
for(int i=2;i<=n;i++)
{
int x=read(),z=read();
add(x,i,z); add(i,x,z);
}
f[0]=g[0]=-inf;
dp(1,0);so(1,0);
for(int i=1;i<=n;i++) printf("%d\n",max(f[i],ans[i]));
}
return 0;
}
这样的话,树形DP入门的六道题就可以完结了,,,(但是树形DP这一章还是不可能就这样翻过去的,,(先发在说,草稿箱里太多了))
好的来补道题吧,,
七号板子题
一道刚考的题:
题目链接:树上距离
题解:
(这个题,我30行+的树形DP直接击溃100+行的点分治(但是不代表不能写点分治))
(一个写DP题的套路:先定义f[]再试状态转移方程)
f
[
i
]
[
j
]
f[i][j]
f[i][j]:以i作为长度为j的路径其中一个点的方案数
状态转移方程:
f
[
x
]
[
j
]
+
=
f
[
y
]
[
j
−
1
]
f[x][j]+=f[y][j-1]
f[x][j]+=f[y][j−1]
A
n
s
+
=
d
p
[
y
]
[
j
]
∗
d
p
[
x
]
[
m
−
j
−
1
]
;
Ans+=dp[y][j]*dp[x][m-j-1];
Ans+=dp[y][j]∗dp[x][m−j−1];
(这里的m就是题目中的k)
代码:
#include<bits/stdc++.h>
using namespace std;
inline int read()
{
int s=0,w=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
while(ch<='9'&&ch>='0')s=s*10+ch-'0',ch=getchar();
return s*w;
}
const int sea=50005;
struct see{int x,ver,next;}e[100005];
int n,m,last[100005],tot,ans,f[sea][520];
void add(int x,int y){e[++tot].x=x;e[tot].ver=y;e[tot].next=last[x];last[x]=tot;}
void dfs(int x,int fa)
{
f[x][0]=1;
for(int i=last[x];i;i=e[i].next)
{
int y=e[i].ver; if(y==fa) continue;
dfs(y,x);
for(int j=0;j<m;j++) ans+=f[y][j]*f[x][m-j-1];
for(int j=1;j<=m;j++) f[x][j]+=f[y][j-1];
}
}
int main()
{
n=read(); m=read();
for(int i=1;i<n;i++)
{
int x,y;
x=read(); y=read();
add(x,y); add(y,x);
}
dfs(1,0);
printf("%d\n",ans);
return 0;
}