题目1174：查找第K小数

最新推荐文章于 2022-04-26 20:07:18 发布

kgduu

最新推荐文章于 2022-04-26 20:07:18 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

分类专栏： OJ

本文链接：https://blog.youkuaiyun.com/xiexingshishu/article/details/19640761

OJ 专栏收录该内容

986 篇文章

订阅专栏

本文介绍了一种使用TreeSet来查找数组中第K小元素的方法。通过读取输入的整数并将其添加到TreeSet中，去除重复值后按顺序遍历集合找到指定位置的元素。此方法适用于竞赛编程及算法实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

查找一个数组的第K小的数，注意同样大小算一样大。
如 2 1 3 4 5 2 第三小数为3。

输入：

输入有多组数据。
每组输入n，然后输入n个整数(1<=n<=1000)，再输入k。

输出：

输出第k小的整数。

样例输入：

6
2 1 3 5 2 2
3

样例输出：

import java.io.IOException;
import java.io.FileReader;
import java.io.InputStreamReader;
import java.io.BufferedReader;
import java.util.Scanner;
import java.util.PriorityQueue;
import java.util.TreeSet;
import java.util.Comparator;
import java.util.Iterator;

class Main
{
	public static final boolean DEBUG = false;
	
	public static void main(String[] args) throws IOException
	{
		Scanner cin;
		int n;
		
		if (DEBUG) {
			cin = new Scanner(new FileReader("d:\\OJ\\uva_in.txt"));
		} else {
			cin = new Scanner(new InputStreamReader(System.in));
		}
		
		Comparator<Integer> cmp = new Comparator<Integer>() {
			public int compare(Integer a, Integer b) 
			{
				return b - a;
			}
		};
		
		while (cin.hasNext()) {
			n = cin.nextInt();
			TreeSet<Integer> ts = new TreeSet<Integer>();
			
			for (int i = 0; i < n; i++) {
				int num = cin.nextInt();
				if (ts.contains(num)) continue;
				else {
					ts.add(num);
				}
			}
			
			//System.out.println("size:" + ts.size());
			Iterator<Integer> it = ts.iterator();
			int cnt = 0;
			int m = cin.nextInt();
			while (it.hasNext()) {
				cnt++;
				int num = it.next();
				if (cnt == m) {
					System.out.println(num);
					break;
				}
				
			}
		}
	}
}