程序笔试 [编程题] 翻转数列

string=input()
n=int(string.split(' ')[0])
m=int(string.split(' ')[-1:][0])
count=0
"""zheng = fu =i=1
count=0
while i <= n:
    if fu <= m:
        count += (-1)*i
        i+=1
        fu+=1
    else :
        if zheng <= m:
            count += i
            i+=1
            zheng+=1
        else:
            zheng=fu=1"""
count=int((n/(m*2))*(m*m))
print(count)

 

最开始使用注释里的代码 运行正确但是时间过长只完成了30% 后来找规律进行编码 百分百通过了

但是  我很怀疑这样写真的ok么