結果

問題 No.1097 Remainder Operation
ユーザー srjywrdnprkt
提出日時 2024-03-06 01:10:07
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 192 ms / 2,000 ms
コード長 997 bytes
コンパイル時間 4,847 ms
コンパイル使用メモリ 196,408 KB
最終ジャッジ日時 2025-02-20 01:10:26
ジャッジサーバーID
(参考情報)
judge2 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>

using namespace std;
using ll = long long;

int main(){
    cin.tie(nullptr);
    ios_base::sync_with_stdio(false);

    ll N, val=0, cnt=1, loop, start;
    cin >> N;
    vector<ll> A(N), ord(N, -1), vals; 
    for (int i=0; i<N; i++) cin >> A[i];
    while(ord[val] == -1){
        ord[val] = cnt;
        vals.push_back(val);
        val = (A[val]+val) % N;
        cnt++;
    }
    start = ord[val];
    loop = cnt - start;
   
    ll M = cnt-1;
    vector<ll> S(M+1);
    for (int i=1; i<=M; i++){
        S[i] = S[i-1] + A[vals[i-1]];
    }

    ll Q, K, q, r, loop_sum, line_sum;
    cin >> Q;
    loop_sum = S[M]-S[start-1];
    line_sum = S[start-1];

    while(Q--){
        cin >> K;
        if(K < start){
            cout << S[K] << endl;
        }else{
            //start->K
            q = (K-start+1) / loop;
            r = (K-start+1) % loop;
            cout << S[start-1] + q*loop_sum + S[r+start-1]-line_sum << endl;
        }
    }

    return 0;
}
0