[bzoj3251][乱搞]树上三角形

Description

给定一大小为n的有点权树，每次询问一对点(u,v)，问是否能在u到v的简单路径上取三个点权，以这三个权值为边
长构成一个三角形。同时还支持单点修改。

Input

第一行两个整数n、q表示树的点数和操作数 第二行n个整数表示n个点的点权
以下n-1行，每行2个整数a、b，表示a是b的父亲（以1为根的情况下） 以下q行，每行3个整数t、a、b 若t=0，则询问(a,b)
若t=1，则将点a的点权修改为b n,q<=100000，点权范围[1,2^31-1]

Output

对每个询问输出一行表示答案，“Y”表示有解，“N”表示无解。

Sample Input

5 5

1 2 3 4 5

1 2

2 3

3 4

1 5

0 1 3

0 4 5

1 1 4

0 2 5

0 2 3

Sample Output

N

Y

Y

N

题解

先考虑给你n个值让你判三角形
把他们排序
从前往后枚举三个值判就可以了…
拓展到树上
我们发现
一个不合法的序列长度不会超过斐波那契数列的一个值
因为每次都会扩大a[i]+a[i-1]
然后…
如果两点距离大于45 肯定合法
斐波那契数列在45左右就超了int了
否则你就暴力判.
没了啊

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
#include<ctime>
#define LL long long
#define mp(x,y) make_pair(x,y)
using namespace std;
inline int read()
{
	int f=1,x=0;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
inline void write(int x)
{
	if(x<0)putchar('-'),x=-x;
	if(x>9)write(x/10);
	putchar(x%10+'0');
}
inline void print(int x){write(x);printf(" ");}
struct node{int x,y,next;}a[210000];int len,last[110000];
void ins(int x,int y){len++;a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;}
int fa[110000][25],dep[110000],bin[25];
void pre_tree_node(int x)
{
	for(int i=1;bin[i]<=dep[x];i++)fa[x][i]=fa[fa[x][i-1]][i-1];
	for(int k=last[x];k;k=a[k].next)
	{
		int y=a[k].y;
		if(y!=fa[x][0])
		{
			fa[y][0]=x;dep[y]=dep[x]+1;
			pre_tree_node(y);
		}
	}
}
int lca(int x,int y)
{
	if(dep[x]<dep[y])swap(x,y);
	for(int i=20;i>=0;i--)if(bin[i]<=dep[x]&&dep[fa[x][i]]>=dep[y])x=fa[x][i];
	if(x==y)return x;
	for(int i=20;i>=0;i--)if(bin[i]<=dep[x]&&fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
int n,m,cal[110000];
int sta[110000],tp;
int main()
{
	bin[0]=1;for(int i=1;i<=20;i++)bin[i]=bin[i-1]<<1;
	n=read();m=read();
	for(int i=1;i<=n;i++)cal[i]=read();
	for(int i=1;i<n;i++)
	{
		int x=read(),y=read();
		ins(x,y);
	}
	pre_tree_node(1);
	while(m--)
	{
		int op=read(),x=read(),y=read();
		if(!op)
		{
			int LA=lca(x,y);
			if(dep[x]+dep[y]-dep[LA]-dep[fa[LA][0]]>45)puts("Y");
			else
			{
				tp=0;
				while(x!=LA)sta[++tp]=cal[x],x=fa[x][0];
				while(y!=LA)sta[++tp]=cal[y],y=fa[y][0];
				sta[++tp]=cal[LA];
				sort(sta+1,sta+1+tp);
				bool bk=false;
				for(int i=3;i<=tp;i++)
					if((LL)sta[i]<(LL)sta[i-1]+sta[i-2]){bk=true;break;}
				if(!bk)puts("N");
				else puts("Y");
			}
		}
		else cal[x]=y;
	}
	return 0;
}