树上三角形 tri

树上三角形 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;
}