Description
求 n n 的本质不同拆分数,要求每个数字出现次数小于
Input
第一行一整数 T T 表示用例组数,每组用例输入两个整数
Output
对于每组用例,输出保证每个数字出现次数小于 k k 的的本质不同拆分数,结果模 109+7 10 9 + 7
Sample Input
4
4 2
4 3
4 4
4 5
Sample Output
2
4
4
5
Solution
考虑划分中 j j 的数量的生成函数
设 pk(n) p k ( n ) 为 n n 的本质不同划分中每种数字出现次数小于的方案数,固定 k k 后考虑的生成函数:
Fk(x)=∏j≥1Pj(x)=∏j≥1(1−xkj)∏j≥1(1−xj)=Q(xk)Q(x)=Q(xk)P(x) F k ( x ) = ∏ j ≥ 1 P j ( x ) = ∏ j ≥ 1 ( 1 − x k j ) ∏ j ≥ 1 ( 1 − x j ) = Q ( x k ) Q ( x ) = Q ( x k ) P ( x )
即 Fk(x)=(1−xk−x2k+x5k+x7k−...)(1+p(1)⋅x+p(2)⋅x2+...) F k ( x ) = ( 1 − x k − x 2 k + x 5 k + x 7 k − . . . ) ( 1 + p ( 1 ) ⋅ x + p ( 2 ) ⋅ x 2 + . . . )
考虑等式两边 xn x n 的系数有 pk(n)=p(n)−p(n−k)−p(n−2k)+p(n−5k)+p(n−7k)−... p k ( n ) = p ( n ) − p ( n − k ) − p ( n − 2 k ) + p ( n − 5 k ) + p ( n − 7 k ) − . . .
预处理出 p(1) p ( 1 ) ~ p(n) p ( n ) 后枚举广义五边形数即可求出 pk(n) p k ( n ) ,时间复杂度 O(nn−−√+Tn−−√) O ( n n + T n )
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100005;
#define mod 1000000007
void inc(int &x,int y)
{
x=x+y>=mod?x+y-mod:x+y;
}
void dec(int &x,int y)
{
x=x-y<0?x-y+mod:x-y;
}
int T,n,k,f[maxn];
int main()
{
f[0]=1;
for(int i=1;i<=100000;i++)
{
for(int j=1,w=1;w<=i;w+=3*j+1,j++)
if(j&1)inc(f[i],f[i-w]);
else dec(f[i],f[i-w]);
for(int j=1,w=2;w<=i;w+=3*j+2,j++)
if(j&1)inc(f[i],f[i-w]);
else dec(f[i],f[i-w]);
}
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&k);
int ans=f[n];
for(int j=1,w=1;w<=n/k;w+=3*j+1,j++)
if(j&1)dec(ans,f[n-w*k]);
else inc(ans,f[n-w*k]);
for(int j=1,w=2;w<=n/k;w+=3*j+2,j++)
if(j&1)dec(ans,f[n-w*k]);
else inc(ans,f[n-w*k]);
printf("%d\n",ans);
}
return 0;
}