做leetcode1712练习题如下:
我们称一个分割整数数组的方案是 好的 ,当它满足:
数组被分成三个 非空 连续子数组,从左至右分别命名为 left , mid , right 。
left 中元素和小于等于 mid 中元素和,mid 中元素和小于等于 right 中元素和。
给你一个 非负 整数数组 nums ,请你返回 好的 分割 nums 方案数目。由于答案可能会很大,请你将结果对 109 + 7 取余后返回。
示例 1:
输入:nums = [1,1,1]
输出:1
解释:唯一一种好的分割方案是将 nums 分成 [1] [1] [1] 。
示例 2:
输入:nums = [1,2,2,2,5,0]
输出:3
解释:nums 总共有 3 种好的分割方案:
[1] [2] [2,2,5,0]
[1] [2,2] [2,5,0]
[1,2] [2,2] [5,0]
import bisect
def waysToSplit(nums) -> int:
cnt = [nums[0]]
n = len(nums)
ans = 0
for i in range(1, n):
cnt.append(cnt[-1] + nums[i])
print(cnt)
for i in range(n - 1):
m = cnt[-1] - cnt[i]
if 2 * m >= cnt[i]:
x = bisect.bisect_right(cnt, cnt[i] // 2)
y = bisect.bisect_right(cnt, cnt[i] - m)
ans += x - y
return ans
最后单位ans总是不能完全符合全部例子的答案
将数列累加形成新的cnt递增数列,然后打算用二分查找计算可能得种类
尝试过改变二分查找的right和left,或者改变边界值
帮忙看一下哪里出了问题
你好,我是有问必答小助手,非常抱歉,本次您提出的有问必答问题,技术专家团超时未为您做出解答
本次提问扣除的有问必答次数,已经为您补发到账户,我们后续会持续优化,扩大我们的服务范围,为您带来更好地服务。