Description
给你四个数 \(N,K,M,P\) ，让你生成一段长度为 \(K\) 严格单调递增序列，并且满足：
- 第一位可以为任意元素；
- 相邻两位的差值不超过 \(M\) ；
- 序列中元素大小不超过 \(N\) 。
求满足上述要求不同的生成序列有多少个，对 \(P\) 取模。
\(1\leq N\leq 10^{18},1\leq K,M,P\leq 10^9\)
Solution
其实容易发现，我们可以生成长度为 \(K\) 以增量为元素的序列 \(A\) 。
等价的变成了：
- 第一位可以为任意元素；
- 其余元素 \(\in [1,M]\) ；
- \(\sum\limits_{i=1}^K A_i\leq N\) 。
我们先不考虑第一位元素的选取情况。其余 \(K-1\) 个元素选取情况为 \(M^{K-1}\) 。那么对于每种生成序列 \(A\) ，它第一位可以选的元素 \(\in\left[1,N-\sum\limits_{i=2}^KA_i\right]\) 。
容易发现，每种 \([2,K]\) 位生成序列第一位的情况有 \(N-\sum\limits_{i=2}^KA_i\) 种。记所有 \(2\sim K\) 位生成排列的集合为 \(\mathbb{S}\) 我们枚举生成的排列。那么答案为：\[\sum_{s\in\mathbb{S}}\left(N-\sum\limits_{i=2}^KA_i\right)\]
等价变形为： \[\begin{aligned}&NM^{K-1}-\sum_{s\in\mathbb{S}}\sum\limits_{i=2}^KA_i\\=&NM^{K-1}-(K-1)\frac{1+2+\cdots+M}{2}M^{K-2}\\=&NM^{K-1}-(K-1)\frac{(M+1)M}{2}M^{K-2}\end{aligned}\]
直接求解即可。
Code
//It is made by Awson on 2018.3.10
#include <bits/stdc++.h>
#define LL long long
#define dob complex<double>
#define Abs(a) ((a) < 0 ? (-(a)) : (a))
#define Max(a, b) ((a) > (b) ? (a) : (b))
#define Min(a, b) ((a) < (b) ? (a) : (b))
#define Swap(a, b) ((a) ^= (b), (b) ^= (a), (a) ^= (b))
#define writeln(x) (write(x), putchar('\n'))
#define lowbit(x) ((x)&(-(x)))
using namespace std;
void read(LL &x) {
char ch; bool flag = 0;
for (ch = getchar(); !isdigit(ch) && ((flag |= (ch == '-')) || 1); ch = getchar());
for (x = 0; isdigit(ch); x = (x<<1)+(x<<3)+ch-48, ch = getchar());
x *= 1-2*flag;
}
void print(LL x) {if (x > 9) print(x/10); putchar(x%10+48); }
void write(LL x) {if (x < 0) putchar('-'); print(Abs(x)); }
LL n, k, m, p, ans, t;
LL quick_pow(LL a, LL b) {
LL ans = 1;
while (b) {
if (b&1) ans = ans*a%p;
a = a*a%p, b >>= 1;
}
return ans;
}
void work() {
read(n), read(k), read(m), read(p); n %= p;
ans = n*quick_pow(m, k-1);
if (m&1) t = (m+1)/2*m%p; else t = m/2*(m+1)%p;
ans = (ans-(k-1)*t%p*quick_pow(m, k-2)%p)%p;
writeln((ans+p)%p);
}
int main() {
work(); return 0;
}