和为S的连续正数序列题目描述
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
输出描述:
输出所有和为S的连续正数序列。序列内按照从小至大的顺序,序列间按照开始数字从小到大的顺序。
思路:
看了大佬的解法,太巧妙了!
用两个指针,控制窗口的大小,将窗口内的值与要求值比较进行调整。
# -*- coding:utf-8 -*-
class Solution:
def FindContinuousSequence(self, tsum):
# 双指针法,一个指向最左边,一个指向最右边,当窗口内值小于100则将右指针+1,当窗口内值大于100则左指针+1
left=1
right=2
res=[]
while left<right:
nsum=(left+right)*(right-left+1)/2
if nsum==tsum:
lis=[]
for i in range(left,right+1):
lis.append(i)
res.append(lis)
left+=1#每得到一个结果后将左指针向右增加,继续向下查找
elif nsum<tsum:
right+=1
else:
left+=1
return res
还有另一个巧妙的思路也放在这里:
链接:https://www.nowcoder.com/questionTerminal/c451a3fd84b64cb19485dad758a55ebe?f=discussion
来源:牛客网
1)由于我们要找的是和为S的连续正数序列,因此这个序列是个公差为1的等差数列,而这个序列的中间值代表了平均值的大小。假设序列长度为n,那么这个序列的中间值可以通过(S / n)得到,知道序列的中间值和长度,也就不难求出这段序列了。
2)满足条件的n分两种情况:
n为奇数时,序列中间的数正好是序列的平均值,所以条件为:(n & 1) == 1 && sum % n == 0;
n为偶数时,序列中间两个数的平均值是序列的平均值,而这个平均值的小数部分为0.5,所以条件为:(sum % n) * 2 == n.
3)由题可知n >= 2,那么n的最大值是多少呢?我们完全可以将n从2到S全部遍历一次,但是大部分遍历是不必要的。为了让n尽可能大,我们让序列从1开始,
根据等差数列的求和公式:S = (1 + n) * n / 2,得到.
最后举一个例子,假设输入sum = 100,我们只需遍历n = 13~2的情况(按题意应从大到小遍历),n = 8时,得到序列[9, 10, 11, 12, 13, 14, 15, 16];n = 5时,得到序列[18, 19, 20, 21, 22]。
题目描述
输入一个递增排序的数组和一个数字S,在数组中查找两个数,使得他们的和正好是S,如果有多对数字的和等于S,输出两个数的乘积最小的。
输出描述:
对应每个测试案例,输出两个数,小的先输出。
利用上一题的思路,只是指针分别从左右开始向中间遍历。
值得注意的一点是要求输出两个数的乘积最小,可以设置最小值进行对比,但是这样需要遍历整个数组。我们可以知道,在递增序列中,最先找到的一组是最靠边的,乘积肯定比中间的值相乘要低,所以找到第一组就可以输出。
证明参照:
链接:https://www.nowcoder.com/questionTerminal/390da4f7a00f44bea7c2f3d19491311b?f=discussion
来源:牛客网
结论证明:
- 假设:找到两组满足条件的数组对(x,y)、(x+a,y-a),其中(x+y=S, 0<a<y-x)
- x*y-[(x+a)(y-a)]
- =xy-xy-(y-x)a+a2
- =a[a-(y-x)]
- 因为0<a<y-x,所以a-(y-x)<0,所以a[a-(y-x)]<0
- 因此(x,y)乘积一定比(x+a,y-a)小
# -*- coding:utf-8 -*-
class Solution:
def FindNumbersWithSum(self, array, tsum):
if not array and len(array)<2:
return []
left=0
right=len(array)-1
res=[]
while left<right:
if array[left]+array[right]==tsum:
res.append(array[left])
res.append(array[right])
return res
elif array[left]+array[right]<tsum:
left+=1
else:
right-=1
return res