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;
}