题目
【求解连续数列】已知连续正整数数列{K}=K1,K2,K3...Ki的各个数之和为S,i=N(0<S<100000,0<N<100000),求此数列K。
输入描述:
输入包含两个参数,1)连续正整数数列和S,2)数列里数的个数N。
输出描述:
如果有解输出数列K,如果无解输出-1
示例1:
输入:
525 6
输出:
85 86 87 88 89 90
分析
连续数列公式
S = (( K1 + Kn ) * n ) / 2
由此可以推出
K1 = 2S / n - Kn
由于Kn = K1 + n - 1
所以K1 = ( 2S/n + 1 - n )/2
故 2S/n - n必须为奇数
代码实现
def get_list(s,n):
a1 = 1
listk=[]
if (2*s %n) != 0:
print "-1"
return
elif (2*s/n-n)%2 == 0:
print "-1"
return
else:
a1 = (2*s/n+1-n)/2
i=0
for i in range(n):
listk.append(a1+i)
print (" ".join(str(j) for j in listk))
print listk
str0 = raw_input()
s,n = (int(var) for var in str0.split(" "))
print s,n
get_list(s,n)
结果输出,达到预期效果
参考链接