結果

問題 No.3040 Aoiスコア
ユーザー Carpenters-Cat
提出日時 2025-02-28 21:49:12
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 891 bytes
コンパイル時間 4,595 ms
コンパイル使用メモリ 252,880 KB
実行使用メモリ 7,844 KB
最終ジャッジ日時 2025-06-20 20:56:29
合計ジャッジ時間 5,077 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 RE * 1
other AC * 23 RE * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using mint = atcoder::modint998244353;
int main () {
	int N, M;
	cin >> N >> M;
	string s;
	cin >> s;
	int nh = 0;
	for (auto& a : s) nh += (a == '?');
	std::vector<vector<int>> gr(N);
	for (int i = 0; i < M; i ++) {
		int a, b;
		cin >> a >> b;
		gr[--a].push_back(--b);
		gr[b].push_back(a);
	}
	mint ans = 0;
	mint pw = 26;
	mint ai = pw.pow(nh), hi = pw.pow(nh - 1), hh = pw.pow(nh - 2);
	mint iv = pw.inv();
	for (int i = 0; i < N; i ++) {
		mint ca = 0, ci = 0, ch = 0;
		if ((s[i] != 'o') && (s[i] != '?')) continue;
		for (auto& j : gr[i]) {
			if (s[j] == 'a') {
				ca ++;
			} else if (s[j] == 'i') {
				ci ++;
			} else if (s[j] == '?') {
				ch ++;
			}
		}
		mint pl = ca * ci * ai + (ca * ch + ch * ci) * hi + (ch * (ch - 1)) * hh;
		if (s[i] == '?') pl *= iv;
		ans += pl;
	}
	cout << ans.val() << endl;
}
0