結果

問題 No.1513 simple 門松列 problem
ユーザー 夜夜
提出日時 2021-05-23 11:58:30
言語 C++14
(gcc 12.3.0 + boost 1.83.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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
6,820 KB
testcase_01 AC 3 ms
7,752 KB
testcase_02 AC 494 ms
130,440 KB
testcase_03 AC 17 ms
68,292 KB
testcase_04 AC 8 ms
34,276 KB
testcase_05 AC 17 ms
68,244 KB
testcase_06 AC 13 ms
56,812 KB
testcase_07 AC 5 ms
17,880 KB
testcase_08 AC 19 ms
70,056 KB
testcase_09 AC 3 ms
7,504 KB
testcase_10 AC 3 ms
7,664 KB
testcase_11 AC 3 ms
7,640 KB
testcase_12 AC 3 ms
9,672 KB
testcase_13 AC 3 ms
11,612 KB
testcase_14 AC 486 ms
131,120 KB
testcase_15 AC 493 ms
129,992 KB
testcase_16 AC 495 ms
129,860 KB
testcase_17 AC 313 ms
95,536 KB
testcase_18 AC 102 ms
30,280 KB
testcase_19 AC 29 ms
74,616 KB
testcase_20 AC 72 ms
51,268 KB
権限があれば一括ダウンロードができます

ソースコード

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