AtCoder Beginner Contest 217 G Groups （排列组合）

最新推荐文章于 2023-01-16 15:28:18 发布

莫说啥

最新推荐文章于 2023-01-16 15:28:18 发布

阅读量248

点赞数

分类专栏：算法题解文章标签：组合计数模运算分组问题快速幂算法优化

本文链接：https://blog.csdn.net/m0_46527476/article/details/120117507

版权

算法题解专栏收录该内容

1 篇文章 0 订阅

订阅专栏

原题链接

题意

给你 $N$ 个整数，标号为 $1$ ~ $N$
要求分成无序非空的 $k$ 组
对 $M$ 取模相同的数字不能分在同一组
对 $\le k \le n$ 都求一次答案
$\le N \le 5000,2 \le M \le N$

思路

根据整数对 $M$ 的取模值会分成 $M$ 种数，这 $M$ 种数的个数最多只会有两种 $\left \lfloor \frac{n}{m}\right \rfloor$ 和 $\left \lfloor \frac{n}{m}\right \rfloor+1$
直接考虑非常困难，我们先考虑分成有序可为空的 $k$ 组的答案
设第 $i$ 种数的个数为 $cnt_i$ ，放置第 $i$ 种数的方案数即为 $A_k^{cnt_i}$
因此 $f_k=\prod_{i=0}^{i=M-1}A_k^{cnt_i}$
然后考虑有序非空 $g_k=f_k-\sum_{i=0}^{i=k-1}C_k^i*f_i$ (把非空组数不到 $k$ 的减掉)
最后在考虑本题，只需要将有序的方案数转换成无序，答案其实就是 $\frac{g_k}{k!}$

代码

时间复杂度 $O(n^2)$

#include <bits/stdc++.h>

using namespace std;




#define ls o<<1
#define rs o<<1|1
#define fi first
#define se second
typedef long long ll;
typedef pair<int,int> pii;
const int maxn=2e5+10;
const int mod=998244353;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3f;
const double PI=acos(-1.0);


ll g[maxn],f[maxn];
ll fac[maxn],inv[maxn];
ll cal1(ll n,ll m){
    if(m>n) return 1;
    return fac[n]*inv[n-m]%mod;
}
ll kpow(ll a,ll k){
    if(!a) return a;
    ll res=1;
    while(k){
        if(k&1) res=res*a%mod;
        a=a*a%mod;
        k>>=1;
    }
    return res;
}
void init(){
    fac[0]=inv[0]=1;
    for(int i=1;i<=5000;i++){
        fac[i]=fac[i-1]*i%mod;
        inv[i]=kpow(fac[i],mod-2);
    }
}
ll cal2(ll n,ll m){
    return fac[n]*inv[n-m]%mod*inv[m]%mod;
}
void solve(){
    int n,m;
    init();
    scanf("%d%d",&n,&m);
    ll cnt1=m-n%m,cnt2=n%m,index1=n/m,index2=n/m+1;
    int mi=(n+m-1)/m;
    for(int i=1;i<=n;i++){
        if(i<mi) {
            f[i]=0;puts("0");
            continue;
        }
        f[i]=kpow(cal1(i,index1),cnt1)*kpow(cal1(i,index2),cnt2)%mod;
        for(int j=1;j<=i-1;j++){
            f[i]=(f[i]-cal2(i,j)*f[j]%mod+mod)%mod;
        }
        if(i)
        printf("%lld\n",f[i]*inv[i]%mod);
    }

}
int main(){
    // clock_t t1=clock();
	#ifndef ONLINE_JUDGE
       freopen("in.in", "r", stdin);
       freopen("out.out","w",stdout);
    #endif 
	// int T;scanf("%d",&T);while(T--)
	solve();
// end:
    // cerr<<"Time used "<<clock()-t1<<" ms"<<endl;
	return 0;
}