蓝魔法师

蓝魔法师

https://ac.nowcoder.com/acm/problem/20811

思路

思路应该算是比较简单吧...
难点在于时间复杂度的证明.
因为这题的范围都是以内的.题目是让你切成若干份,每份大小都不超过,因为是一棵树,所以很容易想到树形.对于树形,我们很容易想到利用子树就行转移.那么我们要转移什么,才能让子树来跟新父节点呢?很显然的一个东西,我可以和子树分割,也可以不和子树分割.我们定义的状态表示到了这个节点所连的联通块大小为的方案数.
那么显然的转移方程就是:

假如不切割这颗子树,.
假如切割这颗子树,那么就相当于一个独立事件,对于的每种连通块大小都有可能,然后相乘即使答案了.

下面就是时间复杂度证明,简单一看好像是的.不过其实跑不满~~.时间真的卡的很紧,但绝对不是的.
倒叙枚举,防止更新的答案作为结果更新后面的答案.

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod=998244353;
const int N=2e3+5;
ll n,k;
vector<int>g[N];
ll f[N][N],sz[N];
void dfs(int u,int fa)
{
    sz[u]=1;f[u][1]=1;
    for(int v:g[u])
    {
        if(v==fa)   continue;
        dfs(v,u);
        ll sum=0;
        for(int i=1;i<=min(sz[v],k);i++)    sum+=f[v][i],sum%=mod;
        for(int i=min(sz[u],k);i>=1;i--)//枚举父亲节点的连通块的大小.
        {
            for(int j=min(sz[v],k);j>=1;j--)//枚举子节点的连通块大小.
                if(i+j<=k) f[u][i+j]+=f[u][i]*f[v][j],f[u][i+j]%=mod;
            f[u][i]*=sum,f[u][i]%=mod;
        }
        sz[u]+=sz[v];
    }
}

int main()
{
    scanf("%lld%lld",&n,&k);
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs(1,1);
    ll ans=0;
    for(int i=1;i<=k;i++)   ans+=f[1][i],ans%=mod;
    printf("%lld\n",ans);
    return 0;
}
lpt的小屋 文章被收录于专栏

我想要一份甜甜的爱情

全部评论

相关推荐

2025-12-12 19:01
南京航空航天大学 C++
秋招没咋投,准备&nbsp;wxg&nbsp;转正之后摆烂了。结果不堪字节&nbsp;HR&nbsp;的骚扰还是面了一下字节。之前想去字节的时候怎么面都挂。现在想着随便面一下结果三面技术面都意外顺利还有加面。十月中旬字节发了意向,wxg&nbsp;转正结果无响应。十月底字节拉了保温群,wxg&nbsp;口头通过,系统显示考核中。十一月初和字节&nbsp;ld&nbsp;交流之后得知&nbsp;base&nbsp;居然能选海外,甚至能小&nbsp;wlb&nbsp;一下,wxg&nbsp;无响应无人联系。十一月中旬把字节&nbsp;base&nbsp;转到了海外,wxg&nbsp;流程灰了,一问超时忘处理了,过两天又变考核中了。十一月下旬字节换了海外&nbsp;HR&nbsp;对接,问了期望薪资,wxg&nbsp;考核终于显示通过,无&nbsp;HR&nbsp;保温,无其他保温。十一月底给字节报了个天价,想吓吓他们,同时告诉微信字节要开了,微信无响应。同样十一月底字节&nbsp;HR&nbsp;告诉我确实给不到那么高,但是能拿期权补上,问能不能接受。微信无响应。同样十一月底字节&nbsp;HR&nbsp;告知了具体方案,符合预期。&nbsp;微信无响应。十二月上旬催&nbsp;wxg&nbsp;不开我就盲拒了,wxg&nbsp;HR&nbsp;火急火燎的打电话问情况,问期望。我给了一个不算夸张的总包数字,因为今年市场在涨,过了三天还不联系我,我再催,约时间下午打电话,非得在我给出的数字上压下去几万,微信又不差这点,为什么不能满足我,让我没有拒绝的理由呢?一番纠结抗争,求稳还是追求挑战,最终选择接受迎接新的挑战,因为堂吉诃德永远不会停下脚步!回想起来,在&nbsp;wxg&nbsp;谈薪的阶段,我认为并没有给予我一定的重视,即使&nbsp;HR&nbsp;表示我在实习期间的表现和之前的面评都很靠前。也没有感觉到想要争取我,虽然我表示拒了&nbsp;offer&nbsp;之后要给我加面委定&nbsp;t6&nbsp;再涨,但我三个月没面试让我面面委那就是白给,还是算了。有缘再见了我亲爱的&nbsp;wxg,再见了曾经的梦中情厂,再见亲爱的&nbsp;mt,再见亲爱的朋友们。也再见,北京的一切。我想润了。秋招结束,卸载牛客,下一个三年,下一个五年,下一个十年后再来看看。
面试中的大熊猫爱吃薯...:我嫉妒得狗眼通红
点赞 评论 收藏
分享
评论
2
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务