CF 1029E Tree with Small Distances 树形DP or 贪心

题意：

给你一颗树，问你要使1号节点到所有节点的距离不超过2，最少添加多少边？

题解：

显然添加的边肯定连1好。

利用树形DP。

我们设dp[i][j]为i号节点到1的距离为j，并且i号节点的子树都满足要求的最少边数。

这样dp[n][2]即可，

dp[i][1] = (i == 1 || par == 1 ? 0 : 1)

但是dp[i][2]有两种情况，一种是i节点经过父亲再到1的距离为2， 一种是经过子节点再到1的距离为2。

这两种情况的DP方程不同，我们把dp[i][2]拆成两种状态，dp[i][2]表示经过父亲再到1,dp[i][3]表示经过子节点再到1。

dp[i][2]  = min(dp[v][1], dp[v][3])

dp[i][3] = dp[i][2] + max( min(dp[i][1] - dp[i][3]), 0)

初始状态，对于叶子节点i：

dp[i][1] = dp[i][1] = (i == 1 || par == 1 ? 0 : 1)

dp[i][2] = 0. 

dp[i][3] = INF

代码：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>
#include <bitset>
#include <map>
#include <vector>
#include <stack>
#include <set>
#include <cmath>
#ifdef LOCAL
#define debug(x) cout<<#x<<" = "<<(x)<<endl;
#else
#define debug(x) 1;
#endif

#define chmax(x,y) x=max(x,y)
#define chmin(x,y) x=min(x,y)
#define lson id<<1,l,mid
#define rson id<<1|1,mid+1,r
#define lowbit(x) x&-x
#define mp make_pair
#define pb push_back
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, int> pii;

const ll MOD = 1e9 + 7;
const double eps = 1e-10;
const int INF = 0x3f3f3f3f;
const ll INFLL = 0x3f3f3f3f3f3f3f3fll;
const int MAXN = 2e5 + 5;

int d[MAXN][4];
vector<int> G[MAXN];


void dfs (int now, int par) {
    d[now][1] = (now == 1 || par == 1 ? 0 : 1);
    int minn = INF;
    for(int i : G[now]) {
        if(i == par) continue;
        dfs(i, now);
        d[now][1] += min(d[i][1], d[i][2]);
        d[now][2] += min(d[i][1], d[i][3]);
        minn = min(d[i][1] - d[i][3], minn);
    }
    d[now][3] = d[now][2] + max(minn, 0);
}


int main() {
#ifdef LOCAL
    freopen ("input.txt", "r", stdin);
#endif
    int n;
    cin >> n;
    for (int i = 2; i <= n; i++) {
        int x, y;
        scanf ("%d %d", &x, &y);
        G[x].pb (y);
        G[y].pb (x);
    }
    dfs(1, 0);
    int ans = 0;
    for(int i : G[1]) ans += d[i][1];
    printf("%d\n", ans);
    return 0;
}