腾讯2020秋招笔试真题——4.求最小非零元素

【题目描述】Q哥给了小Q一个长度为n正整数序列ai。

Q哥要求小Q重复以下操作步骤k轮:

1、发现最小的非零元素x。

2、打印x。

3、将序列中所有非零元素减x。

小Q把这个艰巨的任务交给了你,希望你能帮帮他。

输入描述

输入包括两行。

第一行包括两个正整数n和k(1<=n,k<=105)。

第二行包括n个正整数ai(1<=ai<=109)。

输出描述

输出k行,即每轮中的最小非零整数(如果到某轮所有元素都是0,打印0即可)。

示例1

输入

1
2
7 5
5 8 10 3 6 10 8

输出

3
2
1
2
2

【解题思路】
1）方法1：使用动态数组

bool cmp_num(int i, int j){
	return i < j;
}
int main(){
	int n, k,l;
	cin >> n;
	cin >> k;
	vector<int> vec;
	for (int i = 0; i < n; i++){
		cin >> l;
		vec.push_back(l);
	}
	sort(vec.begin(), vec.end(), cmp_num);
	while (k--){
		int m = -1;//找到第一个不为零的索引
		for (int i = 0; i < n; i++){
			if (vec[i] != 0){
				m = i;
				break;
			}
		}
		cout << vec[m] << endl;
		int x = vec[m];
		for (int j = m; j < n; j++)
		{
			vec[j] -= x;
		}					
	}
}

2）方法2：使用小根堆

priority_queue<int, vector<int>, greater<int>> Q;//小根堆
int main() {
	int n, k;
	cin >> n;
	cin >> k;
	for (int i = 1; i <= n; i++) {
		int x;
		cin >> x;
		Q.push(x);
	}
	int del = 0;
	while (k--) {
		while (!Q.empty() && Q.top() - del <= 0)
			Q.pop();
		if (Q.empty())
			cout<<0<<endl;
		else {
			int x = Q.top();
			Q.pop();
			x -= del;
			cout << x;
			del += x;
		}
	}
}