网易 [编程题] 塔

[编程题] 塔

时间限制：1秒

空间限制：262144K

小易有一些立方体，每个立方体的边长为1，他用这些立方体搭了一些塔。

现在小易定义：这些塔的不稳定值为它们之中最高的塔与最低的塔的高度差。

小易想让这些塔尽量稳定，所以他进行了如下操作：每次从某座塔上取下一块立方体，并把它放到另一座塔上。

注意，小易不会把立方体放到它原本的那座塔上，因为他认为这样毫无意义。

现在小易想要知道，他进行了不超过k次操作之后，不稳定值最小是多少。

 

输入描述:

第一行两个数n,k (1 <= n <= 100, 0 <= k <= 1000)表示塔的数量以及最多操作的次数。
第二行n个数，ai(1 <= ai <= 104)表示第i座塔的初始高度。

输出描述:

第一行两个数s, m，表示最小的不稳定值和操作次数(m <= k)
接下来m行，每行两个数x,y表示从第x座塔上取下一块立方体放到第y座塔上。

输入例子1:

3 2
5 8 5

输出例子1:

0 2
2 1
2 3

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;

bool cmp(pair<int, int> &a, pair<int, int> &b)
{
	return a.second < b.second;
}
int main()
{
	
	int n, k;
	cin >> n >> k;
	vector<pair<int, int>> vec(n + 1, make_pair(0, 0));
	vector<pair<int, int>> res;

	for (int i = 1; i <= n; i++)
	{
		int temp;
		cin >> temp;
		vec[i] = make_pair(i, temp);
	}


	sort(vec.begin() + 1, vec.end(), cmp);
	int i = 0;

	while (i < k)
	{
		int low = vec[1].second;
		int high = vec[vec.size() - 1].second;
		if (low == high)
			break;
		vec[vec.size() - 1].second--;
		vec[1].second++;
		res.push_back(pair<int, int>(vec[vec.size() - 1].first, vec[1].first));
		sort(vec.begin() + 1, vec.end(), cmp);
		i++;
	}

	cout << vec[vec.size() - 1].second - vec[1].second << " " << i << endl;
	for (auto it : res)
	{
		cout << it.first << " " << it.second << endl;
	}
	return 0;

}