bzoj 3251: 树上三角形 暴力

→题目链接←

【想说的话】

啊啊啊啊，搬完博客的第一篇博客

md新博客调了一天，头都大了

【题解】

想一想，可以发现如果答案为no，在int范围内最多只可能同时存在46个数

所以找到lca，判断下路径上的点的个数，如果大于46个就输出Yes，否则暴力判断

【代码】

#include<bits/stdc++.h>#define MAXN 100010typedef long long ll;using namespace std;inline int rd(){int x=0,y=1;char c=getchar();while(c<'0' || c>'9'){if(c=='-')y=-y;c=getchar();}while(c>='0'&&c<='9')x=x*10+c-'0',c=getchar();return x*y;}int n,m;int val[MAXN],fa[MAXN][22],deep[MAXN],a[MAXN],tot=0;vector<int>v[MAXN];void dfs(int x){deep[x]=deep[fa[x][0]]+1;for(int i=1; i<=20; i++)fa[x][i]=fa[fa[x][i-1]][i-1];for(int i=0; i<v[x].size(); i++){int to=v[x][i];if(to==fa[x][0])continue;dfs(to);}}int lca(int x,int y){    if(deep[x]>deep[y])swap(x,y);    int num=deep[y]-deep[x];    for(int i=20; i>=0; i--){        if(num>=(1<<i)){            y=fa[y][i];            num-=(1<<i);        }    }    if(x!=y){        for(int i=20; i>=0; i--){            if(fa[x][i]!=fa[y][i]){                x=fa[x][i];                y=fa[y][i];            }        }        x=fa[x][0];    }    return x;}void solve(){sort(a+1,a+1+tot);for(int i=3; i<=tot; i++){if((ll)a[i]<(ll)a[i-1]+a[i-2]){puts("Y");return;}}puts("N");}int main(){n=rd(),m=rd();for(int i=1; i<=n; i++)val[i]=rd();for(int i=1; i<n; i++){int x=rd(),y=rd();fa[y][0]=x;v[x].push_back(y);v[y].push_back(x);}dfs(1);while(m--){int x=rd(),y=rd(),z=rd();if(x==1)val[y]=z;else{int xx=lca(y,z);if(deep[y]+deep[z]-deep[xx]*2+1>46)puts("Y");else{tot=0;while(y!=xx){a[++tot]=val[y];y=fa[y][0];}while(z!=xx){a[++tot]=val[z];z=fa[z][0];}a[++tot]=val[xx];solve();}}}return 0;}