結果

問題 No.1013 〇マス進む
ユーザー f_stsh
提出日時 2020-09-22 14:40:35
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,379 bytes
コンパイル時間 2,102 ms
コンパイル使用メモリ 175,124 KB
実行使用メモリ 39,752 KB
最終ジャッジ日時 2024-06-26 03:41:26
合計ジャッジ時間 8,883 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 3 WA * 59
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
/* typedef */
typedef long long ll;
typedef pair<int, int> pii;
/* constant */
const int INF = 1 << 30;
const ll LINF = 1LL << 61;
const int NIL = -1;
const int MAX = 10000;
const int MOD = 1000000007;
const double pi = 3.141592653589;
/* global variables */
/* function */
void solve_dubling() {
    int n, k;
    cin >> n >> k;
    int MAXD = log2(1e+9 + 5);
    // init
    vector<int> a(n + 1);
    vector<vector<int>> to(MAXD + 1, vector<int>(n + 1, -1));
    vector<vector<ll>> sum(MAXD + 1, vector<ll>(n + 1, 0));
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        to[0][i] = (i + a[i]) % n;
        if (to[0][i] == 0) to[0][i] = n;
        sum[0][i] = a[i];
    }
    // build
    for (int d = 0; d < MAXD; d++) {
        for (int i = 1; i <= n; i++) {
            to[d + 1][i] = (to[d][i] + to[d][to[d][i]]) % n;
            if (to[d + 1][i] == 0) to[d + 1][i] = n;
            sum[d + 1][i] = sum[d][i] + sum[d][to[d][i]];
        }
    }
    // query
    for (int st = 1; st <= n; st++) {
        int cur = st;
        ll ret = 0;
        for (int d = MAXD; d >= 0; d--) {
            int b = 1 << d;
            if (k & b) {
                ret += sum[d][cur];
                cur = to[d][cur];
            }
        }
        cout << ret + st << '\n';
    }

}
/* main */
int main(){
    solve_dubling();
}
0