結果

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

ソースコード

diff # 

#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;
  }
  int cnt = 0;
  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[j] != '?') 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]) {
        assert(++cnt <= N * M);
        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