【动态规划】之树形DP《收集果子》题解与各部分分附代码

诱敌深入王2小

于 2024-07-29 15:29:19 发布

阅读量312

点赞数 14

分类专栏：动态规划文章标签：动态规划算法 c++

本文链接：https://blog.csdn.net/ExFoler/article/details/140771996

版权

动态规划专栏收录该内容

7 篇文章 0 订阅

订阅专栏

问题描述

一共有 $n$ 个果子分布在 $n$ 个平台上，有的平台有许多个果子，有的平台没有果子。
这 $n$ 个平台间由 $n - 1$ 座桥连接。
$Z$ 君从 $1$ 号平台开始捡果子，最终君捡到了 $n$ 个果子。
$Z$ 君在想，要是有些桥损坏了，我岂不是捡不到 $n$ 个果子了?
于是君想让你算一算，在所有 $2^{n-1}$ 种情况下(每个桥可以损坏或不
损坏)，君有多少种情况可以捡到k个果子。
为了避免 $64$ 位整数输出，请将答案对 $10^9+7$ 取。

输入格式

第一行两个整数 $n, k$ 。
接下来一行 $n$ 个整数，表示每个平台上果子的数量。
接下来 $n - 1$ 行，每行两个整数 $x$ , $y$ ，表示平台连接情况。

输出格式

一行一个整数表示答案。

样例输入

样例输出

【样例解释】
第一个桥坏掉，第二个坏不坏都可以。
【数据规模和约定】
对于 30% 的数据，满足 n,k< 20。
对于 60% 的数据，满足 n,k< 100。
对于 100% 的数据，满足 n,k≤ 1000。

$60$ 分的话其实很好理解，直接一个树上背包

$f [i] [j]$ 表示以 $i$ 为根的子树收集到 $j$ 个果子的方案数

不选 $k$ 号节点的话，它后面的子树节点都不选，一种方案都不剩，这样的话对答案的贡献就相当于是 $f[i][j] * 2^{(size[k])}$
size[k]表示以 $k$ 为根节点的子树的大小

选的话就也跟个背包一样

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod=1e9+7;
const int N=1e3+10;
int n,k;
ll a[N],b[N],c[N];
ll f[N][N],size[N];
int head[N],next[N],ver[N],tot;
void add(ll x,ll y)
{
	ver[++tot]=y;
	next[tot]=head[x];
	head[x]=tot;
}
void dfs(int x,int fa)
{
	size[x]=1;
	f[x][a[x]]=1;
	for(int i=head[x];i;i=next[i])
	{
		int y=ver[i];
		if(y==fa) continue;
		dfs(y,x);
		size[x]+=size[y];
		for(int i=0;i<=k;i++)
		{
            b[i]=c[size[y]-1]*f[x][i];
            for(ll j=0;j<=i;j++)
            {
                b[i]+=f[x][i-j]*f[y][j] % mod;
            }
        }
        for(int i=0;i<=k;i++)
        {
            f[x][i]=b[i];
        }
    }
}
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
    }
    for(int i=1,x,y;i<=n-1;i++)
    {
        scanf("%d%d",&x,&y);
        add(x,y);
		add(y,x);
    }
    c[0]=1;
    for(int i=1;i<=n;i++)
	{
		c[i]=c[i-1]*2%mod;
	}
	dfs(1,-1);
    printf("%lld",f[1][k]);
    return 0;
}

~~30分写法就是在此基础上少取几次mod（我在大佬的30分代码上，多加几个取模就是上面的60分代码）~~

所以 yxc 说：少生孩子多取膜

$30$ 分的就是给所有Bianca编个号，直接用二进制表示有哪些是断的，有哪些是没断的，然后对于每状态做 $D FS$

然后是我考试时候的代码(+后来写的注释)：

//T5 树形dp，f[i][j]表示dp到第i个点收集j个果子的方案数,n三方(还是n方？)60或100
#include<bits/stdc++.h>
using namespace std;
#define N 1010
#define Mod 1000000007
#define int long long
int sz[N],f[N][N],hd[N],n,k,ver[N<<1],a[N],v[N],nxt[N<<1],tot,b[N];
void lian(int x,int y){ver[++tot]=y;nxt[tot]=hd[x];hd[x]=tot;}//正常的连边
void dp(int x,int y)
{
	sz[x]=1;
	for(int i=hd[x];i;i=nxt[i])
	{
		if(ver[i]==y)continue;
		for(int j=n;j>=v[ver[i]];j--)f[ver[i]][j]=f[x][j-v[ver[i]]];//感觉跟背包差不多的状态转移
		dp(ver[i],x);
		sz[x]+=sz[ver[i]];//求以x为根的子树节点大小
		for(int j=0;j<=n;j++)f[x][j]=((a[sz[ver[i]]-1]*f[x][j])%Mod+f[ver[i]][j])%Mod;
/*
从0开始循环，以x为根的子树下边的节点随便要不要都随便，
就有a[sz[ver[i]]-1]种方案,再乘之前所求的f[x][j]然后加上自己子节点的贡献，
记得多取膜
*/
	}
}
signed main()
{
	scanf("%lld%lld",&n,&k);int x,y;a[0]=1;
	for(int i=1;i<=n;i++)a[i]=(a[i-1]<<1)%Mod;//预处理出2的n次方
	for(int i=1;i<=n;i++)scanf("%lld",&v[i]);//输入每个点的权值
	f[1][v[1]]=1;//dp到第一个节点，取v[1]个苹果肯定有一种方案
	for(int i=1;i<n;i++){scanf("%lld%lld",&x,&y);lian(x,y);lian(y,x);}//读入+连边
	dp(1,0);cout<<f[1][k];//dp+输出
	return 0;
}
/*
3 1
1 1 1
1 2
2 3

2
*/