【题目】小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
【思路】下面第一个代码未能通过所有测试用例,因为没对B 进行排序,我自己想的,越写越复杂就不想写了。思路不建议看,写的乱七八糟的,就是,先判断有几个正数相加可以得到这个数。
如果需要正数个数是偶数个M,就判断tsum对M的余数是不是M/2,若是,这个数就是tsum/M前M/2个,和后M/2个,例如7可以被两个数得出,7%2=1等于2/2,所以所需正数就是7/2的前一个(3)和后一个(4)。
如果需要的正数的数目是奇数N,就要判断tsum能不能整除N,若能,就是整除后的前N//2个,整除的结果,后N//2个,例如100由5个正数相加,100/5=20,20的前5//2个(18,19)(20),后5//2个(21,22)。
【代码】
# -*- coding:utf-8
class Solution:
def FindContinuousSequence(self, tsum):
# write code here
B = []
if tsum < 3:
return []
for i in range(2,tsum):
if i % 2 != 0:
if tsum % i==0:
A = []
a = tsum//i
b = a-i//2
if b>0:
for n in range(0,i):
A.append(b+n)
B.append(A)
else :
if tsum % i == i//2:
C = []
a = tsum//i
b=a-i//2+1
if b>0:
for n in range(0,i):
C.append(b+n)
B.append(C)
return B
另一种思路是用滑动窗口的思想,首先定义窗口两端,然后对窗口内的元素求和,因为窗口内是等差数列,用求和公式即可,窗口内值得和等于tsum时,将窗口内的值添加到列表中,左边窗口望前移一格,如果窗口内的值大于tsum,左边窗口移动一步,小于tsum,右边窗口移动一步,直到左边等于右边。
class Solution:
def FindContinuousSequence(self, tsum):
# write code here
a = 1
b = 2
B = []
while a < b :
sum = (a+b)*(b-a+1)/2
if tsum == sum:
A = []
for i in range(a,b+1):
A.append(i)
B.append(A)
a = a + 1
elif tsum < sum:
a = a + 1
else:
b = b + 1
return B