整数拆分

最新推荐文章于 2019-03-02 22:51:00 发布

deprecated_tzg

最新推荐文章于 2019-03-02 22:51:00 发布

阅读量636

点赞数

分类专栏：数论文章标签：五边形数生成函数

本文链接：https://blog.csdn.net/solotzg/article/details/37762163

版权

数论专栏收录该内容

19 篇文章 0 订阅

订阅专栏

hdu 1085 Holding Bin-Laden Captive!

hdu 1398 Square Coins

入门级基础题

hdu 4651 Partition

这该算是数论中的经典题了。网上关于这题的解法大同小异，都不外乎利用生成函数（母函数）和五边形定理。

作为初学者算是费了不少精力才搞懂。

关于生成函数的推导是重点，本题中：

∑p(n)x^n= (1+x^1+x^2+x^3...)(1+x^2+x^4+x^6...)(1+x^3+x^6...)

等比数列求和得 ∑p(n)x^n= (1+x^1+x^2+x^3...)(1+x^2+x^4+x^6...)(1+x^3+x^6...) =

(1-x^(+∞/1 + 1))(1-x^(+∞/2 + 1))... / (1-x)(1-x^2)(1-x^n)

∵ |x|<1

∴ ∑p(n)x^n = 1 / (1-x)(1-x^2)(1-x^n)

结合五边形定理∑p(n)x^n = 1 / (1-x^1-x^2+x^5+x^7...)

即：(p(0) + p(1)x^1 + p(2)x^2 ... )(1-x^1-x^2+x^5+x^7...) = 1

∴ p(n)x^n + p(n-1)x^(n-1)*(-x^1) + p(n-2)x^(n-2)*(-x^2) + p(n-5)x^(n-5)*(x^5)... = 0

∴ p(n) = p(n-1) + p(n-2) - p(n-5) - p(n-7)...
#include<cstdio>
#include<algorithm>
#include<vector>
#include<cmath>
#include<queue>
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
const int MAXN = 100005;
typedef __int64 ll;
ll MOD = 1000000007;
ll num[MAXN];
void init()
{
    num[0] = num[1] = 1;
    for (int i = 2; i<= 100000; ++i)
    {
        ll flg = 1;
        for (int k = 1; ;++k)
        {
            int a = i - (3*k-1)*k/2, b = i - (3*k+1)*k/2;
            ll s1, s2;
            if (a < 0) break;
            s1 = num[a];
            if (b < 0) s2 = 0;
            else s2 = num[b];
            num[i] = (num[i]+flg*(s1+s2))%MOD;
            flg = 0-flg;
        }
        num[i] = (num[i]+MOD)%MOD;
    }
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
#endif
    int t;
    scanf("%d", &t);
    init();
    while(t--)
    {
        int a;
        scanf("%d", &a);
        printf("%I64d\n", num[a]);
    }
    return 0;
}

hdu 4658 Integer Partition

该题是上题的加强版，而且需要用到上题中推出的公式

具体可参见 http://blog.csdn.net/zhoufenqin/article/details/9821617 ，写的非常详细

#include<cstdio>
#include<algorithm>
#include<vector>
#include<cmath>
#include<queue>
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
const int MAXN = 100005;
typedef __int64 ll;
ll MOD = 1000000007;
ll num[MAXN];
void init()
{
	num[0] = num[1] = 1;
	for (int i = 2; i<= 100000; ++i)
	{
		ll flg = 1;
		for (int k = 1; ;++k)
		{
			int a = i - (3*k-1)*k/2, b = i - (3*k+1)*k/2;
			ll s1, s2;
			if (a < 0) break;
			s1 = num[a];
			if (b < 0) s2 = 0;
			else s2 = num[b];
			num[i] = (num[i]+flg*(s1+s2))%MOD;
			flg = 0-flg;
		}
		num[i] = (num[i]+MOD)%MOD;
	}
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
#endif
	int t;
	scanf("%d", &t);
	init();
	while(t--)
	{
		int a, k;
		scanf("%d%d", &a, &k);
		ll res = num[a];
		for (int j = 1, f = -1; ; ++j, f=0-f)
		{
			int c = (3*j-1)*j/2*k, d = (3*j+1)*j/2*k;
			ll s1, s2;
			if (c > a) break;
			s1 = f*num[a-c];
			if (d > a) s2 = 0;
			else s2 = f*num[a-d];
			res = (res + s1 + s2)%MOD;
		}
		res = (res+MOD)%MOD;
		printf("%I64d\n", res);
	}
    return 0;
}