旅游

旅游

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

题意:
有一棵n个节点的树,第一天你在s点住宿,每一天你可以浏览与它距离不超过1的节点,你不能住宿在你已经浏览过的节点,求你最多住宿几天?

思路:
树状dp,以s节点为根。
dp[i][1/0]表示在i为根的子树中是否在i节点住宿的最大住宿次数。
如果u是父节点,v是子节点。
dp[u][1]=图片说明 dp[v][0]+1;(v为u的子节点)
dp[u][0]=图片说明 max(dp[v][1],dp[v][0]);(v为u的子节点)
答案为dp[s][1].

代码:

#include<bits/stdc++.h>
#define ll long long
#define inf 1024523

using namespace std;

inline int read()
{
    int x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')
            f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=(x<<1)+(x<<3)+(ch^48);
        ch=getchar();
    }
    return x*f;
}

vector<int> g[500005];
int dp[500005][2];

void dfs(int v,int f)
{
    for(int i=0;i<g[v].size();i++)
    {
        if(g[v][i]==f)
        {
            continue;
        }
        dfs(g[v][i],v);
        dp[v][1]+=dp[g[v][i]][0];
        dp[v][0]+=max(dp[g[v][i]][1],dp[g[v][i]][0]);
    }
    dp[v][1]++;
}

int main()
{
    int n, s;
    scanf("%d%d",&n,&s);
    for(int i=0;i<n-1;i++)
    {
        int u, v;
        scanf("%d%d",&u,&v);
        g[u].push_back(v);
        g[v].push_back(u);
    }
    dfs(s,-1);
    cout << dp[s][1] << endl;
    return 0;
}
全部评论

相关推荐

评论
1
收藏
分享
牛客网
牛客企业服务