D. Sasha and a Walk in the City

原题链接

题意

树中任意一条路径上黑色点的数量不超过两个,请问存在多少种树

分析

先随便找一个节点作为根节点,然后分类讨论

假如根到叶子节点的路径上有两个黑色节点,则不能再添加其他点了

如果根到叶子节点的路径上有一个黑色节点,则可以还可以在不在这条路径上的地方放黑色节点

在弄清楚规则后,就可以用dp了(真是万能啊,完全没有要用dp的intuition)

其实dp只能算一种主动运用的方法,当后者于前者有继承关系时,就可以用了,状态的设计也是很巧妙

code

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll mod=998244353;

int dp[300005][3]={0};
vector<int> G[300005];

void dfs(int now,int fa)
{
    dp[now][0]=1;
    dp[now][2]=0;

    int tem=1;
    for(auto next:G[now])
    {
        if(next==fa) continue;

        dfs(next,now);
        tem*=(1+dp[next][1]);
        dp[now][2]+=dp[next][2]+dp[next][1];
    }
    dp[now][1]=tem;
}

void solve()
{
    int n;
    cin>>n;

    for(int i=1;i<n;i++)
    {
        int x,y;

        cin>>x>>y;

        G[x].push_back(y);
        G[y].push_back(x);
    }

    dfs(1,1);

    cout<<dp[1][1]+dp[1][2]+1<<'\n';

    for(int i=1;i<=n;i++) G[i].clear();
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int t=1;
    cin>>t;
    while(t--) solve();
    return 0;
}

posted @   纯粹的  阅读(5)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· Manus的开源复刻OpenManus初探
· AI 智能体引爆开源社区「GitHub 热点速览」
· 从HTTP原因短语缺失研究HTTP/2和HTTP/3的设计差异
· 三行代码完成国际化适配,妙~啊~
点击右上角即可分享
微信分享提示