【FJWC2016】树上三角形 / LCA

题

给定一棵 $n$ 个点的树，每个点有个 $[1,2^{31}-1]$ 的点权，有 $Q$ 个操作。
操作类型 $0$ ：修改某个点的点权。
操作类型 $1$ ：给定 $a,b$ ，判断从 $a,b$ 之间的唯一路径上的所有点（路径上的点当然也包括 $a$ 和 $b$ 啦）中能否选出三个点，满足能以这三个点的点权为边长构成三角形。
$n，Q\le100000$。

解

暴力：把路径上所有点存下来，按权值排序，for一遍判断。复杂度平方。
正解：看上去要什么数据结构？？naive！
看到“三角形”这种东西就应该到想到斐波那契数列（有微妙的关系）
我们考虑暴力存下来排序完后的数列。设这个数列叫 $a$，如果是无解，必然有
$a_1 + a_2 \le a_3 ，a_2+a_3\le a_4...$
我们发现要满足这样条件的 a a <script type="math/tex" id="MathJax-Element-830">a</script> 数列增长非常快…因为即使我们全部取等号，它就是个斐波那契数列，仍然增长得吓人，更不用说是小等于了。
增长得有多快呢？我们经过程序打表，很容易发现斐波那契数列在五十多项就超出了题目中的范围，所以我们算一下路径上的点数，超过这个数就直接输出NO。否则暴力判断，复杂度是线性乘上一个五十多的常数。
代码：

#include <bits/stdc++.h>

using namespace std;

#define Maxn 100005
#define R register
#define LL long long
#define LG 19

int first[Maxn],next[Maxn],to[Maxn],cnt;
void link(R int x,R int y) {next[++cnt]=first[x];first[x]=cnt;to[cnt]=y;}

int n,q,f[Maxn][32],dep[Maxn],tail;
void dfs(R int now){for (R int i=first[now];i;i=next[i])dep[to[i]]=dep[now]+1,dfs(to[i]);}
LL que[Maxn],w[Maxn];
inline bool check()
{
    for(R int i=3;i<=tail;++i)if(que[i-2]+que[i-1]>que[i]) return 1;
    return 0;
}
int main()
{
    scanf("%d %d",&n,&q);
    for (R int i=1;i<=n;++i) scanf("%lld",&w[i]);
    for (R int i=1,a,b;i<n;++i) scanf("%d %d",&a,&b),f[b][0]=a,link(a,b);
    dfs(1);
    for (R int i=1;i<=LG;++i)
     for (R int j=1;j<=n;++j)
      f[j][i]=f[f[j][i-1]][i-1];
    while(q--)
    {
        R int t,a,b;
        scanf("%d %d %d",&t,&a,&b);
        if(t) w[a]=(LL)b;
        else
        {
            if(dep[a] > dep[b]) swap(a,b);
            R int ta=a,tb=b,tmp=dep[b]-dep[a];
            for(R int j=0;j<LG;++j)if(tmp&(1<<j))b=f[b][j];
            for(R int j=LG-1;j>=0;--j)if(f[a][j]^f[b][j])a=f[a][j],b=f[b][j];
            (a^b)?a=f[a][0]:0;
            R int len=dep[ta]+dep[tb]-(dep[a]<<1);
            if(len >= 60) 
            {
                puts("Y");
                continue;
            }
            tail=0;
            while(ta^a)que[++tail]=w[ta],ta=f[ta][0];
            while(tb^a)que[++tail]=w[tb],tb=f[tb][0];
            que[++tail]=w[a];
            sort(que+1,que+tail+1);
            check()?puts("Y"):puts("N");
        }
    }
    return 0;
}