牛牛P50591 解的分配问题，高精度组合数

stdforces

于 2022-11-14 10:30:35 发布

阅读量204

点赞数

文章标签：算法 c++ 开发语言

本文链接：https://blog.csdn.net/stdforces/article/details/127842907

版权

题意：

给出不定方程
$a_{1}+a_{2}+...+a_{k}=x^{x}\ (mod\ 1000)$
求出有多少组正整数解

Solution：

解的分配问题考虑隔板法，两个隔板之间的小球数是一个解，每个隔板之前的到前一个隔板的小球数就是一个解，第一个小球前不用放置一个隔板，但最后一个小球后面一定需要放置一个隔板，那么只需要在剩下的 $x^{x}(mod\ 1000)-1$ 个位置中放置 $k - 1$ 个隔板，即
$C_{x^{x}(mod\ 1000)-1}^{k-1}$
这个需要高精度加法或乘法，切莫使用NTT做，自带取余，可以考虑FFT求高精度积，总而言之，高精度加是最优的

也可以考虑生成函数，每个解的生成函数为
$f(x')=\sum_{i=1}^{x^{x}(mod\ 1000)}x'^{i}$
对这个多项式求 $k$ 次幂即可，但数字太大，不如高精度，并且需要NTT，自带取余，会使得答案不准确

// #include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<ctime>
#include<vector>
#include<bitset>
#include<map>
using namespace std;

using ll=long long;
const int N=2e7,inf=0x3fffffff;
const long long INF=0x3f3f3f3f3f3f,mod=1e3;

ll qpow(ll a,ll b)
{
	ll ret=1,base=a;
	while(b)
	{
		if(b&1) ret=ret*base%mod;
		base=base*base%mod;
		b>>=1;
	}
	return ret;
}

vector<int>operator+(vector<int>val1,vector<int>val2)
{
    if(val1.size()<val2.size()) swap(val1,val2);
    reverse(val1.begin(),val1.end()); reverse(val2.begin(),val2.end());
    vector<int>ret(val1.size());
    for(int i=0;i<val2.size();i++) ret[i]=val1[i]+val2[i];
    for(int i=val2.size();i<val1.size();i++) ret[i]=val1[i];
    for(int i=0;i<ret.size();i++)
    {
        if(ret[i]>9)
        {
            if(i+1<ret.size()) ret[i+1]+=ret[i]/10;
            else ret.push_back(ret[i]/10);
            ret[i]%=10;
        }
    }
    reverse(ret.begin(),ret.end());
    return ret;
}

ll k,x;
vector<int>C[1005][105];

int main()
{
    #ifdef stdjudge
        freopen("in.txt","r",stdin);
    #endif 
    cin>>k>>x;
    ll tmp=qpow(x,x);
    C[0][0]={1};
    for(int i=1;i<=tmp;i++)
    {
        for(int j=0;j<=105;j++)
        {
            if(j==0||j==i) C[i][j]={1};
            else C[i][j]=C[i-1][j-1]+C[i-1][j];
        }
    }
    if(k>tmp) cout<<0;
    else
    {
        for(int i=0;i<C[tmp-1][k-1].size();i++) printf("%d",C[tmp-1][k-1][i]);
    }
    return 0;
}