时间限制:1.000S  空间限制:32MB

题目描述

小明很喜欢玩积木。一天,他把许多积木块组成了好多高度不同的堆,每一堆都是一个摞一个的形式。然而此时,他又想把这些积木堆变成高度相同的。但是他很懒,他想移动最少的积木块来实现这一目标,你能帮助他吗?

输入描述

输入包含多组测试样例。每组测试样例包含一个正整数n,表示小明已经堆好的积木堆的个数。
接着下一行是n个正整数,表示每一个积木堆的高度h,每块积木高度为1。其中1<=n<=50,1<=h<=100。
测试数据保证积木总数能被积木堆数整除。
当n=0时,输入结束。

输出描述

对于每一组数据,输出将积木堆变成相同高度需要移动的最少积木块的数量。
在每组输出结果的下面都输出一个空行。

输入示例
6
5 2 4 1 7 5
0
输出示例
5

答案

package main
 
import (
    "bufio"
    "os"
    "fmt"
    "strings"
    "strconv"
)
 
var (
    n int
    arr []string
    intArr []int
    cur int
    sum int
    avg int
    step int
)
 
func main() {
    input := bufio.NewScanner(os.Stdin)
    for input.Scan() {
        n, _ = strconv.Atoi(input.Text())
        if n == 0 {
            break
        }
        input.Scan()
        arr = strings.Split(input.Text(), " ")[:n]
        intArr = make([]int, n)
        for i, v := range arr {
            cur, _ = strconv.Atoi(v)
            sum += cur
            intArr[i] = cur
        }
        avg = sum / n
        for _, v := range intArr {
	        if v - avg > 0 {
		        step += v - avg
	        }
        }
        fmt.Printf("%d\n\n", step)
        sum = 0
        step = 0
    }
}