2019年网易有道事业部 计算机视觉工程师内推笔试第1道编程题

2019年网易有道事业部 计算机视觉工程师内推笔试第1道编程题

题目

有n座塔，每座塔都有一定的高度，每次可以将一座塔的高度移动一到另外一座塔的上面，给定最大可移动次数k，求最大高度塔和最小高度塔之差，以及移动的次数。

输入 :
      3 2
      5 8 5
输出：
      0 2 //0为最小高度差
      2 1//将第2个塔移动1到第1个塔
      2 3//将第2个塔移动1到第3个塔

#include<iostream>
#include<vector>
#include<algorithm>
#include<string>
using namespace std;
struct Data
{
    int idx;
    int value;
};
struct ANS
{
    int x;
    int y;
};
bool cmp(const Data& a,const Data& b)
{
    if(a.value < b.value)//首先按照值的大小排
    {
        return true;
    }
    else if(a.value == b.value)
    {
        if(a.idx < b.idx)//值的大小相等 按照idx排
            return true;
        else
            return false;
    }
    else
        return false;
}
int main()
{
    int n,k;
    vector<Data>arr;
    vector<ANS>ans;
    scanf("%d %d",&n,&k);
    for(int i = 0;i < n; i++)
    {
        Data temp;
        scanf("%d",&temp.value);
        temp.idx = i + 1;
        arr.push_back(temp);
    }
    sort(arr.begin(),arr.end(),cmp);
    int s = MAX;
    int m = 0;
    for(int i = 0;i < k;i++)
    {
        if(arr[n-1].value - arr[0].value > 1)
        {
            m++;
            arr[n-1].value -= 1;
            arr[0].value += 1;
            ANS temp;
            temp.x = arr[n-1].idx;
            temp.y = arr[0].idx;
            ans.push_back(temp);
            sort(arr.begin(),arr.end(),cmp);
        }
        else
        {
            break;
        } 
    }
    s = arr[n-1].value - arr[0].value;
    printf("%d %d\n",s,m);
    for(int i = 0; i< m;i++)
    {
        printf("%d %d\n",ans[i].x,ans[i].y);
    }
    return 0;
}