[树形dp]旅游

it2023-08-16  72

题目:旅游

分析:

一道2星的题,一开始还想成贪心找叶子,太菜了。 dp[i][0/1]表示不选/选当前结点的最大时间 还是菜,毕竟树形dp做的太少了

代码:

#include <bits/stdc++.h> using namespace std; typedef long long ll;//三年竞赛一场空,不开long long见祖宗 //typedef __int128 lll; #define print(i) cout << "debug: " << i << endl #define close() ios::sync_with_stdio(0), cin.tie(0), cout.tie(0) #define mem(a, b) memset(a, b, sizeof(a)) #define pb(a) push_back(a) #define x first #define y second typedef pair<int, int> par; const ll mod = 1e9 + 7; const int maxn = 1e6 + 10; const int inf = 0x3f3f3f3f; vector<int> g[maxn]; int dp[maxn][2]; int n, s; void dfs(int u, int fa) { dp[u][1] = 1; if(!g[u].size()) return; for(int v : g[u]) { if(v == fa) continue; dfs(v, u); dp[u][1] += dp[v][0]; dp[u][0] += max(dp[v][0], dp[v][1]); } } int main() { cin >> n >> s; for(int i = 1; i < n; i++) { int a, b; cin >> a >> b; g[a].pb(b), g[b].pb(a); } dfs(s, -1); cout << dp[s][1] << endl; }
最新回复(0)