結果

問題 No.1097 Remainder Operation
ユーザー srjywrdnprktsrjywrdnprkt
提出日時 2024-03-06 01:10:07
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 160 ms / 2,000 ms
コード長 997 bytes
コンパイル時間 5,524 ms
コンパイル使用メモリ 205,100 KB
実行使用メモリ 6,676 KB
最終ジャッジ日時 2024-03-06 01:10:22
合計ジャッジ時間 9,729 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,676 KB
testcase_01 AC 2 ms
6,676 KB
testcase_02 AC 2 ms
6,676 KB
testcase_03 AC 2 ms
6,676 KB
testcase_04 AC 2 ms
6,676 KB
testcase_05 AC 2 ms
6,676 KB
testcase_06 AC 2 ms
6,676 KB
testcase_07 AC 17 ms
6,676 KB
testcase_08 AC 17 ms
6,676 KB
testcase_09 AC 17 ms
6,676 KB
testcase_10 AC 17 ms
6,676 KB
testcase_11 AC 17 ms
6,676 KB
testcase_12 AC 158 ms
6,676 KB
testcase_13 AC 158 ms
6,676 KB
testcase_14 AC 157 ms
6,676 KB
testcase_15 AC 158 ms
6,676 KB
testcase_16 AC 159 ms
6,676 KB
testcase_17 AC 158 ms
6,676 KB
testcase_18 AC 158 ms
6,676 KB
testcase_19 AC 159 ms
6,676 KB
testcase_20 AC 158 ms
6,676 KB
testcase_21 AC 159 ms
6,676 KB
testcase_22 AC 160 ms
6,676 KB
権限があれば一括ダウンロードができます

ソースコード

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