BZOJ3251 : 树上三角形

BZOJ AC1000题纪念~~~

 

将x到y路径上的点权从小到大排序

如果不存在b[i]使得b[i]+b[i+1]>b[i+2]则无解

此时b数列增长速度快于斐波那契数列，当达到50项时就会超过int的范围

所以暴力查询，如果路径长度达到50就肯定有解，否则暴力排序检验

 

#include<cstdio>

#include<algorithm>

#define N 100010

using namespace std;

int n,q,i,x,y,op,a[N],d[N],f[N],g[N],nxt[N],v[N],ed,b[N],t;

inline void read(int&a){char ch;while(!(((ch=getchar())>='0')&&(ch<='9')));a=ch-'0';while(((ch=getchar())>='0')&&(ch<='9'))(a*=10)+=ch-'0';}

inline void add(int x,int y){f[v[++ed]=y]=x;nxt[ed]=g[x];g[x]=ed;}

void dfs(int x){d[x]=d[f[x]]+1;for(int i=g[x];i;i=nxt[i])dfs(v[i]);}

inline bool ask(int x,int y){

  for(t=0;x!=y;){

    if(d[x]<d[y])swap(x,y);

    b[++t]=a[x],x=f[x];

    if(t>48)return 1;

  }

  for(b[++t]=a[x],sort(b+1,b+t+1),i=2;i<t;i++)if((long long)b[i-1]+b[i]>b[i+1])return 1;

  return 0;

}

int main(){

  read(n),read(q);

  for(i=1;i<=n;i++)read(a[i]);

  for(i=1;i<n;i++)read(x),read(y),add(x,y);

  dfs(1);

  while(q--){

    read(op),read(x);

    if(op)read(a[x]);else read(y),puts(ask(x,y)?"Y":"N");

  }

  return 0;

}