結果

問題 No.1513 simple 門松列 problem
ユーザー 夜夜
提出日時 2021-05-23 11:57:49
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,390 bytes
コンパイル時間 874 ms
コンパイル使用メモリ 96,812 KB
実行使用メモリ 131,568 KB
最終ジャッジ日時 2024-04-19 20:29:12
合計ジャッジ時間 4,176 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
7,784 KB
testcase_02 AC 446 ms
131,248 KB
testcase_03 AC 28 ms
128,756 KB
testcase_04 AC 7 ms
32,356 KB
testcase_05 AC 21 ms
106,200 KB
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 AC 2 ms
6,944 KB
testcase_10 AC 2 ms
9,924 KB
testcase_11 AC 2 ms
7,636 KB
testcase_12 AC 2 ms
7,656 KB
testcase_13 AC 3 ms
11,996 KB
testcase_14 AC 451 ms
130,088 KB
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
権限があれば一括ダウンロードができます

ソースコード

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 << " " << ans2 << endl;
}
0