0
点赞
收藏
分享

微信扫一扫

查找第K小数(北邮机试)

东林梁 2022-03-11 阅读 23
算法

Time Limit: 1000 ms
Memory Limit: 256 mb
查找一个数组的第K小的数,注意同样大小算一样大。 如 2 1 3 4 5 2 第三小数为3。

输入输出格式
输入描述

输出描述

输入输出样例
输入样例

6
2 1 3 5 2 2
3

输出样例

3

此题有两种思路:
①用普通数组对数据进行存储,然后sort排序再去重即可,代码如下:

#include <bits/stdc++.h>
using namespace std;

int main()
{
	int a[1000];
	int n,q;
	while(cin>>n)
	{
		for(int i=0;i<n;i++)
			cin>>a[i];
		sort(a,a+n);
		cin>>q;
		int cnt=0;
		for(int i=0;i<n;i++)
		{

			if(i!=0)//去重
				while(a[i]==a[i-1])
					i++;
			cnt++;
			if(cnt==q)
			{
				cout<<a[i]<<endl;
				break;
			}
		}
	}
	return 0;
}

②采用set集合对元素进行存储,insert后数据自动按照从小到大的顺序进行排列,代码如下:

#include <bits/stdc++.h>
using namespace std;

int main()
{
	int n;
	while(cin>>n){
		int a[n];
		for(int i=0; i<n; i++)
			cin>>a[i];
		set<int> s;
		for(int i=0; i<n; i++)
			s.insert(a[i]);
		int k;
		cin>>k;
		auto p=s.begin();
		while(--k) p++;
		cout<<*p;
	}	
}

这里对查找问题进行一个总结:

  • map对百分之99的查找问题都能够很好的解决(这题由于需要排序就用的是set,map排序只能直接对key进行排序)
  • string,int型查找数据是否在可以直接用find()对value进行查找
  • 如果直接是看某int数据是否在则可以直接用int作为key值,存在则map[key]++
举报

相关推荐

0 条评论