码迷,mamicode.com
首页 > 其他好文 > 详细

整数拆分

时间:2018-06-20 00:49:46      阅读:192      评论:0      收藏:0      [点我收藏+]

标签:.net   pair   open   csdn   oid   push   str   TE   inf   

技术分享图片

知识点:母函数(其实更建议去看《组合数学》),五边形数定理(百度总结的也不错)。

技术分享图片

技术分享图片
#include<bits/stdc++.h>
#define ll long long
#define P pair<int,int>
#define pb push_back
#define lson root << 1
#define INF (int)2e9 + 7
#define maxn (int)1e5 + 7
#define rson root << 1 | 1
#define LINF (unsigned long long int)1e18
#define mem(arry, in) memset(arry, in, sizeof(arry))
using namespace std;

const int mod = (int)1e9 + 7;

int T, n, k;
int dp[maxn];

void Inite(){
    mem(dp, 0);
    dp[0] = 1;
    for(int i = 1; i <= 100000; i++){
        for(int j = 1, r = 1; i - (3 * j * j - j) / 2 >= 0; j++, r *= -1){
            dp[i] += dp[i - (3 * j * j - j) / 2] * r;
            dp[i] %= mod;
            dp[i] =(dp[i] + mod) % mod;
            if(i - (3 * j * j + j) / 2 >= 0){
                dp[i] += dp[i - (3 * j * j + j) / 2] * r;
                dp[i] %= mod;
                dp[i] =(dp[i] + mod) % mod;
            }
        }
    }
}

int main()
{
    Inite();
    cin >> T;
    while(T--){
        cin >> n >> k;
        cout << (dp[n] - dp[n - k] + mod) % mod << endl;
    }
    return 0;
}
View Code

 

整数拆分

标签:.net   pair   open   csdn   oid   push   str   TE   inf   

原文地址:https://www.cnblogs.com/zgglj-com/p/9201784.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!