結果

問題 No.2356 Back Door Tour in Four Seasons
ユーザー Carpenters-CatCarpenters-Cat
提出日時 2023-06-16 22:59:23
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 143 ms / 2,000 ms
コード長 1,255 bytes
コンパイル時間 2,037 ms
コンパイル使用メモリ 202,552 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-24 16:00:26
合計ジャッジ時間 6,415 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
ll const m = 998244353;
ll mpow(ll a, ll n) {
	ll ret = 1;
	while (n) {
		if (n & 1) ret = (ret * a) % m;
		n >>= 1;
		a = (a * a) % m;
	}
	return ret;
}
int main () {
	int N;
	cin >> N;
	string key = "UFWP";
	ll al = (N - 2);
	al = (al * mpow(N - 1, m - 2)) % m;
	ll num[] = {0, 0, 0, 0}, xi[] = {0, 0, 0, 0};
	ll ss = 0;
	for (int i = 0; i < N; i ++) {
		char c;
		cin >> c;
		int fl = 0;
		while (key[fl] != c) fl ++;
		ll a;
		cin >> a;
		ss += a;
		num[fl] ++;
		xi[fl] = (xi[fl] + mpow(al, a)) % m;
	}
	ll bs = mpow(N - 1, ss);
	ll ans1 = bs;
	for (int i = 0; i < 4; i ++) ans1 = (ans1 * num[i]) % m;
	ll ans2 = 0;
	for (int i = 0; i < 3; i ++) {
		ll kj = xi[i];
		for (int j = 1; j < 4; j ++) {
			kj = (kj * num[(i + j) % 4]) % m;
		}
		ans2 = (ans2 + kj * bs) % m;
	}
	ll ans3 = 0;
	for (int p = 0; p < 3; p ++) {
		for (int q = p + 1; q < 3; q ++) {
			ll kj = (xi[p] * xi[q]) % m;
			for (int i = 0; i < 4; i ++) {
				if (i != p && i != q) kj = (kj * num[i]) % m;
			}
			ans3 = (ans3 + kj) % m;
		}
	}
	ans3 = (ans3 * bs) % m;
	ll ans4 = (bs * num[3]) % m;
	for (int i = 0; i < 3; i ++) ans4 = (ans4 * xi[i]) % m;
	cout << (m * 4 + ans1 - ans2 + ans3 - ans4) % m << endl;
}
0