結果

問題 No.1097 Remainder Operation
ユーザー IKyoproIKyopro
提出日時 2020-06-26 21:40:15
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 273 ms / 2,000 ms
コード長 991 bytes
コンパイル時間 2,182 ms
コンパイル使用メモリ 198,560 KB
最終ジャッジ日時 2025-01-11 11:05:24
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
template <class T, class U> using Pa = pair<T, U>;
template <class T> using vec = vector<T>;
template <class T> using vvec = vector<vec<T>>;

int main(){
    cin.tie(0);
    ios::sync_with_stdio(false);
    int N;
    cin >> N;
    const int K = 40;
    vec<ll> A(N);
    vvec<ll> ne(K+1,vec<ll>(N)),val = ne;
    for(int i=0;i<N;i++){
        cin >> A[i];
        ne[0][i] = (i+A[i])%N;
        val[0][i] = A[i];
    }
    for(int k=1;k<=K;k++){
        for(int i=0;i<N;i++){
            ne[k][i] = ne[k-1][ne[k-1][i]];
            val[k][i] = val[k-1][i]+val[k-1][ne[k-1][i]];
        }
    }
    int Q;
    cin >> Q;
    while(Q--){
        ll X;
        cin >> X;
        ll ans = 0;
        int now = 0;
        for(int k=K;k>=0;k--){
            if(X>>k&1){
                ans += val[k][now];
                now = ne[k][now];
                X -= 1LL<<k;
            }
        }
        cout << ans << "\n";
    }
}
0