Wannafly挑战赛27 C 蓝魔法师(树上dp)

2 篇文章 0 订阅

链接:https://ac.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

输入

复制

5 2
1 2
1 3
2 4
2 5

输出

复制

7
#include<bits/stdc++.h>
using namespace std;
#define Sheryang main
const int maxn=1e6+7;
typedef long long ll;
const int mod=998244353;
vector<int>G[2005];
ll dp[2005][2005],tmp[2005];
int sz[2005],n,k;

void dfs(int u,int fa)
{
    sz[u]=dp[u][1]=1;
    for(int i=0;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(v==fa) continue;
        dfs(v,u);

        memset(tmp,0,sizeof(tmp));
        for(int ii=1;ii<=sz[u];ii++) //此时的sz[u]是已被遍历的u的子节点数量
            for(int j=0;j<=sz[v] && ii+j<=k;j++)
                tmp[ii+j]=(tmp[ii+j]+dp[u][ii]*dp[v][j]%mod)%mod;
                                    // 此时dp[u][ii]是左侧已被遍历的子树的种类数 不包括此时的子节点v 因此不会重复
        for(int ii=1;ii<=k;ii++)
            dp[u][ii]=tmp[ii];
        sz[u]+=sz[v];
    }

    for(int i=1;i<=k;i++)
        dp[u][0]=(dp[u][0]+dp[u][i])%mod; //dp[u][0]存储所有大小联通块的方案数和
}

int Sheryang()
{
    scanf("%d%d",&n,&k);

    for(int i=1;i<n;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        G[u].push_back(v);
        G[v].push_back(u);
    }

    dfs(1,0);

    printf("%lld\n",dp[1][0]);
    return 0;
}

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值