結果

問題 No.2356 Back Door Tour in Four Seasons
ユーザー Carpenters-Cat
提出日時 2023-06-16 22:49:59
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 940 bytes
コンパイル時間 2,131 ms
コンパイル使用メモリ 195,256 KB
最終ジャッジ日時 2025-02-14 21:14:41
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 WA * 1
other AC * 5 WA * 23
権限があれば一括ダウンロードができます

ソースコード

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 = bs * (xi[0] + xi[1] + xi[2]) % m;
	ll ans3 = (xi[0] * xi[1] + xi[1] * xi[2] + xi[2] * xi[0]) % m;
	ans3 = (ans3 * bs) % m;
	ll ans4 = bs;
	for (int i = 0; i < 3; i ++) ans4 = (ans4 * xi[i]) % m;
	cout << (m * 4 + ans1 - ans2 + ans3 - ans4) % m << endl;
}
0