洛谷P4345 [SHOI2015]超能粒子炮·改

最新推荐文章于 2021-02-20 22:27:49 发布

Cr.rech

最新推荐文章于 2021-02-20 22:27:49 发布

阅读量176

点赞数 1

分类专栏：做题笔记学习笔记文章标签：数论组合数学

本文链接：https://blog.csdn.net/bcr_233/article/details/95658178

版权

学习笔记同时被 2 个专栏收录

25 篇文章 0 订阅

订阅专栏

做题笔记

18 篇文章 0 订阅

订阅专栏

题目链接： $\rm [SHOI2015]超能粒子炮·改$
感谢 $\rm Venus$ 神仙帮助完成 $\LaTeX$ 公式

显然题目让我们求的是 $\rm \left[\sum\limits_{i=0}^k\binom ni\right]\%p$ 的值，其中 $\rm p=2333$
设 $\rm \operatorname{sum}(n,k)=\left[\sum\limits_{i=0}^k\binom ni\right]\%p$
然后我们开始推式子（大波 $\LaTeX$ 公式警告）：

$\begin{aligned} \rm \operatorname{sum}(n,k) & =\rm \left[ \sum_{i=0}^k\binom ni \right] \%p \quad\color{#66ccff}\textbf{（为保证界面美观，以下省略\%p）}\\ & =\rm \sum_{i=0}^k\binom{n/p}{i/p}\binom{n\%p}{i\%p} \color{#66ccff}\textbf{（卢卡斯定理）}\\ & =\rm \left[ \sum_{i=0}^{p-1}\binom{n/p}0\binom{n\%p}i\right] +\left[\sum_{i=0}^{p-1}\binom{n/p}1\binom{n\%p}i\right]+\cdots \\ & \rm \quad\ +\left[\sum_{i=0}^{p-1}\binom{n/p}{k/p-1} \binom{n\%p}i\right] +\left[\sum_{i=0}^{k\% p}\binom{n/p}{k/p}\binom{n\%p}i \right] \color{#66ccff}\textbf{（整除分块）}\\ & =\rm \left[\binom{n/p}0\sum_{i=0}^{p-1}\binom{n\%p}i\right] +\left[\binom{n/p}1\sum_{i=0}^{p-1}\binom{n\%p}i\right]+\cdots \\ & \rm\quad\ +\left[\binom{n/p}{k/p-1}\sum_{i=0}^{p-1}\binom{n\%p}i\right] +\left[\binom{n/p}{k/p}\sum_{i=0}^{k\%p}\binom{n\%p}i\right] \\ & =\rm \left[\sum_{i=0}^{p-1}\binom{n\%p}i\right] \times\left[\sum_{j=0}^{k/p-1}\binom{n/p}j\right] +\binom{n/p}{k/p}\sum_{i=0}^{k\%p}\binom{n\%p}i \end{aligned}$

再套用一下 $\operatorname{sum}$ 的定义就有：
$\rm \operatorname{sum}(n,k)= \left [\operatorname{sum}(n\%p,p-1) \times\operatorname{sum}(n/p,k/p-1) +\binom{n/p}{k/p}\operatorname{sum}(n\%p,k\%p)\right] \%p$

化简完毕。那么各部分如何处理呢？

$\begin{aligned} & \rm \operatorname{sum}(n\%p,p-1),\operatorname{sum}(n\%p,k\%p) 的参数都小于 p ，可以预处理出来； \\ & \rm 2333是质数，所以\binom{n/p}{k/p}显然可以用Lucas搞出来； \\ & \rm \operatorname{sum}(n/p,k/p-1)可以递归求解。 \end{aligned}$

代码

#include<cstdio>
#include<cctype>
#define getc (l==r&&(r=(l=f)+fread(f,1,1<<21,stdin),l==r)?EOF:*l++)
//fread优化快读
typedef long long ll;
const int p=2333;

char f[1<<21],*l=f,*r=f;
inline ll read() { //快读
	ll x=0; char ch=getc;
	while (!isdigit(ch)) ch=getc;
	while (isdigit(ch)) { x=x*10+(ch^48); ch=getc; }
	return x;
}

int c[p][p],s[p][p];
void work() {
	//杨辉三角预处理组合数
	for (int i=0; i<p; ++i) c[i][0]=c[i][i]=s[i][0]=1;
	for (int i=1; i<p; ++i)
		for (int j=1; j<i; ++j)
			c[i][j]=(c[i-1][j]+c[i-1][j-1])%p;
	//前缀和预处理
	for (int i=0; i<p; ++i)
		for (int j=1; j<p; ++j)
			s[i][j]=(s[i][j-1]+c[i][j])%p;
}
int C(ll n,ll m) {
	return m?C(n/p,m/p)*c[n%p][m%p]%p:1; //卢卡斯
}
int S(ll n,ll m) {
	if (!n||!m) return 1;
	if (n<p&&m<p) return s[n][m];
	int ans=(m>=p?s[n%p][p-1]*S(n/p,m/p-1)%p:0);
	(n/p>=m/p)&&(ans+=C(n/p,m/p)*s[n%p][m%p]%p);
	return ans%p;
}

int main() {
	work();
	int T=read();
	while (T--) {
		ll n=read(),m=read();
		printf("%d\n",S(n,m));
	}
}