結果

問題 No.3040 Aoiスコア
コンテスト
ユーザー ripity
提出日時 2025-02-28 21:53:46
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,363 bytes
コンパイル時間 4,416 ms
コンパイル使用メモリ 263,576 KB
実行使用メモリ 7,844 KB
最終ジャッジ日時 2025-06-20 20:57:18
合計ジャッジ時間 10,664 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12 WA * 13 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;
 
#include <atcoder/all>
using namespace atcoder;
 
#pragma GCC target("avx")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
 
#define FASTIO cin.tie(0)->sync_with_stdio(0)
 
using ll = long long;
using mint = modint998244353;

int main() {
  FASTIO;
  int N, M;
  string S;
  cin >> N >> M >> S;
  vector<unordered_set<int>> G(N);
  for(int i = 0; i < M; i++) {
    int a, b;
    cin >> a >> b;
    a--, b--;
    G[a].insert(b);
    G[b].insert(a);
  }
  int q = 0;
  for(int i = 0; i < N; i++) {
    if(S[i] == '?') q++;
  }
  vector<mint> pw(N + 1, 1);
  for(int i = 0; i < N; i++) {
    pw[i + 1] = pw[i] * 26;
  }
  mint ans;
  for(int i = 0; i < N; i++) {
    for(int j = 0; j < N; j++) {
      if(i == j) continue;
      if(S[i] != 'a' && S[i] != '?') continue;
      if(S[j] != 'i' && S[i] != '?') continue;
      int co = 0, cq = 0;
      int s = (G[i].size() < G[j].size() ? i : j);
      int t = (G[i].size() < G[j].size() ? j : i);
      for(const int &k : G[s]) {
        if(!G[t].count(k)) continue;
        if(S[k] == 'o') co++;
        if(S[k] == '?') cq++;
      }
      int q2 = q - cq - (S[i] == '?') - (S[j] == '?');
      if(cq == 0) {
        ans += pw[q2] * co;
      }else {
        ans += pw[q2] * pw[cq - 1] * (26 * co + cq);
      }
    }
  }
  cout << ans.val() << endl;
}
0