結果

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

ソースコード

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;
  }
  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]) {
        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