首页 > 其他 > 详细

1259 整数划分 V2

时间:2017-11-12 21:23:14      阅读:256      评论:0      收藏:0      [点我收藏+]

设dp[n]为整数n的分割函数,由五边形定理得到:

dp[n] = dp[n-1] + dp[n-2] - dp[n-5] - dp[n-7]……

我们将其分为两部分计算

第一部分为 :( dp[n-1] - dp[n-5] …… ) 奇数项为加,偶数项为减,第j项括号内的值为 : n-(j*(3*j-1)/2)

第二部分为:(dp[n-2] - dp[n-7]……)  奇数项为加,偶数项为减,第j项括号内的值为 : n-(j*(3*j+1)/2)

如此递推便可求出n的分割函数 dp[n]

 

技术分享
#include <iostream>
#include <cmath>
#include <string.h>
#include <cstdio>
#include <queue>
#include <algorithm>
using namespace std;
#define LL long long
#define MAXSIZE 1000005
#define INF 1e11

const LL mod=1e9+7;

LL dp[MAXSIZE];

LL Solve(int n)
{
    dp[0] = 1;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j*(3*j-1)/2<=i;j++)
        {
            if(j&1)
                dp[i] = (dp[i] + dp[i-(j*(3*j-1)/2)] + mod)%mod;
            else
                dp[i] = (dp[i] - dp[i-(j*(3*j-1)/2)] + mod)%mod;
        }

        for(int j=1;j*(3*j+1)/2<=i;j++)
        {
            if(j&1)
                dp[i] = (dp[i] + dp[i-(j*(3*j+1)/2)] + mod)%mod;
            else
                dp[i] = (dp[i] - dp[i-(j*(3*j+1)/2)] + mod)%mod;
        }
    }
    return dp[n];
}

int main()
{
    int n;
    scanf("%d",&n);
    LL ans = Solve(n);
    printf("%lld\n",ans);
    return 0;
}
View Code

 

1259 整数划分 V2

原文:http://www.cnblogs.com/alan-W/p/7822783.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!