• codeforces 675D D. Tree Construction(线段树+BTS)


    题目链接:

    D. Tree Construction

    D. Tree Construction
    time limit per test
    2 seconds
    memory limit per test
    256 megabytes
    input
    standard input
    output
    standard output

    During the programming classes Vasya was assigned a difficult problem. However, he doesn't know how to code and was unable to find the solution in the Internet, so he asks you to help.

    You are given a sequence a, consisting of n distinct integers, that is used to construct the binary search tree. Below is the formal description of the construction process.

    1. First element a1 becomes the root of the tree.
    2. Elements a2, a3, ..., an are added one by one. To add element ai one needs to traverse the tree starting from the root and using the following rules:
      1. The pointer to the current node is set to the root.
      2. If ai is greater than the value in the current node, then its right child becomes the current node. Otherwise, the left child of the current node becomes the new current node.
      3. If at some point there is no required child, the new node is created, it is assigned value ai and becomes the corresponding child of the current node.
     
    Input
     

    The first line of the input contains a single integer n (2 ≤ n ≤ 100 000) — the length of the sequence a.

    The second line contains n distinct integers ai (1 ≤ ai ≤ 109) — the sequence a itself.

    Output
     

    Output n - 1 integers. For all i > 1 print the value written in the node that is the parent of the node with value ai in it.

    Examples
     
    input
    3
    1 2 3
    output
    1 2
    input
    5
    4 2 3 1 6
    output
    4 2 2 4


    题意:

    给一个序列,构造一个二叉搜索树,然后输出每个节点的父节点;


    思路:

    在构造二叉搜索树的时候,每插入一个节点时它的插入位置是一定的,要么插在最大的比它小的数的右边,要么插在最小的比它
    大的数左边,用线段树维护最大最小值就可以了;也可以用set+map模拟建树的过程;

    AC代码:


    #include <bits/stdc++.h>
    /*
    #include <iostream>
    #include <queue>
    #include <cmath>
    #include <map>
    #include <cstring>
    #include <algorithm>
    #include <cstdio>
    */
    using namespace std;
    #define Riep(n) for(int i=1;i<=n;i++)
    #define Riop(n) for(int i=0;i<n;i++)
    #define Rjep(n) for(int j=1;j<=n;j++)
    #define Rjop(n) for(int j=0;j<n;j++)
    #define mst(ss,b) memset(ss,b,sizeof(ss));
    typedef long long LL;
    const LL mod=1e9+7;
    const double PI=acos(-1.0);
    const int inf=0x3f3f3f3f;
    const int N=1e5+25;
    int n,l[N],r[N],f[N];
    struct Tree
    {
        int l,r,mmin,mmax;
    }tree[4*N];
    void pushup(int node)
    {
        tree[node].mmin=min(tree[2*node].mmin,tree[2*node+1].mmin);
        tree[node].mmax=max(tree[2*node].mmax,tree[2*node+1].mmax);
    }
    void build(int node,int L,int R)
    {
        tree[node].l=L;
        tree[node].r=R;
        tree[node].mmax=0;
        tree[node].mmin=inf;
        if(L==R)return ;
        int mid=(L+R)>>1;
        build(2*node,L,mid);
        build(2*node+1,mid+1,R);
    }
    void update(int node,int pos)
    {
       // cout<<tree[node].l<<" "<<tree[node].r<<" "<<pos<<"@"<<"
    ";
        if(tree[node].l==tree[node].r&&tree[node].l==pos)
        {
            tree[node].mmax=tree[node].mmin=pos;
            return ;
        }
        int mid=(tree[node].l+tree[node].r)>>1;
        if(pos<=mid)update(2*node,pos);
        else update(2*node+1,pos);
        pushup(node);
    }
    int query(int node,int L,int R,int flag)
    {
        if(L<=tree[node].l&&R>=tree[node].r)
        {
            if(flag)return tree[node].mmax;
            else return tree[node].mmin;
        }
        int mid=(tree[node].l+tree[node].r)>>1;
        if(R<=mid)return query(2*node,L,R,flag);
        else if(L>mid)return query(2*node+1,L,R,flag);
        else
        {
            if(flag)return max(query(2*node,L,mid,flag),query(2*node+1,mid+1,R,flag));
            else return min(query(2*node,L,mid,flag),query(2*node+1,mid+1,R,flag));
        }
    }
    struct Po
    {
        int a,pos,num;
    }po[N];
    int cmp1(Po x,Po y)
    {
        return x.a<y.a;
    }
    int cmp2(Po x,Po y)
    {
        return x.pos<y.pos;
    }
    int main()
    {
        scanf("%d",&n);
        build(1,1,n);
        Riep(n)
        {
            scanf("%d",&po[i].a);
            po[i].pos=i;
        }
        sort(po+1,po+n+1,cmp1);
        Riep(n)po[i].num=i,f[i]=po[i].a;
        sort(po+1,po+n+1,cmp2);
        update(1,po[1].num);
        for(int i=2;i<=n;i++)
        {
            int s=query(1,1,po[i].num,1);
            if(s==0||(s!=0&&r[s]))
            {
                s=query(1,po[i].num,n,0);
                l[s]=po[i].num;
            }
            else
            {
                r[s]=po[i].num;
            }
            update(1,po[i].num);
            printf("%d ",f[s]);
        }
    
        return 0;
    }
    #include <iostream>
    #include <cstdio>
    #include <cstring>
    #include <algorithm>
    #include <cmath>
    #include <bits/stdc++.h>
    #include <stack>
    #include <map>
     
    using namespace std;
     
    #define For(i,j,n) for(int i=j;i<=n;i++)
    #define mst(ss,b) memset(ss,b,sizeof(ss));
     
    typedef  long long LL;
     
    template<class T> void read(T&num) {
        char CH; bool F=false;
        for(CH=getchar();CH<'0'||CH>'9';F= CH=='-',CH=getchar());
        for(num=0;CH>='0'&&CH<='9';num=num*10+CH-'0',CH=getchar());
        F && (num=-num);
    }
    int stk[70], tp;
    template<class T> inline void print(T p) {
        if(!p) { puts("0"); return; }
        while(p) stk[++ tp] = p%10, p/=10;
        while(tp) putchar(stk[tp--] + '0');
        putchar('
    ');
    }
     
    const LL mod=1e9+7;
    const double PI=acos(-1.0);
    const int inf=1e9+10;
    const int N=1e5+10;
    const int maxn=1e3+20;
    const double eps=1e-12;
    
    
    set<int>s;
    map<int,int>le,ri;
    set<int>::iterator it; 
    int main()
    {
        int n,x;
        read(n);
        read(x);s.insert(x);
        For(i,2,n)
        {
            read(x);
            it=s.lower_bound(x);
            int pos=*it;
            if(le[pos]==0&&it!=s.end())le[pos]=x;
            else 
            {
                it--;
                pos=*it;
                ri[pos]=x;
            }
            s.insert(x);
            printf("%d ",pos);
        }
        return 0;
    }
    

      

  • 相关阅读:
    Java学习开篇
    《我的姐姐》
    世上本无事,庸人自扰之
    这48小时
    补觉
    淡定
    es java api 设置index mapping 报错 mapping source must be pairs of fieldnames and properties definition.
    java mongodb groupby分组查询
    linux 常用命令
    mongodb too many users are authenticated
  • 原文地址:https://www.cnblogs.com/zhangchengc919/p/5502749.html
Copyright © 2020-2023  润新知