树上三角形 tri
时间限制: 1 Sec 内存限制: 256 MB
题目描述
给定一大小为n的有点权树,每次询问一对点(u,v),问是否能在u到v的简单路径上取三个点权,以这三个权值为边长构成一个三角形。同时还支持单点修改
输入
第一行两个整数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
输出
对每个询问输出一行表示答案,“Y”表示有解,“N”表示无解
样例输入
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
样例输出
N
Y
Y
N
【数据范围】
对于10%的数据,n,q<=100
对于另外40%的数据,无修改操作
对于100%的数据,n,q<=100000,点权范围[1,2^31-1]
题解
三条边形成一个三角形必然被满足a+b>=c。当a+b=c,可以发现是一个斐波那契数列,鉴于点权<2^31-1,即为斐波那契第47项。所以当两点间节点数超过47,则必然满足题意。对于节点数小于47的部分包里判断即可。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#define ll long long
#define N 100010
using namespace std;
int n,Q,s[N],dep[N],t[50],fa[N];
int k,la[N],ff[N*2],q[N];
struct node{int a,b;}map[N*2];
void add(int a,int b)
{
map[++k]=(node){a,b};
ff[k]=la[a];la[a]=k;
}
void bfs()
{
int l=1,r=2;q[1]=1;dep[1]=1;
while(l<r)
{
int x=q[l];l++;
for(int a=la[x];a;a=ff[a])
q[r]=map[a].b,dep[q[r]]=dep[x]+1,r++;
}
}
int main()
{
int type,a,b;
scanf("%d%d",&n,&Q);
for(int i=1;i<=n;i++)scanf("%d",&s[i]);
for(int i=1;i<n;i++)
scanf("%d%d",&a,&b),fa[b]=a,add(a,b);
bfs();
while(Q--)
{
scanf("%d%d%d",&type,&a,&b);
if(type){s[a]=b;continue;}
int sum=0,flag=0;
while(a!=b&&sum<47)
{
if(dep[a]<dep[b])swap(a,b);
t[++sum]=s[a];a=fa[a];
}
if(sum==47){printf("Y\n");continue;}
t[++sum]=s[a];sort(t+1,t+sum+1);
for(int i=1;i<=sum-2;i++)
if((ll)t[i]+t[i+1]>(ll)t[i+2]){flag=1;break;}
if(!flag)printf("N\n");
else printf("Y\n");
}
return 0;
}