codeforces 675D Tree Construction 数据结构

题目链接：http://www.codeforces.com/contest/675/problem/D

题意：

依次给出每个节点的值，要求最终得到一颗二叉排序树，第一个节点是根，求每个节点的父节点。

思路：

对于每个插入的节点，要么成为比它小的最大的节点右儿子，要么成为比它大的最小节点的左儿子。

把每个节点依次放入set里面用lower_bound查找就可以了。

顺便复习了一下set:

lower_bound(x)返回集合中大于或者等于x的第一个值的位置，如果找不到返回s.end()

upper_bound(x)返回集合中大于x的第一个位置，如果找不到返回s.end()

指针变量不能加和减，但是可以++或者--。

set中最后一个数的值应该为*(--s.end())

#include <bits/stdc++.h>
using namespace std;
int n;
int a[100010];
int main() 
{
    //freopen("in.txt", "r", stdin);
    while(~scanf("%d", &n))
    {
        set <int> s;
        scanf("%d", &a[1]);
        s.insert(a[1]);
        
        bool flag = false;
        map <int, int> l, r;
        for(int i = 2; i <= n; i++)
        {
            if(i > 2) printf(" ");
            scanf("%d", &a[i]);
            set <int>::iterator it = s.lower_bound(a[i]);
            if(it == s.end())  //找第一个比它大的数,如果没找到,说明这个数最大,只能在最后的右儿子
            {
                set <int>::iterator it2 = --s.end();
                printf("%d", *it2);
                r[*it2] = 1;
            } 
            else //如果找到了一个最接近的比它大的数
            {
                int val = *it;
                if(l[val] == 1)  //如果左节点已经有了
                {
                    set <int>::iterator it2 = --it;
                    printf("%d", *it2);
                    r[*it2] = 1;
                }
                else            //如果左节点没有
                {
                    printf("%d", val);
                    l[val] = 1;
                }
            }
            s.insert(a[i]);
        }
        printf("\n");
        
    }
    return 0;
}