今日头条面试记录

只问了我一道题,如何求第k大并且不能改变原数组顺序。。太久没做题,脑子不太好使,面试官很nice,给了我提示,但是我还是没当场想出来。。。后来我想了下,好像是记录最大值和最小值,然后二分答案转换为判断问题,复杂度O(nlogn),额如果这个不是正解的话,感觉我也想不出其他解法了,记录一下自己失败经历吧。。。。最后感谢老朋友的内推,是我自己不争气。

#include <cstdio>
#include <ctime>
#include <cstring>
#include <random>
#include <cmath>
#include <stack>
#include <vector>
#include <queue>
#include <map>
#include <string>
#include <algorithm>
using namespace std;
#define MAXN 15
#define MAXM 130
typedef long long LL;
const int INF = 0x3f3f3f3f;
int a[MAXN], n, k;
int check(int mid){
	int sum = 0;
	for (int i = 0; i < n; i++){
		if (a[i] > mid){
			sum++;
		}
	}
	return sum + 1;
}
int main(){
	//freopen("input.txt", "r", stdin);
	scanf("%d%d", &n, &k);
	int minmum = INF, maxmum = -INF;
	for (int i = 0; i < n; i++){
		scanf("%d", &a[i]);
		minmum = min(minmum, a[i]);
		maxmum = max(maxmum, a[i]);
	}
	int ans;
	while (minmum <= maxmum){
		int mid = minmum + maxmum >> 1;
		if (check(mid)<=k){
			ans = mid;
			maxmum = mid - 1;
		}
		else{
			minmum = mid + 1;
		}
	}
	printf("%d\n", ans);
	//fclose(stdin);
}
这里应该还要加特判k是否小于等于数组长度,代码就不加了,懂就好了。。。

猜你喜欢

转载自blog.csdn.net/qq_34262582/article/details/80582261