Editorial for Lecture Gap
Use this editorial only when stuck, and do not copy-paste code from it.
Please be respectful to the problem author and editorialist. Submitting an official solution before solving the problem yourself is a bannable offence.
Approach
The gap condition xi+1≥xi+g makes the sequence strictly spaced. Shift it back to an ordinary non-decreasing sequence by setting
yi=xi−(i−1)g.
Then xi+1≥xi+g becomes yi+1≥yi, and the bounds become
1≤y1≤y2≤⋯≤yk≤n−(k−1)g.
Let m=n−(k−1)g. If m<1, no such y exists and the answer is 0 (do not evaluate a binomial coefficient with a negative upper index).
Otherwise the number of non-decreasing sequences of length k with values in {1,…,m} is combinations with repetition:
(km+k−1).
Special cases worth checking by hand:
- g=0: no forced gap, so the answer is (kn+k−1) (vanilla non-decreasing sequences in [1,n]);
- g=1: the sequence is strictly increasing, so m=n−k+1 and the answer is (kn);
- k=1: m=n and the answer is n.
Precompute factorials and inverse factorials modulo 109+7 up to n+k. Time is O(n+k).
Solution (Python)
MOD = 10**9 + 7
MAXN = 2 * 10**6 + 5
fact = [1] * MAXN
for i in range(1, MAXN):
fact[i] = fact[i - 1] * i % MOD
invfact = [1] * MAXN
invfact[MAXN - 1] = pow(fact[MAXN - 1], MOD - 2, MOD)
for i in range(MAXN - 2, -1, -1):
invfact[i] = invfact[i + 1] * (i + 1) % MOD
def nCr(n, k):
if k < 0 or k > n:
return 0
return fact[n] * invfact[k] % MOD * invfact[n - k] % MOD
n, k, g = map(int, input().split())
m = n - (k - 1) * g
if m < 1:
print(0)
else:
print(nCr(m + k - 1, k))
Comments0
No comments yet
Be the first to comment.
New comment
Log in to join the discussion.