C T3 最少移动

C T3 最少移动

随机选两个相邻的数，一个加1，一个减1，最后要所有数都一样，那么就都是平均数。所以一开始想判断是否每个数都能变成相同。如果可以，就从尾到头遍历，和平均值比较，大了就减，然后前一个加。

for _ in range(int(input())):
    n = int(input())
    lst = list(map(int , input().split()))
    if sum(lst) % len(lst) != 0 :
        print(-1)
    else:
        ans=0
        ave = sum(lst) // len(lst)
        for i in range(n-1 , -1 , -1):
            if lst[i] != ave:
                ans += abs (lst[i] - ave)
                if lst[i] > ave:
                    lst[i-1] += abs (lst[i] - ave)
                else:
                    lst[i-1] -= abs (lst[i] - ave)
        print(ans)