結果

問題 No.1013 〇マス進む
ユーザー 0w1
提出日時 2020-11-13 10:17:19
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 115 ms / 2,000 ms
コード長 931 bytes
コンパイル時間 2,132 ms
コンパイル使用メモリ 200,536 KB
最終ジャッジ日時 2025-01-15 22:18:02
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 62
権限があれば一括ダウンロードができます

ソースコード

diff #

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

int main() {
  ios::sync_with_stdio(false);

  int N, K; {
    cin >> N >> K;
  }

  vector<int> P(N); {
    for (int i = 0; i < N; ++i) cin >> P[i];
  }

  vector<int> rp(N + 1); {
    for (int i = 0; i < N; ++i) rp[P[i]] = i + 1;
  }

  vector<vector<int>> pf(30, vector<int>(N + 1));
  vector<vector<int64_t>> pp(30, vector<int64_t>(N + 1)); {
    for (int i = 0; i < N; ++i) pf[0][P[i]] = i + P[i] >= N;
    for (int i = 0; i < N; ++i) pp[0][P[i]] = P[(i + P[i]) % N];
    for (int i = 0; i < 29; ++i) {
      for (int u = 1; u <= N; ++u) {
        pf[i + 1][u] = pf[i][u] + pf[i][pp[i][u]];
        pp[i + 1][u] = pp[i][pp[i][u]];
      }
    }
  }

  for (int p : P) {
    int64_t c = 0;
    int r = p;
    for (int i = 29, k = K; k; --i) if (k >= 1 << i) {
      k -= 1 << i;
      c += (int64_t) N * pf[i][r];
      r = pp[i][r];
    }
    cout << (c + rp[r]) << "\n";
  }
}

0