题解
生成函数写出来:
1 1 − x 2 ( 1 + x ) ( 1 + x + x 2 ) ( x 1 1 − x 2 ) ( 1 1 − x 4 ) ( 1 + x + x 2 + x 3 ) ( 1 + x ) ( 1 1 − x 3 ) \frac{1}{1-x^2}(1+x)(1+x+x^2)(x\frac{1}{1-x^2})(\frac{1}{1-x^4})(1+x+x^2+x^3)(1+x)(\frac{1}{1-x^3}) 1−x21(1+x)(1+x+x2)(x1−x21)(1−x41)(1+x+x2+x3)(1+x)(1−x31)
然后根据 ( 1 − x ) ( 1 + x + x 2 ) = ( 1 − x 3 ) (1-x)(1+x+x^2)=(1-x^3) (1−x)(1+x+x2)=(1−x3), ( 1 − x ) ( 1 + x + x 2 + x 3 ) = ( 1 − x 4 ) (1-x)(1+x+x^2+x^3)=(1-x^4) (1−x)(1+x+x2+x3)=(1−x4),带进去化简,就得到:
x 1 − x 4 \frac{x}{1-x^4} 1−x4x
然后展开
x ∑ k = 0 ∞ ( − 1 ) k ( − 4 k ) x k + 1 x\sum_{k=0}^{\infin} (-1)^k \binom{-4}{k} x^{k+1} xk=0∑∞(−1)k(k−4)xk+1
最后得到答案为
n ( n + 1 ) ( n + 2 ) 6 \frac{n(n+1)(n+2)}{6} 6n(n+1)(n+2)
代码
n = int(input())
print((n*(n+1)*(n+2)//6)%10007)