[编程题] 翻转数列

时间限制：1秒

空间限制：32768K

小Q定义了一种数列称为翻转数列:
给定整数n和m, 满足n能被2m整除。对于一串连续递增整数数列1, 2, 3, 4..., 每隔m个符号翻转一次, 最初符号为'-';。
例如n = 8, m = 2, 数列就是: -1, -2, +3, +4, -5, -6, +7, +8.
而n = 4, m = 1, 数列就是: -1, +2, -3, + 4.
小Q现在希望你能帮他算算前n项和为多少。

输入描述:

输入包括两个整数n和m(2 <= n <= 109, 1 <= m), 并且满足n能被2m整除。

 

输出描述:

输出一个整数, 表示前n项和。

 

输入例子1:

8 2

 

输出例子1:

8

这是腾讯的一道题，只要用到列表空间复杂度一定会超，只能借助规律和生成器来完成。

input=list(map(int,input().split(" ")))
n1=input[0]
n2=input[1]
a=sum(x for x in range(1,n2+1))
b=sum(x for x in range(n2+1,1+(n2*2)))
res=b-a
l=n1//(n2*2)
print(res*l)