网易算法题,分苹果

分苹果

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

输入描述:

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

输出描述:

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

示例

输入:
         4
         7 15 9 5
输出:
         3

python代码


a=input("")
b=raw_input("")
b=[int(e) for e in b.split()]
sum_data=sum(b)
average_data=sum_data/int(a)
if sum_data%int(a)!=0:
    print -1
else:
    judge_data=[(average_data-e)%2 for e in b]
    if judge_data.count(1)!=0:
        print -1
    else:
        less_data=[average_data-e for e in b if e<average_data]
        print sum(less_data)/2

猜你喜欢

转载自blog.csdn.net/chr1341901410/article/details/80770188