python 数据分析 笔试_腾讯2019年暑期实习生招聘在线笔试技术研究和数据分析方向第二题（python）...

def printindex(n,arr):

# n = int(input())

# arr = list(map(int,input().split(' ')))

li1=[]

li2=[]

for i in range(n):

if(arr[i]<0):

li1.append(i)

else:

li2.append(i)

index = 0

for i in li2:

for j in li1:

if abs(arr[i]>abs(arr[j])):

index +=abs(j-i)*abs(arr[j])

arr[i] = arr[i] -abs(arr[j])

arr[j] =0

else:

index +=abs(j-i)*abs(arr[i])

arr[j] = arr[j]+arr[i]

arr[i]=0

break

return index

# print(index)

if __name__ == '__main__':

test1 = {'l1': [5], 'l2': [5, -4, 1, -3,1], 'result': [9]}

# test2 = {'l1': [2, 2], 'l2': [4, 6], 'result': [4 ,2]}

test = [test1]

count = 0

for t in test:

if printindex(t['l1'], t['l2']) == t['result']:

count += 1

else:

print('l1:', t['l1'], '\nl2:', t['l2'], '\nOutput:', printindex(t['l1'], t['l2']), '\nExpect:',

t['result'])

print('Case通过率：', count * 100 / len(test), '%')