题意:
有一棵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;
}

京公网安备 11010502036488号