用五边形数解决整数拆分问题

最新推荐文章于 2021-11-02 19:16:32 发布

*ACoder*

最新推荐文章于 2021-11-02 19:16:32 发布

阅读量298

点赞数

本文链接：https://blog.csdn.net/FSAHFGSADHSAKNDAS/article/details/105929556

版权

典型例题：洛谷P6189

点击跳转

五边形数定理

$\prod_{k=1}^\infin (1-x^k) = \prod _{i=-\infin} ^ \infin (-1)^i x ^{\frac{3i^2-i}{2}}$

整数拆分

生成函数写出来：

$\prod_{k=1}^\infin \frac{1}{1-x^k}$

假设这个多项式是 $a_0x^0 + a_1x^1 + a_2x^2 + a_3x^3 + ...$

现在二者结合

很显然
$(a_0x^0 + a_1x^1 + a_2x^2 + a_3x^3 + ...) \prod_{k=1}^\infin (1-x^k) = 1$

因此
$(\prod _{i=-\infin} ^ \infin (-1)^i x ^{\frac{3i^2-i}{2}} ) (a_0x^0 + a_1x^1 + a_2x^2 + a_3x^3 + ...) = 1$

对比等式左右，就知道 $a_0=1$ ，而后面的项如果递推求的话， $i$ 次方项我可以在 $O(\sqrt i)$ 时间内求出

若要求第 $n$ 项，则

最终的时间复杂度是
$\sum_{i=0}^n i^{\frac{1}{2}} ) = O( \int_0^{n} x^{\frac{1}{2}} ) = O( n^{\frac{3}{2}} )$

代码

ll i, j;
cin >> n >> p;
a[0]=1;
rep(i,1,n)
{
    for(j=1;(3*j*j-j)/2<=i;j++)( j&1 ? a[i]+=a[i-(3*j*j-j)/2] : a[i]-=a[i-(3*j*j-j)/2] );
    for(j=-1;(3*j*j-j)/2<=i;j--)( j&1 ? a[i]+=a[i-(3*j*j-j)/2] : a[i]-=a[i-(3*j*j-j)/2] );
    a[i]=(a[i]%p+p)%p;
}
cout << a[n] << endl;