- 合并果子
- 2017-10-03 18:18:58 @
请大家指教,哪里可以优化呢?是sort()耗时吗?
```Python 3
series = int(input())
nums = list(map(int,(input().split())[:series]))
total = 0
if series ==1:
print(0)
elif series == 2:
print(sum(nums))
elif series > 2:
while True:
nums.sort()
min2sum=nums[0]+nums[1]
nums = nums[2:]
nums.append(min2sum)
total += min2sum
if len(nums) == 2:
total += sum(nums)
print(total)
break
```
import bisect
。。。。。。。。。。。
while True:
min2sum=nums[0]+nums[1]
nums = nums[2:]
bisect.insort(nums,min2sum)
。。。。。。。。。。。。。。。。
1 条评论
-
idou8 LV 7 @ 2017-10-05 22:21:50
已经解决,使用bisect.insort
不用每次都排序了
- 1