题目来源:传送门
题目描述:
给一非负整数数组. 取数组中的一部分元素, 使得它们的和大于数组中其余元素的和, 求出满足条件的元素数量最小值. 数组中至少有一个正数。
输入示例:
示例1:
输入: nums = [3, 1, 7, 1],
输出: 1
示例2:
输入: nums = [2, 1, 2]
输出: 2
解题思路:
今天这道题看上去没那么复杂,不知道为啥题库将它标注为中等难度…我的思路首先将输入的列表按从大到小排序,然后对列表求和用变量保存,遍历列表,取出列表第一个元素和剩余元素和相比,如果满足,则返回遍历次数加1(遍历一次相当于取出一个元素,遍历两次相当于取出两个元素,因此遍历次数就是满足条件的元素数量最小值,但是遍历是从0开始遍历的,所以最终遍历次数需加1),如果不满足就继续遍历,知道条件满足。
废话少说,上代码:
def minElements(arr):
num = 0 # 用于保存列表和
d = 0 # 用于保存取出元素的和
arr.sort(reverse=True) # 对列表排序
for i in arr:
num += i # 列表求和
# 遍历列表
for i in range(len(arr)):
d += arr[i]
if d > num - d:
return i + 1
data = [3, 1, 7, 1]
print(minElements(data))
data = [2, 1, 2]
print(minElements(data))
因此该程序只是实现了功能,算法还有巨大的优化空间。评论区交给各位大佬!!!