算法练习——分苹果

题目描述

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

输入描述:

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

输出描述:

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

 

输入

4
7 15 9 5

输出

3

import java.util.Scanner;

public class Main{
	public static void main(String[] args) {
		Scanner sc=new Scanner(System.in);
		int n=sc.nextInt();
		int []arr=new int[n];
		for (int i = 0; i < n; i++) {
			arr[i]=sc.nextInt();
		}
		System.out.println(solve(arr,n));
	}
	public static int solve(int []arr,int n) {
		int sum=0;
		int count=0;
		int avg=0;
		int temp=0;
		for (int i = 0; i < n; i++) {
			sum=sum+arr[i];
		}
		if (sum%n!=0) {
			count=-1;
			return count;
		}
		else {
			avg=sum/n;
			for (int i = 0; i < n; i++) {
				if ((Math.abs(avg-arr[i]))%2!=0) {
					count=-1;
					break;
				}
				if(arr[i]<avg){
                    count=count+(avg-arr[i])/2;
                }
				
			}
		}
		return count;
	}
}