Cf #353 D. Tree Construction

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

 

题目大意是将一个没有相同数字的数列中的数字依次插入到二叉搜索树中,问除了第一个数字以外,其他数字的父亲节点是哪个数字。

求每个数字之前最小的比它大的数字以及最大的比它小的数字。然后看那两个数字哪个数字比较后插入,则父亲节点为后插入的数字。当然如果只找出一个数字的话,那么就直接输出那个数字就好。

用Java写了一下,我原先一直不清楚java中类似于C++集合的lower_bound和upper_bound是什么函数。查了下资料,其实是有四个分别是lower higher floor ceiling 其中前两个是严格大于或者小于的,后两个是不严格大于小于的。

 

 1 import java.io.OutputStream;
 2 import java.io.IOException;
 3 import java.io.InputStream;
 4 import java.io.PrintWriter;
 5 import java.util.StringTokenizer;
 6 import java.util.Map;
 7 import java.util.HashMap;
 8 import java.io.IOException;
 9 import java.io.BufferedReader;
10 import java.io.InputStreamReader;
11 import java.util.TreeSet;
12 import java.io.InputStream;
13 
14 
15 public class Main {
16     public static void main(String[] args) {
17         InputStream inputStream = System.in;
18         OutputStream outputStream = System.out;
19         InputReader in = new InputReader(inputStream);
20         PrintWriter out = new PrintWriter(outputStream);
21         TaskD solver = new TaskD();
22         solver.solve(1, in, out);
23         out.close();
24     }
25 
26     static class TaskD {
27         public void solve(int testNumber, InputReader in, PrintWriter out) {
28             int n = in.nextInt();
29             TreeSet<Integer> set = new TreeSet<Integer>();
30             Map<Integer, Integer> map = new HashMap<Integer, Integer>();
31             map.put(0, -1);
32             for (int i = 0; i < n; i++) {
33                 int a = in.nextInt();
34                 if (i > 0) {
35                     int d = (set.lower(a) == null ? 0 : set.lower(a));
36                     int u = (set.higher(a) == null ? 0 : set.higher(a));
37                     out.println((map.get(d) > map.get(u) ? d : u) + " ");
38                 }
39                 map.put(a, i);
40                 set.add(a);
41             }
42         }
43 
44     }
45 
46     static class InputReader {
47         private BufferedReader reader;
48         private StringTokenizer tokenizer;
49 
50         public InputReader(InputStream stream) {
51             reader = new BufferedReader(new InputStreamReader(stream));
52             tokenizer = null;
53         }
54 
55         public String next() {
56             while (tokenizer == null || !tokenizer.hasMoreTokens()) {
57                 try {
58                     tokenizer = new StringTokenizer(reader.readLine());
59                 } catch (IOException e) {
60                     throw new RuntimeException(e);
61                 }
62             }
63             return tokenizer.nextToken();
64         }
65 
66         public int nextInt() {
67             return Integer.parseInt(next());
68         }
69 
70     }
71 }

 

posted @ 2016-05-23 14:31  活在夢裡  阅读(222)  评论(0编辑  收藏  举报