bzoj3251 树上三角形 乱搞

Description

---

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

若t=0，则询问(a,b)
若t=1，则将点a的点权修改为b
n,q<=100000，点权范围[1,2^31-1]

Solution

---

考虑如果不存在会是什么情况。我们把所有边从小到大排序，无解即是$\forall i\in [1,n-2]$，$a_i+a_{i+1}\le a_{i+2}$
取相等的情况其实就是斐波那契数列。由于点权范围是int的，因此超过50个数字就一定有解了，否则暴力即可

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (register int i=st;i<=ed;++i)

typedef long long LL;
const int N=100005;

struct edge {int y,next;} e[N];

int ls[N],fa[N],dep[N],edCnt;

LL a[N],v[N];

int read() {
	int x=0,v=1; char ch=getchar();
	for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
	for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
	return x*v;
}

void add_edge(int x,int y) {
	e[++edCnt]=(edge) {y,ls[x]}; ls[x]=edCnt;
}

void dfs(int now) {
	for (int i=ls[now];i;i=e[i].next) {
		dep[e[i].y]=dep[now]+1;
		dfs(e[i].y);
	}
}

int main(void) {
	freopen("data.in","r",stdin);
	freopen("myp.out","w",stdout);
	int n=read(),q=read();
	rep(i,1,n) a[i]=read();
	rep(i,2,n) {
		int x=read(),y=read();
		fa[y]=x; add_edge(x,y);
	}
	dfs(dep[1]=1);
	for (int opt,x,y;q--;) {
		opt=read(),x=read(),y=read();
		if (!opt) {
			bool flag=false; v[0]=0;
			while (x!=y&&v[0]<50) {
				if (dep[x]>dep[y]) v[++v[0]]=a[x],x=fa[x];
				else v[++v[0]]=a[y],y=fa[y];
			}
			v[++v[0]]=a[x];
			if (v[0]>=50) flag=true;
			if (!flag) {
				std:: sort(v+1,v+v[0]+1);
				rep(i,1,v[0]-2) {
					if (v[i]+v[i+1]>v[i+2]) {
						flag=true; break;
					}
				}
			}
			flag?puts("Y"):puts("N");
		} else a[x]=y;
	}
	return 0;
}