題目

來源于 PythonTip 。

            
              6 的因子有 1, 2, 3 和 6, 它們的平方和是 1 + 4 + 9 + 36 = 50. 如果 f(N) 代表正整數 N 所有因子的平方和, 那么 f(6) = 50.
現在令 F 代表 f 的求和函數, 亦即 F(N) = f(1) + f(2) + .. + f(N), 顯然 F 一開始的 6 個值是: 1, 6, 16, 37, 63 和 113.
那么對于任意給定的整數 N (1 <= N <= 10^8), 輸出 F(N) 的值.

            
          

解析

根據題目要求一步一步來，可以實現該功能，但是考慮到實際 N 值的大小，程序時間復雜度會變得極大，因此需要從代碼層面進行優化。

在優化之前首先需要了解一下平方和的計算， 計算1到100的平方的和

平方求和公式：

代碼實現：

            
              def sumsqr(n):
    return int(n * (n + 1) * (2 * n + 1) / 6)
print(sumsqr(100))

            
          

接著分析本題目，在原始方法不可取的情況下，對數據進行規律分析。

列出從 1 到 N 的因子列表。

            
              N = 6
def get_factors(n):
    dp = []
    for i in range(1, n + 1):
        if n % i == 0:
            dp.append(i)
    return dp


def cal_sums():
    global N
    dp = []
    for i in range(1, N + 1):
        dp.append(get_factors(i))
    return dp

            
          

輸出結果為：

            
              [[1], [1, 2], [1, 3], [1, 2, 4], [1, 5], [1, 2, 3, 6]]

            
          

發現 1 有 6 個，2 有 3 個，3 有 2 個。。。替換 N 值，依然可以發現此現象。
總結 F(N)==1^2*(N//1)+2^2*(N//2)+...+N^2*(N//N)

得到改進版如下：

            
              N = 6
s = 0
for i in range(1,N+1):
	s = s + i**2*(N//i)
print (s)

            
          

但是時耗依然很大，對于每個數平方需要乘積的次數 N//i 進行分析。

            
              N = 10
L1 = list(range(1,N+1))
L2 = [N//i for i in L1]
print (L1)
print (L2)

            
          

結果為：

            
              [1, 2, 3, 4, 5, 6]
[6, 3, 2, 1, 1, 1]

            
          

測試發現后半部分的個數都是 1，所以對之前的程序進行修改。

            
              N = 6
s = 0
for i in range(1,N//2+1):
	s = s + i**2*(N//i)
 
def sumsqr(n):
	return int(n*(n+1)*(2*n+1)/6)
s = s + (sumsqr(N) - sumsqr(N//2))
 
print (s)

            
          

這里運用到了平方和求差公式，我們需要計算 6**2*1+5**2*1+4**2*1 ，即可簡化為 sumsqr(6)-sumsqr(3) 。
提交程序后，運行時耗依然很大，不通過。那就需要對循環再次進行簡化，也就是對循環次數 N//2 進行壓縮，那么我們再看一下上述平方和乘積次數。

            
              [1, 2, 3, 4, 5, 6]
[6, 3, 2, 1, 1, 1]

            
          

和 N 相關的數值，除了 N//2 平均數，那就只有 sqrt(N) 平方根，這兩個是比較常見的數值。
L2 中的 1 一直到最后，2 到第三項，恰好 sqrt(6) 為 2 代表第二項，所以第三項之后的可以用平方差進行求和計算。 3**2*2+4**2*1+5**2*1+6**2*1 = （3**2）+ （3**2+4**2+5**2+6**2） ，再進一步轉換為 F(6)=1+2**2*3+sumsqr(6)-sumsqr(2)+sumsqr(3)-sumsqr(2) ，進而得到以下代碼：

            
              def sumsqr(n):
    return int(n * (n + 1) * (2 * n + 1) / 6)

def factors_sums():
    N = 6
    s = 0
    m = int(sqrt(N))
    i = 1
    while i <= m:
        s += pow(i,2)*(N//i)
        s += sumsqr(N//i) - sumsqr(m)
        i += 1
    return s

            
          

最后提交代碼成功，時間復雜度也是最低的。