HNOI 2013 数列 题解

题目传送门

题目大意： 有一个长度为 $K$ 的严格单调上升序列，最大值不超过 $N$，相邻两项的差不超过 $M$，问这个序列有多少种方案。

题解

显然需要考虑对这个序列进行差分，差分数组设为 $b$，那么需要满足：${\sum }_{i=1}^{K-1}{b}_i\le N$。

考虑一个差分数组对应多少个原序列，若 ${\sum }_{i=1}^{K-1}{b}_i=c$，那么序列第一位就有 $N-c$ 种取值，所以方案数为 $N-c$。

答案为：
$$\begin{array}{rl}\sum _{b_1}\sum _{b_2}...\sum _{b_{K-1}}(N-\sum _{i=1}^{K-1}{b}_i)& =N\times M^{K-1}-\sum _{i=1}^{K-1}\sum _{b_1}...\sum _{b_{K-1}}{b}_i\\ & =N\times M^{K-1}-\sum _{i=1}^{K-1}\sum _{b_i=1}^M{b}_i\times M^{K-2}\\ & =N\times M^{K-1}-(K-1)\frac{(1+M)M}{2}{M}^{K-2}\end{array}$$

代码如下：

#include <cstdio>

long long n;
int m,k,mod;
int ksm(int x,int y){int re=1;for(;(y&1?re=1ll*re*x%mod:0),y;y>>=1,x=1ll*x*x%mod);return re;}

int main()
{
	scanf("%lld %d %d %d",&n,&k,&m,&mod);
	int ans=n%mod*ksm(m,k-1)%mod;
	ans=(ans-1ll*(k-1)*(1ll*(1+m)*m/2%mod)%mod*ksm(m,k-2)%mod+mod)%mod;
	printf("%d",ans);
}