蓝魔法师（树形dp）

链接：https://www.nowcoder.com/acm/contest/215/C
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述

“你，你认错人了。我真的，真的不是食人魔。”--蓝魔法师

给出一棵树，求有多少种删边方案，使得删后的图每个连通块大小小于等于k，两种方案不同当且仅当存在一条边在一个方案中被删除，而在另一个方案中未被删除，答案对998244353取模

输入描述:

第一行两个整数n,k, 表示点数和限制
2 <= n <= 2000, 1 <= k <= 2000
接下来n-1行，每行包括两个整数u,v，表示u,v两点之间有一条无向边
保证初始图联通且合法

输出描述:

共一行，一个整数表示方案数对998244353取模的结果

示例1

输入

复制

输出

复制

题解：

用dp[i][j]表示以i为根结点的子树中，当前包含节点i的连通块的大小为j的方案个数。

其中dp[x][0]特殊，用来表示∑dp[x][i] (1≤i≤k）

然后每一个节点u来说，他的一部分v儿子已经搜完了，此时的dp[u][p]就是加上了搜到过的v的，联通块大小为p的个数。然后继续搜的时候就继续用dp【u】【i】*dp【v】【k-i】加到原来的dp【u】【p】上。这样是不重复的。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn=10000;
const int mod=998244353;
LL dp[maxn][maxn];
vector<int>G[maxn];
int temp[maxn];
int siz[maxn];
int n,k;
void dfs(int u,int fa){dp[u][1]=1;siz[u]=1;int v;for(int i=0;i<G[u].size();i++){v=G[u][i];if(v==fa)continue;dfs(v,u);memset(temp,0,sizeof(temp));for(int ii=1;ii<=siz[u];ii++){for(int j=0;j<=min(k-ii,siz[v]);j++){temp[ii+j]=(temp[ii+j]%mod+dp[u][ii]*dp[v][j]%mod)%mod;}}for(int ii=1;ii<=k;ii++){dp[u][ii]=temp[ii];}siz[u]+=siz[v];}for(int i=1;i<=k;i++)///到最末端节点{///    cout<<"*******dp["<<u<<"]["<<0<<"]:"<<dp[u][0]<<endl;dp[u][0] = (dp[u][0]+dp[u][i])%mod;}
}
int main()
{int a,b;while(scanf("%d%d",&n,&k)!=EOF){for(int i=1;i<=n-1;i++){scanf("%d%d",&a,&b);G[a].push_back(b);G[b].push_back(a);}dfs(1,0);cout<<dp[1][0]<<endl;}return 0;
}

蓝魔法师（树形dp）相关推荐

蓝魔法师——树形DP
链接:https://ac.nowcoder.com/acm/problem/20811 来源:牛客网时间限制:C/C++ 1秒,其他语言2秒空间限制:C/C++ 262144K,其他语言5242 ...
Wannafly挑战赛27 C 蓝魔法师(树上dp)
链接:https://ac.nowcoder.com/acm/contest/215/C 来源:牛客网时间限制:C/C++ 1秒,其他语言2秒空间限制:C/C++ 262144K,其他语言5242 ...
Wannafly挑战赛27: C. 蓝魔法师（树形DP）
链接:https://www.nowcoder.com/acm/contest/215/C 来源:牛客网题目描述 "你,你认错人了.我真的,真的不是食人魔."--蓝魔法师给出一 ...
刷题记录:牛客NC20811蓝魔法师
传送门:牛客题目描述: "你,你认错人了.我真的,真的不是食人魔."--蓝魔法师给出一棵树,求有多少种删边方案,使得删后的图每个连通块大小小于等于k,两种方案不同当且仅当存在 ...
牛客网【每日一题】8月5日题目精讲—蓝魔法师
来源:牛客网: 文章目录题目描述题解: 代码: 题目描述 "你,你认错人了.我真的,真的不是食人魔."–蓝魔法师给出一棵树,求有多少种删边方案,使得删后的图每个连通块大小小于 ...
牛客网 Wannafly挑战赛27 蓝魔法师
蓝魔法师链接: https://www.nowcoder.com/acm/contest/215/C 来源:牛客网题目描述 "你,你认错人了.我真的,真的不是食人魔."--蓝魔 ...
牛客wannafly27 C 树形dp
Description "你,你认错人了.我真的,真的不是食人魔."–蓝魔法师给出一棵树,求有多少种删边方案,使得删后的图每个连通块大小小于等于k,两种方案不同当且仅当存在一条边 ...
BNUOJ 52305 Around the World 树形dp
题目链接: https://www.bnuoj.com/v3/problem_show.php?pid=52305 Around the World Time Limit: 20000msMemory ...
[树形dp] Jzoj P5233 概率博弈
Description 小A和小B在玩游戏.这个游戏是这样的: 有一棵n个点的以1为根的有根树,叶子有权值.假设有m个叶子,那么树上每个叶子的权值序列就是一个1->m 的排列. 一开始在1号点有 ...

蓝魔法师（树形dp）

题目描述

输入描述:

输出描述:

输入

输出

蓝魔法师（树形dp）相关推荐

最新文章

热门文章