百度2017春招笔试题一——买帽子

百度2017春招笔试题一——买帽子

题目来自牛客网，感谢！

从一列有可能重复的数中取出第三小的数值。

输入格式

第一行输入N，代表数的个数
第二行输入N个数

输出格式

第一行输出第三小的数值（相同数值视为同一），若不存在则输出-1

输入范例

10
10 10 10 10 20 20 30 30 40 40

输出范例

30

重点在于数据的去重排序。在Java中，使用TreeSet即可实现这一目的。

我的代码实现

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
import java.util.Set;
import java.util.TreeSet;

public class PriceOfHat {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		Set<Integer> set = new TreeSet<Integer>();
		for (int i = 0; i < n; i++)
			set.add(sc.nextInt());
		sc.close();

		if (set.size() <= 2)
			System.out.println(-1);
		else {
			List<Integer> list = new ArrayList<Integer>(set);
			System.out.println(list.get(2));
		}
	}
}