仓库 算法提高 java（蓝桥杯 ）

资源限制

内存限制：256.0MB   C/C++时间限制：1.0s   Java时间限制：3.0s   Python时间限制：5.0s

问题描述

　　沿着一条道路有 N 个仓库,相邻两个仓库间距离为 1 公里。假设第 i个仓库存有 Xi 个单位的货物。现在希望将货物重新分配,使得最终每个仓库内的货物量 Xi 相等。
　　假设将 1 个单位的货物运送 1 公里的耗费为 1 个金币。请你计算出为了让各个仓库内的货物相等最少需要花费多少个金币。

输入格式

　　输入第一行有一个数,N (1 ≤ N ≤ 1000) ,表示仓库的个数。
　　第二行为用空格隔开的 N 个正整数,依次表示 X1 到 XN 。假定 Xi 不超过 1000的非负整数。所有 Xi 的和一定为 N 的倍数。

输出格式

　　输出文件只有一行,一个整数,表示最少花费。

样例输入

3
1 6 2

样例输出

3
对样例的解释
先从 2 号仓库运送 2 个单位的货物给 1 号仓库,然后从 2 号仓库运送
1 个单位的货物给 3 号仓库,所以总费用为 2 × 1+ 1 × 1 = 3。

package lanqiao;

import java.util.Scanner;

public class 仓库 {
	public static void main(String[] args) {
		Scanner scanner=new Scanner(System.in);
		int n=scanner.nextInt();
		int arr[]=new int[n];
		int num=0;
		for (int i = 0; i < arr.length; i++) {
			arr[i]=scanner.nextInt();
			num+=arr[i];
		}
		num=num/n;
		int money=0;
		for (int i = 0; i < arr.length; i++) {
			if (arr[i]!=num) {
				arr[i+1]+=(arr[i]-num);
				money+=Math.abs(arr[i]-num);
				arr[i]=num;
			}
		}
		System.out.println(money);
	}
}

思路是：遍历数组 如果arr[i]小于最后的平均值 那么从后面直接拿过来 后面减少 如果大于的话 直接放到后面 最后求出来就是最优解