Hrbust 2064 萌萌哒十五酱的宠物~(树链剖分+线段树）

最新推荐文章于 2019-08-16 13:24:44 发布

吃货妍

最新推荐文章于 2019-08-16 13:24:44 发布

阅读量472

点赞数

分类专栏：补题

本文链接：https://blog.csdn.net/qq_33183401/article/details/77996504

版权

补题专栏收录该内容

76 篇文章 0 订阅

订阅专栏

萌萌哒十五酱的宠物~

Time Limit: 2000 MS Memory Limit: 32768 K

Total Submit: 48(16 users) Total Accepted: 22(14 users) Rating: Special Judge: No

Description

十五酱养了一只很可爱的宠物。。。。。。 

一只虫子。。。。TAT 

这只虫子特别喜欢爬树。。。。。它在这棵树上生活了很久，对它的构造了如指掌。所以它在树上从来都是走最短路，不会绕路。它还还特别喜欢三角形，所以当它在树上爬来爬去的时候总会在想，如果把刚才爬过的那几根树枝/树干锯下来，能不能从中选三根出来拼成一个三角形呢？

Input

多组数据 

每组数据第一行包含一个整数 N（N<=100000），表示树上节点的个数（从 1 到 N 标号）。 
接下来的 N-1 行包含三个整数 a, b, len（1<=a<=N,1<=b<=N,1<=len<=10^9），表示有一根长度为 len 的树枝/树干在节点 a 和节点 b 之间。
接下来一行包含一个整数 M（M<=100000），表示询问数。
接下来M行每行两个整数 S, T，表示毛毛虫从 S 爬行到了 T，询问这段路程中的树枝/树干是否能拼成三角形。

Output
多组数据
对于每个询问输出一行，包含”Yes”或“No”，表示是否可以拼成三角形。
Sample Input
5
1 2 5
1 3 20
2 4 30
4 5 15
2
3 4
3 5

Sample Output
No
Yes
题解：要想数列不能组成三角形，则应满足斐波那契数列，ai<1e9.如果边数>50,则一定能组成三角形。不行就暴力判断。
代码：

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<vector>
#include<queue>
#include<stack>
#include<set>
#include<algorithm>
#include<map>
#include<time.h>
#include<math.h>
//#define pb push_back
//#define mp make_pair
#define INF 0x3f3f3f3f
using namespace std;
typedef long long int ll;
typedef pair<int,int>pp;
const int N=1e3+100;
const int mod=1e9+7;
int read()
{
    int x=0;
    char ch = getchar();
    while('0'>ch||ch>'9')ch=getchar();
    while('0'<=ch&&ch<='9')
    {
        x=(x<<3)+(x<<1)+ch-'0';
        ch=getchar();
    }
    return x;
}
/***********************************************************/
struct node
{
    int to;
    int w;
    int next;
}e[250000];
int val[150000];
int head[150000];
int n,cont;
void add(int from,int to,int w)
{
    e[cont].to=to;
    e[cont].w=w;
    e[cont].next=head[from];
    head[from]=cont++;
}
#define lson l,m,rt*2
#define rson m+1,r,rt*2+1
int tree[155000*10];
void pushup(int rt)
{
    tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}
void update(int L,int R,int c,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        tree[rt]=c;
        return ;
    }
    int m=(l+r)/2;
    if(L<=m)
    {
        update(L,R,c,lson);
    }
    if(m<R)
    {
        update(L,R,c,rson);
    }
    pushup(rt);
}
int Query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        return tree[rt];
    }
    else
    {
        int m=(l+r)>>1;
        int ans=0;
        if(L<=m)
        {
            ans+=Query(L,R,lson);
        }
        if(m<R)
        {
            ans+=Query(L,R,rson);
        }
        pushup(rt);
        return ans;
    }
}
void build( int l ,int r , int rt )
{
    if( l == r )
    {
        tree[rt]=0;
        return ;
    }
    else
    {
        int m = (l+r)>>1 ;
        build(lson) ;
        build(rson) ;
        pushup(rt) ;
        return ;
    }
}
int cnt;
int depth[150000];
int son[150000];
int fa[150000];
int size[150000];
int dfn[150000];
int Top[150000];
void Dfs(int u,int from,int d)
{
    size[u]=1;fa[u]=from;son[u]=-1;depth[u]=d;
    for(int i=head[u];i!=-1;i=e[i].next)
    {
        int v=e[i].to;
        if(v==from)continue;
        Dfs(v,u,d+1);
        size[u]+=size[v];
        if(son[u]==-1||size[v]>size[son[u]])
        {
            son[u]=v;
        }
    }
}
void Dfs2(int u,int top)
{
    Top[u]=top;dfn[u]=++cnt;
    if(son[u]!=-1)Dfs2(son[u],top);
    for(int i=head[u];i!=-1;i=e[i].next)
    {
        int v=e[i].to;
        if(v==fa[u]||v==son[u])continue;
        Dfs2(v,v);
    }
}
void Slove2(int x,int y)
{
    int cnt=0;
    int path[500];
    while(x!=y)
    {
        if(depth[x]<depth[y])swap(x,y);
        path[cnt++]=val[x];
        x=fa[x];
    }
    int flag=0;
    sort(path,path+cnt);
    for(int i=2;i<cnt;i++)
    {
        if(path[i-2]+path[i-1]>path[i])flag=1;
    }
    if(flag==1)printf("Yes\n");
    else printf("No\n");
}
void Slove(int x,int y)
{
    int prex=x,prey=y;
    int sum=0;
    int fx=Top[x],fy=Top[y];
    while(fx!=fy)
    {
        if(depth[fx]<depth[fy])
        {
            swap(fx,fy);swap(x,y);
        }
        sum+=Query(dfn[fx],dfn[x],1,n,1);
        x=fa[fx];fx=Top[x];
    }
    if(depth[x]<depth[y])swap(x,y);
    if(x!=y)
    sum+=Query(dfn[son[y]],dfn[x],1,n,1);
    if(sum>=100)printf("Yes\n");
    else Slove2(prex,prey);
}
int xx[150000],yy[150000],ww[150000];
int main()
{
    while(~scanf("%d",&n))
    {
        cnt=0;
        cont=0;
        memset(head,-1,sizeof(head));
        for(int i=1;i<=n-1;i++)
        {
            int x,y,w;scanf("%d%d%d",&x,&y,&w);
            xx[i]=x;yy[i]=y;ww[i]=w;
            add(x,y,w);
            add(y,x,w);
        }
        Dfs(1,-1,1);
        Dfs2(1,1);
        build(1,n,1);
            for(int i=1;i<=n-1;i++)
        {
            if(depth[xx[i]]<depth[yy[i]])swap(xx[i],yy[i]);
            val[xx[i]]=ww[i];
        }
        for(int i=2;i<=n;i++)update(dfn[i],dfn[i],1,1,n,1);
        int q;
        scanf("%d",&q);
        while(q--)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            Slove(x,y);
        }
    }
    return 0;
}