BZOJ.4151.[AMPPZ2014]The Cave(結論)

BZOJ

不是很懂他們爲何都要DFS三次。因而穩拿Rank1 qwq。
（三道題兩個Rank1一個Rank3效率是否是有點高qwq？）

---

記以\(1\)爲根DFS時每一個點的深度是\(dep_i\)。對於一條限制\((a,b,D)\)，知足條件的點\(x\)知足：\(dep_a+dep_b-2dep_x\leq D\Rightarrow dep_x\geq\frac{dep_u+dep_v-D}{2}\)。
那麼\(dep_x\)最小是\(\max\{0,\ \lceil\frac{dep_u+dep_v-D}{2}\rceil\}\)（先不考慮負數的狀況）。
記\(p\)爲全部\(m\)條限制中，\(\max\{0,\ \lceil\frac{dep_u+dep_v-D}{2}\rceil\}\)最大的那個點\(x\)。
有一個結論是：若\(p\)不能知足全部限制，則必定無解。不然\(p\)爲一個合法的解。
\(p\)就是要求最嚴格的限制所對應的那個點。對於和\(p\)不在同一棵子樹裏的限制，顯然該限制要知足\(p\)；在同一棵子樹內的，由於\(p\)是最嚴格的，因此只要知足\(p\)就能夠啦。

---

//13032kb   76ms
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define gc() getchar()
#define MAXIN 500000
//#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
typedef long long LL;
const int N=3e5+5;

int Enum,H[N],nxt[N<<1],to[N<<1],fa[N],dep[N],A[N],B[N],D[N];
char IN[MAXIN],*SS=IN,*TT=IN;

inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-48,c=gc());
    return now;
}
inline void AE(int u,int v)
{
    to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum;
    to[++Enum]=u, nxt[Enum]=H[v], H[v]=Enum;
}
void DFS(int x)
{
    for(int i=H[x],v; i; i=nxt[i])
        if((v=to[i])!=fa[x]) fa[v]=x, dep[v]=dep[x]+1, DFS(v);
}

int main()
{
    for(int T=read(); T--; )
    {
        const int n=read(),m=read();
        Enum=0, memset(H,0,n+1<<2);
        for(int i=1; i<n; ++i) AE(read(),read());
        dep[1]=0, fa[1]=0, DFS(1);
        int mx=0,p=0;
        for(int i=1; i<=m; ++i)
        {
            int t=dep[A[i]=read()]+dep[B[i]=read()]-(D[i]=read());
            if(t>mx) mx=t, p=i;
        }
        if(!mx) {puts("TAK 1"); continue;}
        int x=A[p];
        for(int t=dep[x]-(mx+1)/2; t>0; --t,x=fa[x]);
        dep[x]=0, fa[x]=0, DFS(x);
        bool fg=1;
        for(int i=1; i<=m; ++i)
            if(dep[A[i]]+dep[B[i]]>D[i]) {fg=0; break;}
        fg?printf("TAK %d\n",x):puts("NIE");
    }
    return 0;
}