分苹果 python实现

题目描述

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

输入描述:

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

输出描述:

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

示例

输入：

4

15 7 5 9

输出：

3

我的代码：

num = int(raw_input())
nums = [int(i) for i in raw_input().split()]
if num ==1:
	print 0
else:
	isum = sum(nums)
	ave = isum*1.0/num
	inums = [i-ave for i in nums]
	# print inums
	count = 0
	for i in inums:
		# print i%2

		if i%2 != 0.0:
			print -1
			break
		elif i > 0:
			count += int(i/2)
	if count !=0:
		print count