【网易】塔

我怕爱的太早我们不能终老 提交于 2019-11-29 06:23:46

题目描述
小易有一些立方体,每个立方体的边长为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

思路:
思路很简单,就是每次操作将最高的塔搬一块到最低的塔,一开始想用排序加map,发现由于塔的高度在不断改变,map会失效
观察到塔的数量不是很多,因此直接遍历找最高和最低的塔即可,这样塔的顺序也没有被改变
值得注意的是两个地方:
1.k为0时,不要打印出操作
2.最坑的地方–>如果最后几个操作的不稳定值没变,可以选择不做最后几步,因此我们还要回滚s数组。

#include <iostream>
#include <vector>
#include <unordered_map>
#include <limits.h>
#include <algorithm>
using namespace std;

int main()
{
    int n,k;
    cin>>n>>k;
    vector<int> tower(n);
    vector<pair<int,int>> record;
    vector<int> s;
    for(int i=0;i<n;++i)
       cin>>tower[i];
    int m;
    for(m=0;m<=k;++m)
    {
        int high=0,low=0;
        for(int i=0;i<n;++i)
        {
            if(tower[i]>tower[high]) high=i;
            if(tower[i]<tower[low]) low=i;
        }
        s.push_back(tower[high]-tower[low]);
        if(s.back()<2) break;
        record.push_back(make_pair(high+1,low+1));
        tower[high]--;
        tower[low]++;
    }
    /* 回滚数组 */ 
    while(s.size()>1&&s[s.size()-1]==s[s.size()-2])
        s.pop_back();
    m=s.size()-1;
    if(0==k) m=0;
    cout<<s.back()<<" "<<m<<endl;
    for(int i=0;i<m;++i)
        cout<<record[i].first<<" "<<record[i].second<<endl;
    return 0;
}
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!