贪心--均分纸牌

问题 B: 【贪心】均分纸牌

时间限制: 1 Sec   内存限制: 64 MB
提交: 164   解决: 95
[ 提交][ 状态][ 讨论版]

题目描述

有N堆纸牌，编号分别为1，2，…，N。每堆上有若干张，但纸牌总数必为N的倍数。可以在任一堆上取若干张纸牌，然后移动。
移牌规则为：在编号为1堆上取的纸牌，只能移到编号为2的堆上；在编号为N的堆上取的纸牌，只能移到编号为N-1的堆上；其他堆上取的纸牌，可以移到相邻左边或右边的堆上。
现在要求找出一种移动方法，用最少的移动次数使每堆上纸牌数都一样多。
例如 N=4，4堆纸牌数分别为：
① 9　② 8　③ 17　④ 6
移动3次可达到目的：
从③ 取4张牌放到④ （9 8 13 10）-> 从③ 取3张牌放到② （9 11 10 10）->从② 取1张牌放到① （10 10 10 10）。

输入

输入文件中包括两行数据。
第一行为N堆纸牌数（1<=N<=100）。
第二行为N堆纸牌中每堆纸牌初始数A1，A2，…，An（l<=Ai<=10000）。

输出

输出文件中仅一行，即所有堆均达到相等时的最少移动次数。

样例输入

4

9 8 17 6

样例输出

3

贪心经典例题，只考虑最左边未考虑的元素，若大于平均值则将大于的部分全部都给右边第一个元素，若小于平均值则拿右边第一个直到该元素到平均值（将右边元素拿到负数的情况转换成不拿到负数的最优情况，只需要改变移动的顺序即可，所以拿到负数依然最优）

#include<stdio.h>
#include <algorithm>
#include<iostream>
#include<string.h>
#include<vector>
#include<stdlib.h>
#include<math.h>
#include<queue>
#include<deque>
#include<ctype.h>
#include<map>
#include<set>
#include<stack>
#include<string>
#include<algorithm>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
 
int a[105];
 
 
int main()
{
    int n;
    scanf("%d",&n);
 
    int sum=0;
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
        sum+=a[i];
    }
    sum/=n;
 
    int ans=0;
    for(int i=0;i<n;i++)
    {
        if(a[i]==sum)
            continue;
 
        for(int j=i+1;j<n;j++)
            if(a[i]<sum)
            {
                int temp=sum-a[i];
                a[j]-=temp;
                a[i]+=temp;
                ans++;
            }
            else if(a[i]==sum)
                break;
 
        for(int j=i+1;j<n;j++)
            if(a[i]>sum)
            {
                int temp=a[i]-sum;
                a[i]-=temp;
                a[j]+=temp;
                ans++;
            }
            else if(a[i]==sum)
                break;
    }
 
    printf("%d\n",ans);
    return 0;
}