校招真题:分苹果——快排qsort

题目描述

n 只奶牛坐在一排,每个奶牛拥有 a i 个苹果,现在你要在它们之间转移苹果,使得最后所有奶牛拥有的苹果数都相同,每一次,你只能从一只奶牛身上拿走恰好两个苹果到另一个奶牛上,问最少需要移动多少次可以平分苹果,如果方案不存在输出 -1。

输入描述:

每个输入包含一个测试用例。每个测试用例的第一行包含一个整数 n(1 <= n <= 100),接下来的一行包含 n 个整数 ai(1 <= ai <= 100)。

输出描述:

输出一行表示最少需要移动多少次可以平分苹果,如果方案不存在则输出 -1。

思路:  判断-1的情况: 给2的时候总是多1或者少1 平均数与数字之间的差值总是不为2的倍数

         可以选择用快排从小到大

#include <iostream>
#include<cstdlib>
using namespace std;
int num[110];
int cmp(const void *a, const void *b){
    return *(int *)a > *(int *)b; //函数返回值 大于0 则qsort排序认为前者“大于”后者 本题中按值从小到大排序
}
int main()
{
    int n;
    cin>>n;
    int sum = 0;
    for(int i=0; i<n; i++)
    {
        cin>>num[i];
        sum += num[i];
    }
    if(sum%n)
    {
        cout<<-1;
        return 0;   
    }
    qsort(num, n, sizeof(int), cmp);
   /*
    for(int i=0; i<n; i++)
    {
        cout<<num[i]<<' ';
    }
    */
    int step = 0;
    int avg = sum/n;
    for(int i=n-1,j=0;i>=0&&j<=i;)
    {
        if(num[j] == avg && num[i] == avg)
            break;
        if((avg - num[j])%2 || (avg - num[i])%2)
        {
            cout<<-1;
            return 0;  
        }
        while(num[j]<avg && num[i]>avg)
        {
            num[j] += 2;
            num[i] -= 2;
            step++;
        }
        if(num[j] == avg)
            j++;
        if(num[i] == avg)
            i--;
    }
    cout<<step<<endl;
    //system("pause");
    return 0;
}

猜你喜欢

转载自blog.csdn.net/yanbao4070/article/details/80981986
今日推荐