【AMPPZ2014】【BZOJ4151】The Cave

Description

给定一棵有n个节点的树，相邻两点之间的距离为1。
请找到一个点x，使其满足所有m条限制，其中第i条限制为dist(x,a[i])+dist(x,b[i])<=d[i]。
Input

第一行包含一个正整数t(1<=t<=1000)，表示数据组数。
对于每组数据，第一行包含两个正整数n,m(2<=n,m<=300000)，表示点数、限制数。
接下来n-1行，每行两个正整数x,y(1<=x,y<=n)，表示树上的一条边。
接下来m行，每行三个正整数a[i],b[i],di，描述一条限制。
输入数据保证所有n之和不超过300000，所有m之和也不超过300000。
Output

输出t行。第i行输出第i组数据的答案，如果无解输出NIE，否则输出TAK，
然后输出x，如有多组解，输出任意一组。
Sample Input

2

5 3

1 2

2 3

2 4

3 5

1 4 2

5 5 5

3 2 1

3 2

1 2

2 3

1 1 2

3 3 1
Sample Output

TAK 2

NIE
HINT

Source

鸣谢Claris上传

这题我真不会做了..
Claris的AMPPZ2014

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 300010
#define GET (ch>='0'&&ch<='9')
using namespace std;
int T,n,m,top;
int a[MAXN],b[MAXN],d[MAXN],dis[3][MAXN];
struct edge
{
    int to;
    edge *next;
}e[MAXN<<1],*prev[MAXN];
void insert(int u,int v)    {e[++top].to=v;e[top].next=prev[u];prev[u]=&e[top];}
void in(int &x)
{
    char ch=getchar();x=0;
    while (!GET)    ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();
}
void dfs(int x,int f,int d,int now)
{
    dis[now][x]=d++;
    for (edge *i=prev[x];i;i=i->next)   if (i->to!=f)   dfs(i->to,x,d,now);
}
int main()
{
    in(T);int u,v,maxn;
    while (T--)
    {
        top=0;for (int i=1;i<=n;i++)    prev[i]=NULL;
        in(n);in(m);
        for (int i=1;i<n;i++)   in(u),in(v),insert(u,v),insert(v,u);
        for (int i=1;i<=m;i++)  in(a[i]),in(b[i]),in(d[i]);
        dfs(1,0,0,0);u=0;maxn=0;
        for (int i=1;i<=m;i++)
        {
            v=dis[0][a[i]]+dis[0][b[i]]-d[i];
            if (u==0||v>maxn)   u=i,maxn=v;
        }
        dfs(a[u],0,0,1);dfs(b[u],0,0,2);v=0;
        for (int i=1;i<=n;i++)
            if (dis[1][i]+dis[2][i]<=d[u]&&(!v||dis[0][i]<maxn))    v=i,maxn=dis[0][i];
        if (!v) {puts("NIE");continue;}
        dfs(v,0,0,0);
        for (int i=1;i<=m;i++)
            if (dis[0][a[i]]+dis[0][b[i]]>d[i]) {v=0;break;}
        if (v)  printf("TAK %d\n",v);else   puts("NIE");
    }
}