HDU-5682 zxa and leaf(二分+dfs)

题目: 传送门
思路:
        要求最大值最小,不难想到可以二分答案后检验.
        我们可以维护每个节点的取值范围,如果没有初始值就是[1,1e9],如果有初始值k就是[k,k].然后dfs维护每个节点的取值,如果某个节点的取值范围[l,r]出现l>r的情况则说明不符合条件.

AC_Code

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <vector>
using namespace std;

const int inf = 1e9;
const int maxn = 5e4+7;

struct node
{
    int L,R,l,r;
}s[maxn];

int n,m;
vector<int> v[maxn];

void init(){
    for(int i=0;i<=n;i++){
        s[i].L=1;
        s[i].R=inf;
        v[i].clear();
    }
}

bool dfs(int now,int fath,int len) { // 先dfs到最低端划定每个点的初始范围
    s[now].l = s[now].L;
    s[now].r = s[now].R;
    int up = v[now].size();
    for(int i=0;i<up;i++) {
        int nex = v[now][i];
        if( nex == fath) continue;
        if(!dfs(nex,now,len)) return false;
        s[now].l = max(s[now].l,max(1,s[nex].l-len)); //看是否满足区间
        s[now].r = min(s[now].r,min(s[nex].r+len,inf));
        if(s[now].l>s[now].r) return false;
    }
    return s[now].l<=s[now].r;
}

bool is_ok(int x) {
    return dfs(1,0,x);
}

int main() {
    int t;
    scanf("%d",&t);
    while(t--) {
        scanf("%d%d",&n,&m);
        init();
        for(int i=0;i<n-1;i++) {
            int l,r;
            scanf("%d%d",&l,&r);
            v[l].push_back(r);
            v[r].push_back(l);
        }
        for(int i =0;i<m;i++) {
            int node,k;
            scanf("%d%d",&node,&k);
            s[node].L= s[node].R = k;
        }
        int l=0,r=1e9,ans;
        while(l<=r) {
            int mid = (l+r) >>1;
            if(is_ok(mid)) r = (ans=mid)-1;
            else l=mid+1;
        }
        printf("%d\n",ans);
    }
    return 0;
}