题目大意: 有一个长度为 K K K 的严格单调上升序列,最大值不超过 N N N,相邻两项的差不超过 M M M,问这个序列有多少种方案。
题解
显然需要考虑对这个序列进行差分,差分数组设为 b b b,那么需要满足: ∑ i = 1 K − 1 b i ≤ N \sum_{i=1}^{K-1} b_i\leq N ∑i=1K−1bi≤N。
考虑一个差分数组对应多少个原序列,若 ∑ i = 1 K − 1 b i = c \sum_{i=1}^{K-1}b_i=c ∑i=1K−1bi=c,那么序列第一位就有 N − c N-c N−c 种取值,所以方案数为 N − c N-c N−c。
答案为:
∑
b
1
∑
b
2
.
.
.
∑
b
K
−
1
(
N
−
∑
i
=
1
K
−
1
b
i
)
=
N
×
M
K
−
1
−
∑
i
=
1
K
−
1
∑
b
1
.
.
.
∑
b
K
−
1
b
i
=
N
×
M
K
−
1
−
∑
i
=
1
K
−
1
∑
b
i
=
1
M
b
i
×
M
K
−
2
=
N
×
M
K
−
1
−
(
K
−
1
)
(
1
+
M
)
M
2
M
K
−
2
\begin{aligned} \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}^Mb_i\times M^{K-2}\\ &=N\times M^{K-1}-(K-1)\frac {(1+M)M} 2M^{K-2}\\ \end{aligned}
b1∑b2∑...bK−1∑(N−i=1∑K−1bi)=N×MK−1−i=1∑K−1b1∑...bK−1∑bi=N×MK−1−i=1∑K−1bi=1∑Mbi×MK−2=N×MK−1−(K−1)2(1+M)MMK−2
代码如下:
#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);
}