結果

問題 No.1513 simple 門松列 problem
ユーザー 夜
提出日時 2021-05-23 11:58:30
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 495 ms / 3,000 ms
コード長 2,410 bytes
コンパイル時間 906 ms
コンパイル使用メモリ 97,832 KB
実行使用メモリ 131,120 KB
最終ジャッジ日時 2024-10-11 12:43:53
合計ジャッジ時間 4,276 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
#include <iomanip>
#include <cmath>
#include <stdio.h>
#include <queue>
#include <deque>
#include <cstdio>
#include <set>
#include <map>
#include <bitset>
#include <stack>
#include <cctype>
using namespace std;
long long dp[202][202][202] = {}, dp1[202][202][202] = {};
long long mod = 998244353;
int main() {
	long long n, m;
	cin >> n >> m;
	for (int i = 0; i < m; i++) {
		for (int j = 0; j < m; j++) {
			if (i != j) {
				dp[2][i][j] = 1;
				dp1[2][i][j] = i + j;
			}
		}
	}
	for (int i = 1; i < m; i++) {
		for (int j = 0; j < m; j++) {
			dp[2][i][j] += dp[2][i - 1][j];
			dp1[2][i][j] += dp1[2][i - 1][j];
			dp[2][i][j] %= mod;
			dp1[2][i][j] %= mod;
		}
	}
	for (int i = 3; i <= n; i++) {
		for (int j = 0; j < m; j++) {
			for (int k = 0; k < m; k++) {
				if (j == k) {
					dp[i][j][k] = 0;
					dp1[i][j][k] = 0;
				}
				if (j > k) {
					dp[i][j][k] += dp[i - 1][j - 1][j];
					dp[i][j][k] -= dp[i - 1][k][j];
					if (k != 0) {
						dp[i][j][k] += dp[i - 1][k - 1][j];
					}
					dp1[i][j][k] += dp1[i - 1][j - 1][j];
					dp1[i][j][k] += dp[i - 1][j - 1][j] * k;
					dp1[i][j][k] -= dp1[i - 1][k][j];
					dp1[i][j][k] -= dp[i - 1][k][j] * k;
					if (k != 0) {
						dp1[i][j][k] += dp1[i - 1][k - 1][j];
						dp1[i][j][k] += dp[i - 1][k - 1][j] * k;
					}
				}
				if (j < k) {
					dp[i][j][k] += dp[i - 1][m - 1][j];
					dp[i][j][k] -= dp[i - 1][j][j];
					dp[i][j][k] -= dp[i - 1][k][j];
					if (k != 0) {
						dp[i][j][k] += dp[i - 1][k - 1][j];
					}
					dp1[i][j][k] += dp1[i - 1][m - 1][j];
					dp1[i][j][k] += dp[i - 1][m - 1][j] * k;
					dp1[i][j][k] -= dp1[i - 1][j][j];
					dp1[i][j][k] -= dp[i - 1][j][j] * k;
					dp1[i][j][k] -= dp1[i - 1][k][j];
					dp1[i][j][k] -= dp[i - 1][k][j] * k;
					if (k != 0) {
						dp1[i][j][k] += dp1[i - 1][k - 1][j];
						dp1[i][j][k] += dp[i - 1][k - 1][j] * k;
					}
				}
				dp[i][j][k] %= mod;
				dp1[i][j][k] %= mod;
			}
		}
		for (int j = 1; j < m; j++) {
			for (int k = 0; k < m; k++) {
				dp[i][j][k] += dp[i][j - 1][k];
				dp1[i][j][k] += dp1[i][j - 1][k];
				dp[i][j][k] %= mod;
				dp1[i][j][k] %= mod;
			}
		}
	}
	long long ans1 = 0, ans2 = 0;
	for (int i = 0; i < m; i++) {
		ans1 += dp[n][m - 1][i];
		ans2 += dp1[n][m - 1][i];
		ans1 %= mod, ans2 %= mod;
	}
	cout << (ans1+mod)%mod << " " << (ans2+mod)%mod << endl;
}
0