算法题（四十二）：网易2017笔试题——“分苹果”

再思考（大神过来满足成就感啊）

如果是每次移动只能给相邻的牛，并且去掉“恰好移动2”条件的话如何解？这样题的难度增加了不少。

初步思路是从左到右遍历时，如果arr[i]大于均值，则它一定是向与它最近的一个小于均值的arr[j]补充；如果arr[i]小于均值，则它一定是从与它最近的一个大于均值的arr[j]里补充。代码如代码二所示。

输入示例：7, 3, 5, 7, 5, 5, 5, 3

输出示例：5

输入示例：7,3,3,7,7,3,5,5

输出示例：3

是否正确，或有其他解还请看到的大神指点一下。

题目描述

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

输入描述:

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

输出描述:

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

示例1

输入

4
7 15 9 5

输出

3

---

分析

只要理解题意，就变得很简单了。如果苹果的平均数不是整数，直接返回-1。如果牛的苹果数高于平均值，但与平均值的差额不能被2整除，说明无论怎么移，都有不满足“只能从一只奶牛身上拿走恰好两个苹果”，因此直接返回-1；如果能整除，则移动次数就是差额/2。按此方法遍历数组即可。

笔者在答题时遇到如此情况，按照苹果数量是高于平均值来做（即去差额）可以完全AC，按照苹果数量是否低于平均值来做（即补差额）只能通过90%。后来想到，因为补差额时，拿来的2个不一定放到同一个地方。

代码

import java.util.Scanner;
public class Main{
    public static void main(String[] args){
        Scanner input = new Scanner(System.in);
        int len = input.nextInt();
        int[] arr = new int[len];
        for(int i=0; i<len; i++){
            arr[i]=input.nextInt();
        }
        int res = fun(arr);
        System.out.println(res);
    }
    public static int fun(int[] arr){
		int count = 0;
		int sum = 0;
		int ave = 0;
		int len = arr.length;
		for(int i=0; i<len; i++){
			sum += arr[i];
		}
		if(sum%len != 0){
			return -1;
		}
		ave = sum/len;
		for(int i=0; i<len; i++){
			if(ave<arr[i]){
                //错误答案if(0 != (ave-arr[i])%2)
				if(0 != ((arr[i]-ave)%2)){
					return -1;
				}
				count += (arr[i]-ave)/2;
			}
		}
		return count;
	}
}

代码二

public class AppleAdvance {

	public static void main(String[] args) {
		//int[] arr = { 7, 3, 5, 7, 5, 5, 5, 3 };//5
		int[] arr = { 7,3,3,7,7,3,5,5};//3
		int res = fun(arr);
		System.out.println(res);
	}

	public static int fun(int[] arr) {
		int len = arr.length;
		// int sum = 0;
		int ave = 0, sum = 0, count = 0;
		boolean[] flag = new boolean[len];
		for (int i = 0; i < len; i++) {
			sum += arr[i];
		}
		if (sum % len != 0) {
			return -1;
		}
		ave = sum / len;
		for (int i = 0; i < len; i++) {
			if (arr[i] > ave) {
				flag[i] = true;
			}
		}
		for (int i = 0; i < len; i++) {
			if (arr[i] < ave) {
				for (int j = 0; j < len; j++) {
					if (j != i && flag[j]) {
						int give = (ave - arr[i]) > (arr[j] - ave) ? (arr[j] - ave) : (ave - arr[i]);
						count += Math.abs(i - j);
						if (arr[j] - give == ave) {
							flag[j] = false;
						}

						if (arr[i] + give == ave) {
							break;
						}

					}
				}
			}
			/*
			 * if(arr[i]>ave){
			 * 
			 * }
			 */
		}
		ave = sum / len;
		return count;
	}
}