提出 #51655276

---

ソースコード 拡げる

#include <bits/stdc++.h>
using namespace std;

int main() {
  int N; cin >> N;
  vector<int> P(N);
  for (int i = 0; i < N; i++) {
    cin >> P.at(i);
    P.at(i)--;
  }
  string S; cin >> S;
  
  queue<string> q;
  q.push(S);
  while (q.front().find('?') != string::npos) {
    string s = q.front(); q.pop();
    int index = s.find_first_of('?');
    s.at(index) = 'L'; q.push(s);
    s.at(index) = 'R'; q.push(s);
  }
  
  unsigned long long ans = 0;
  
  while (!q.empty()) { // ある利き手の組み合わせ
    string domHands = q.front(); q.pop();
    vector<bool> spoons(N, true); 
    for (int i = 0; i < N; i++) { // 一人ずつ取ってく
      int p = P.at(i);
      bool leftSpoon = spoons.at(p), rightSpoon = spoons.at((p + 1) % N);
      if (leftSpoon && rightSpoon) { // スプーン2つアリ
        if (domHands.at(p) == 'L') spoons.at(p) = false;
        else spoons.at((p + 1) % N) = false;
      }
      else if (leftSpoon) { // 左だけアリ
        spoons.at(p) = false;
      }
      else if (rightSpoon) { // 右だけアリ
        spoons.at((p + 1) % N) = false;
      }
      else { // スプーンなし
        break;
      }
      if (i == N - 1) ans = (ans + 1) % 998244353;
    }
  }
  
  cout << ans % 998244353 << endl;
}

提出情報

ジャッジ結果