Description
给定一大小为n的有点权树,每次询问一对点(u,v),问是否能在u到v的简单路径上取三个点权,以这三个权值为边
长构成一个三角形。同时还支持单点修改。
若t=0,则询问(a,b)
若t=1,则将点a的点权修改为b
n,q<=100000,点权范围[1,2^31-1]
Solution
考虑如果不存在会是什么情况。我们把所有边从小到大排序,无解即是
∀
i
∈
[
1
,
n
−
2
]
\forall i\in [1,n-2]
∀i∈[1,n−2],
a
i
+
a
i
+
1
≤
a
i
+
2
a_i +a_{i+1}\le a_{i+2}
ai+ai+1≤ai+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;
}