阿里春招实习生笔试3.19第一题

阿里春招实习生笔试3.19第一题

1. n个人参加一个促销活动，每个人手里有个数字啊 a[i]，如果某个人手里的数字是平方数（4，9，16…），则他中奖！
2. 现有一种修改券，可以让人们手中的数字 +1 或者 -1，每个人可以用多张修改券。
3. 想要保证n/2的人中奖，最少要准备多少张修改券?
4. 题目测试集保证 n 为偶数。

python解法

import math

n = int(input())
lst = list(map(int,input().split()))
lst_1 = lst[:]
a,c,f= [0]*n,[0]*n,[0]*n

for i in range(n):
	ab = (math.sqrt(lst[i]))%1
	while ab != 0.0:
		lst[i] += 1
		ab = (math.sqrt(lst[i]))%1
		a[i] += 1

for i in range(n):
	b = (math.sqrt(lst_1[i]))%1
	while b != 0.0:
		lst_1[i] -= 1
		b = (math.sqrt(lst_1[i]))%1
		c[i] += 1

for i in range(n):
	f[i] = min(a[i],c[i])

f.sort()
x = int(n//2)
sum = 0
for i in range(x):
	sum += f[i]
print(sum)