组合数学(隔板法) + 高精度 - 方程的解 - AcWing 1308

组合数学(隔板法) + 高精度 - 方程的解 - AcWing 1308

$佳佳碰到了一个难题，请你来帮忙解决。$

$对于不定方程a_1+a_2+\cdots +a_{k-1}+a_k=g(x)，其中k\ge 1且k\in N\ast ，x是正整数，$

$g(x)=x^{x}mod1000（即x^x除以1000的余数），x,k是给定的数。$

$我们要求的是这个不定方程的正整数解组数。$

$举例来说，当k=3,x=2时，方程的解分别为：$

$\{\begin{array}{l}{a}_1=1\\ a_2=1\\ a_3=2\end{array}\{\begin{array}{l}{a}_1=1\\ a_2=2\\ a_3=1\end{array}\{\begin{array}{l}{a}_1=2\\ a_2=1\\ a_3=1\end{array}$

输入格式

$有且只有一行，为用空格隔开的两个正整数，依次为k,x。$

输出格式

$有且只有一行，为方程的正整数解组数。$

数据范围

$1\le k\le 100,1\le x<2^{31},k\le g(x)$

输入样例：

3 2

输出样例：

3

---

分析：

隔板法：

$我么可以将g(x)=x^{x}mod1000看作n个点，$

$接着我们用k-1个隔板将n个点分成k个部分(每个部分点的数量均大于0)，$

$这样，每个部分点的数量就对应一个正整数a_i，每一种分法就对应一组可行解。$

$故方案总数为C_{n-1}^{k-1}。$

$方案总数极限情况下可能会达到C_{999}^{99}左右。$

$递推求C_n^m的时间复杂度为O(nm)可行，但需要手写一个高精度。$

代码：

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

const int N=150, mod=1000;

int k,x;    
int f[1000][100][N];

int quick_pow(int a,int b,int mod)
{
    int res=1;
    while(b)
    {
        if(b&1) res=res*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return res;
}

void add(int c[],int a[],int b[])
{
    for(int i=0,t=0;i<N;i++)
    {
        t+=a[i]+b[i];
        c[i]=t%10;
        t/=10;
    }
}

int main()
{
    cin>>k>>x;
    
    int n=quick_pow(x%1000,x,mod);  
    
    for(int i=0;i<n;i++)        //C[n-1][k-1]
        for(int j=0;j<=i && j<k;j++)
            if(!j) f[i][j][0]=1;
            else add(f[i][j],f[i-1][j],f[i-1][j-1]);
            
    int *ans=f[n-1][k-1];
    int i=N-1;
    while(!ans[i]) i--;
    while(i>=0) cout<<ans[i--];
    
    return 0;
}