小猴编程周赛C++ | 2022年秋第2周T2间距数列

学习C++从娃娃抓起！记录下在学而思小猴编程学习过程中的题目，记录每一个瞬间。侵权即删，谢谢支持！

附上汇总贴：小猴编程C++ | 汇总-CSDN博客

---

【题目描述】

满足以下条件的长为n的整数数列$a_1,a_2,\dots ,a_n$有多少个？

（1）$1\le a_i\le m(1\le i\le n)$

（2）$|a_i-a_{i+1}|\ge k(1\le i\le n-1)$

输出答案除以998244353的余数。

【输入】

1行，3个整数n,m,k。

【输出】

输出答案以998244353的余数。

【输入样例】

2 3 1

【输出样例】

6

【代码详解】

#include <bits/stdc++.h>
using namespace std;
#define int long long
int n, m, k, ans;
int s[1005][1005], dp[1005][1005];
signed main()
{
    cin >> n >> m >> k;
    if (k==0) {  // 对于0进行特判
        int mul = 1;
        for (int i=1; i<=n; i++) {  // 计算m^n
            mul = (mul * m)%998244353; 
        }
        cout << mul << endl;
        return 0;
    }
    for (int j=1; j<=m; j++) {
        dp[1][j] = 1;  // 初始dp[1][1~m]=1
        s[1][j] = j;  // 初始s[1][j]=j
    }
    for (int i=2; i<=n; i++) {
        for (int j=1; j<=m; j++) {
            dp[i][j] = 0;
            if (j-k>=0) {
                dp[i][j] = (dp[i][j] + s[i-1][j-k])%998244353;  // s[i-1][j-k]计算dp[i-1][1~j-k]这连续一段的和
            }
            if (j+k<=m) {
                dp[i][j] += (s[i-1][m] - s[i-1][j+k-1])%998244353;  // s[i-1][m]-s[i-1][j+k-1]计算dp[i-1][j+k~m]这一段的和‍```
            }
            s[i][j] = (s[i][j-1] + dp[i][j])%998244353;
        }
    }
    for (int i=1; i<=m; i++) {  // 计算所有可能长度的结果，并进行累加
        ans = (ans + dp[n][i])%998244353;
    }
    cout << ans << endl;
    return 0;
}

【运行结果】

2 3 1
6