C - 1111gal password

题目描述

题目链接

C++代码

#include<bits/stdc++.h>
using namespace std;

const int N = 1e6 + 10;
const int mod = 998244353;

int f[N][11];

int main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	
	int n;
	cin >> n;
	
	for(int j = 1; j <= 9; j ++) f[1][j] = 1;
	
	for(int d = 2; d <= n; d ++)
		for(int i = 1; i <= 9; i ++)
			for(int j = max(1, i - 1); j <= min(9, i + 1); j ++){
			f[d][j] += f[d - 1][i];
			f[d][j] %= mod;
		}
	
	int res = 0;
	for(int j = 1; j <= 9; j ++){
		res += f[n][j];
		res %= mod;
	}
	
	
	cout << res  << '\n';
	return 0;
}

参考资料

链接地址