码迷,mamicode.com
首页 > 其他好文 > 详细

Codeforces Round #353 (Div. 2) D. Tree Construction (二分,stl_set)

时间:2016-05-17 19:34:01      阅读:194      评论:0      收藏:0      [点我收藏+]

标签:

题目链接:http://codeforces.com/problemset/problem/675/D

给你一个如题的二叉树,让你求出每个节点的父节点是多少。

用set来存储每个数,遍历到a[i]的时候查找比a[i]大的数的位置,然后插入,而父亲就是刚好比a[i]小的数或刚好大的数。

然后讨论是哪一个数。

比如给你3 1 2 ,如图

技术分享

1的父亲是3 ,2的父亲是1。

那我其实只要找左边或右边出现最晚的数就行了,用pair的first表示a[i],second表示出现的顺序i。

 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 typedef pair <int , int> P;
 4 int a[int(1e5 + 5)] , ans[int(1e5 + 5)];
 5 set <P> v;
 6 int main()
 7 {
 8     int n;
 9     scanf("%d" , &n);
10     for(int i = 0 ; i < n ; ++i) {
11         if(i == 0) {
12             scanf("%d" , a + i);
13             v.insert(P(a[i] , i));
14             continue;
15         }
16         scanf("%d" , a + i);
17         auto it = v.upper_bound(P(a[i] , 0));
18         if(it == v.begin()) {
19             ans[i] = it->first;
20         }
21         else if(it == v.end()) {
22             ans[i] = (--it)->first;
23         }
24         else {
25             if(it->second > (--it)->second)
26                 ans[i] = (++it)->first;
27             else
28                 ans[i] = it->first;
29         }
30         v.insert(P(a[i] , i));
31     }
32     for(int i = 1 ; i < n - 1 ; ++i)
33         printf("%d " , ans[i]);
34     printf("%d\n" , ans[n - 1]);
35 }

 

技术分享

Codeforces Round #353 (Div. 2) D. Tree Construction (二分,stl_set)

标签:

原文地址:http://www.cnblogs.com/Recoder/p/5502592.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!