結果
問題 |
No.1013 〇マス進む
|
ユーザー |
|
提出日時 | 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 |
ソースコード
#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"; } }