ZOJ 3557 - How Many Sets II（Lucas定理模板）

最新推荐文章于 2021-09-21 12:04:41 发布

SingleK

最新推荐文章于 2021-09-21 12:04:41 发布

阅读量156

点赞数

分类专栏：数学-----------组合数学文章标签：组合数学

本文链接：https://blog.csdn.net/xiao_k666/article/details/83151712

版权

数学-----------组合数学专栏收录该内容

13 篇文章 0 订阅

订阅专栏

该博客主要探讨了如何利用Lucas定理解决ZOJ 3557题，即计算在不取相邻小球的情况下，从n个相同小球中取m个的方案数。博主给出了问题的思路，指出可以将剩余的小球插入到空位中，形成 Ans=C(n-m+1,m)，并分享了Lucas定理的模板，用于求解模p条件下的组合数。" 133214077,20037727,Open3D中八叉树的点云数据操作,"['点云', '数据结构', 'Open3D', '算法']

摘要由CSDN通过智能技术生成

题目链接 https://cn.vjudge.net/problem/ZOJ-3557

【题意】
从n个相同小球中取m个小球，不能取相邻的小球的方案数

【思路】
首先拿出 $m$ 个小球，还剩下 $n - m$ 个小球。这 $n - m$ 个小球一共有 $n - m + 1$ 个空（左右两边也可以），把这 $m$ 个小球插入到这 $n - m + 1$ 个空里就是答案，即 $Ans=C_{n-m+1}^{m}$ ，记录一下 $L u c a s$ 定理的模板

Lucas 定理

求解 $C_n^m \ \ (mod p)$
首先将n和m分解为p进制：
$n=n_kp^k+n_{k-1}p^{k-1}+...+n_1p+n_0$ $m=m_kp^k+m_{k-1}p^{k-1}+...+m_1p+m_0$ 那么 $Ans≡∏_{i=0}^{k}C_{n_i}^{m_i} \ \ (modp)$

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxn=10050;
int mod;

ll pw(ll x,ll n){
	ll ans=1LL;
	while(n){
		if(n&1) ans=ans*x%mod;
		x=x*x%mod;
		n>>=1;
	}
	return ans;
}

ll inv(ll a){return pw(a,mod-2);}

ll C(int n,int m){
    if(m>n) return 0LL;
    ll up=1LL,down=1LL;
    for(int i=n-m+1;i<=n;++i) up=up*i%mod;
    for(int i=1;i<=m;++i) down=down*i%mod;
    return up*inv(down)%mod;
}

ll Lucas(int n,int m){
	if(m>n) return 0LL;
    ll ans=1LL;
    for (;m;n/=mod,m/=mod)
        ans=ans*C(n%mod,m%mod)%mod;
    return ans;
}

int main(){
	int n,m;
	while(scanf("%d%d%d",&n,&m,&mod)==3){
		printf("%lld\n",Lucas(n-m+1,m));
	}
	return 0;
}