二分+樹型 dp hdu5682 zxa and leaf

傳送門：點擊打開鏈接

題意：一棵樹n個點，其中有一些點已經有權值，現在給剩下的點安排權值，使得樹中相鄰兩點的之差絕對值的最大值最小。

思路：如果我們首先就想到了二分，那後面很好想了。。

直接二分答案，之後check中，我們隨便取1個點爲根節點，然後從下向上按拓撲序做樹型dp。設SL[u]和SR[u]表示節點u能填的數字的範圍

我們從下往上，然後只要判斷是否有交集，即有解，我們就能知道當前答案是否可以使用了。

Trick：下次再用G++交用了#prama擴棧的代碼剁手。。

#include <map>
#include <set>
#include <cmath>
#include <ctime>
#include <stack>
#include <queue>
#include <cstdio>
#include <cctype>
#include <bitset>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <functional>
#define fuck(x) cout<<"["<<x<<"]";
#define FIN freopen("input.txt","r",stdin);
#define FOUT freopen("output.txt","w+",stdout);
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;

const int MX = 1e5 + 5;
const LL INF = 0x3f3f3f3f3f3f3f3fLL;

struct Edge {
    int nxt, v;
} E[MX];
int Head[MX], erear;
void edge_init() {
    erear = 0;
    memset(Head, -1, sizeof(Head));
}
void edge_add(int u, int v) {
    E[erear].v = v;
    E[erear].nxt = Head[u];
    Head[u] = erear++;
}

int is[MX], val[MX];
LL SL[MX], SR[MX];

bool DFS(int u, int f, int x) {
    if(is[u]) SL[u]  = SR[u] = val[u];
    else SL[u] = -INF, SR[u] = INF;
    for(int i = Head[u]; ~i; i = E[i].nxt) {
        int v = E[i].v;
        if(v == f) continue;
        if(!DFS(v, u, x)) return false;
        if(SL[v] != INF) SL[u] = max(SL[u], SL[v] - x);
        if(SR[v] != INF) SR[u] = min(SR[u], SR[v] + x);
    }
    if(SL[u] > SR[u]) return false;
    return true;
}
int solve() {
    int l = 0, r = 1e9, m;
    while(l <= r) {
        m = (l + r) >> 1;
        if(DFS(1, -1, m)) r = m - 1;
        else l = m + 1;
    }
    return r + 1;
}

int main() {
    int T, n, k; //FIN;
    scanf("%d", &T);
    while(T--) {
        edge_init();
        scanf("%d%d", &n, &k);
        for(int i = 1; i <= n; i++) is[i] = 0;
        for(int i = 1; i <= n - 1; i++) {
            int u, v;
            scanf("%d%d", &u, &v);
            edge_add(u, v); edge_add(v, u);
        }
        for(int i = 1; i <= k; i++) {
            int u, w;
            scanf("%d%d", &u, &w);
            is[u] = 1; val[u] = w;
        }
        printf("%d\n", solve());
    }
    return 0;
}