[bzoj4151][AMPPZ2014]The Cave

来自FallDream的博客，未经允许，请勿转载，谢谢。

---

给定一棵有n个节点的树，相邻两点之间的距离为1。

请找到一个点x，使其满足所有m条限制，其中第i条限制为dist(x,a[i])+dist(x,b[i])<=d[i]。

n<=300000,m<=300000

 

一个点如果满足题意，那么他和点1的距离至少为$max(0,\frac{ai+bi-di}{2})$

找出最大的那个点。如果有解，那么那个点肯定可以。暴力判断行不行就可以了。

#include<iostream>
#include<cstdio>
#define MN 300000
#define getchar() (*S++)
char B[<<],*S=B;
using namespace std;
inline int read()
{
    int x = ; char ch = getchar();
    while(ch < '' || ch > '')  ch = getchar();
    while(ch >= '' && ch <= ''){x = x *  + ch - '';ch = getchar();}
    return x;
}
int head[MN+],n,m,cnt=,dep[MN+],a[MN+],fa[MN+],b[MN+],d[MN+];
struct edge{int to,next;}e[MN*+];
inline void ins(int f,int t)
{
    e[++cnt]=(edge){t,head[f]};head[f]=cnt;
    e[++cnt]=(edge){f,head[t]};head[t]=cnt;
}

void Dfs(int x,int f)
{
    fa[x]=f;
    for(int i=head[x];i;i=e[i].next)
        if(e[i].to!=f)
            dep[e[i].to]=dep[x]+,Dfs(e[i].to,x);
}

int main()
{
    fread(B,,<<,stdin);
    for(int T=read();T;--T)
    {
        n=read();m=read();cnt=;
        for(register int i=;i<=n;++i) head[i]=;
        for(register int i=;i<n;++i) ins(read(),read());
        dep[]=;Dfs(,);int mx=,From=;
        for(register int i=;i<=m;++i)
            a[i]=read(),b[i]=read(),d[i]=read(),
            (dep[a[i]]+dep[b[i]]-d[i]+)/>mx?(mx=(dep[a[i]]+dep[b[i]]-d[i]+)/,From=i):;
        for(From=a[From];dep[From]>mx;From=fa[From]);
        dep[From]=;Dfs(From,);bool flag=;
        for(register int i=;i<=m;++i)
            if(dep[a[i]]+dep[b[i]]>d[i]) {flag=;break;}
        if(flag) printf("TAK %d\n",From);
        else puts("NIE");
    }
    return ;
}