网易校招题之塔
题目描述
小易有一些立方体，每个立方体的边长为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
输出
0 2 2 1 2 3
#include<iostream>
#include<algorithm>
#include<stdio.h>
#include <vector>
#include<string>
#include<map>
using namespace std;
bool cmp(pair<int, int>& a, pair<int, int>& b)
{
if (a.second == b.second)
{
return a.first > b.first;
}
return a.second > b.second;
}
int main()
{
int n, k;
while (cin >> n >> k)
{
vector<pair<int, int>> nums;
vector<pair<int, int>> res;
for (int i = 0; i < n; ++i)
{
int temp;
cin >> temp;
nums.push_back(make_pair(i + 1, temp));//哈希表，将下标和塔的高度相对应
}
for (int i = 0; i < k; ++i)
{
sort(nums.begin(), nums.end(), cmp);
if (nums[0].second - nums[n - 1].second <= 1) //注意是小于等于1
break;
nums[0].second--;
nums[n - 1].second++;
res.push_back(make_pair(nums[0].first, nums[n - 1].first));//将移动的下标保存
}
sort(nums.begin(), nums.end(), cmp);
cout << nums[0].second - nums[n - 1].second << " " << res.size() << endl;
for (int i = 0; i < res.size(); ++i)
{
cout << res[i].first << " " << res[i].second << endl;
}
}
return 0;
} 
投递牛客等公司8个岗位 >
