結果

問題 No.2045 Two Reflections
ユーザー SSRSSSRS
提出日時 2022-08-19 21:57:35
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 33 ms / 2,000 ms
コード長 909 bytes
コンパイル時間 1,983 ms
コンパイル使用メモリ 198,868 KB
最終ジャッジ日時 2025-01-31 01:03:38
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
const long long MOD = 998244353;
int main(){
  int N, p, q;
  cin >> N >> p >> q;
  if (p == 1 && q == 1){
    cout << 1 << endl;
  } else if (p == 1 || q == 1){
    cout << 2 << endl;
  } else {
    vector<int> A(N);
    for (int i = 0; i < N; i++){
      A[i] = i;
    }
    reverse(A.begin(), A.begin() + p);
    reverse(A.end() - q, A.end());
    vector<int> c;
    vector<bool> used(N, false);
    for (int i = 0; i < N; i++){
      if (!used[i]){
        used[i] = true;
        int cnt = 1;
        int p = i;
        while (true){
          p = A[p];
          if (used[p]){
            break;
          }
          used[p] = true;
          cnt++;
        }
        c.push_back(cnt);
      }
    }
    int cnt = c.size();
    long long ans = 1;
    for (int i = 0; i < cnt; i++){
      ans = lcm(ans, c[i]);
    }
    cout << ans * 2 % MOD << endl;
  }
}
0