18 142
297 328
388 412
1998 2002
这道题目假设用C++能够直接枚举,非常快就能够过,并且时间,可是这样对我们学习数论知识没有一点帮助。由于数论不仅仅是简单的枚举很多其它的是公式的推导,所以我对于数论题目尽可能的使用耗时长一点的语言。来让我将代码变得更加简短,高速,比方这道题目。用一种方法python超时,可是c++46ms就能够过了,可是假设我用python将这道题目过了,用c++直接就是0ms。
我使用了一个公式推导式针对開始的前后两个数之差进行枚举计算
m = math.sqrt(float(2 * n) + pow(a * 0.5,2.0)) - a * 0.5
if m == int(m):
print i + 1,i + int(m)
这个会超时,原因是,无论这个数符不符合条件,你都要进行这个式子的运算
会导致这种结果,最后一个数据会超时:
如此进行代码优化:
对于等差数列公式得:(2a + m)(m + 1) = 2n -> 2a(m + 1) = 2n - m(m + 1) - > 2a = 2n / (k + 1) - m
又由于a为整数所以。2n % (k + 1)不为零的直接排除,接着是(2n / (k + 1) - m) % 2不为零的能够排除
这样非常多情况仅仅要推断一下就能够了,根本不须要进行什么计算。复杂度自然会降低非常多
接着就是答案输出了
这里提供pythonAC代码:
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import math
n = int(raw_input())
cnt = int(math.sqrt(2 * n))
i = cnt
while cnt > 0:
if not ((2 * n) % (cnt + 1)):
m = 2 * n / (cnt + 1)
m -= cnt
m >>= 1
if (2 * m + cnt) * (cnt + 1) / 2 == n and m >= 0:
print m,m + cnt
cnt -= 1