結果

問題 No.1097 Remainder Operation
ユーザー DriceDrice
提出日時 2020-06-27 00:50:47
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 42 ms / 2,000 ms
コード長 1,548 bytes
コンパイル時間 155 ms
コンパイル使用メモリ 31,572 KB
実行使用メモリ 6,160 KB
最終ジャッジ日時 2023-09-18 12:21:16
合計ジャッジ時間 2,856 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 1 ms
4,376 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 1 ms
4,380 KB
testcase_04 AC 1 ms
4,380 KB
testcase_05 AC 1 ms
4,380 KB
testcase_06 AC 1 ms
4,376 KB
testcase_07 AC 5 ms
4,380 KB
testcase_08 AC 5 ms
4,376 KB
testcase_09 AC 5 ms
4,380 KB
testcase_10 AC 5 ms
4,380 KB
testcase_11 AC 6 ms
4,380 KB
testcase_12 AC 41 ms
4,404 KB
testcase_13 AC 42 ms
5,020 KB
testcase_14 AC 42 ms
5,244 KB
testcase_15 AC 42 ms
5,352 KB
testcase_16 AC 42 ms
4,692 KB
testcase_17 AC 42 ms
4,380 KB
testcase_18 AC 40 ms
6,160 KB
testcase_19 AC 40 ms
6,128 KB
testcase_20 AC 40 ms
6,104 KB
testcase_21 AC 40 ms
6,124 KB
testcase_22 AC 41 ms
6,056 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<cstdio>
long long a[100005];
long long sum[100005];
int vis[100005];
int step[100005];
long long pre[100005];

long long getSum(int r,int k,int n){
    long long mask = r, res = 0;
    for(int i = 0; i < k; i++){
        res += a[mask];
        mask = (mask+a[mask])%n;
    }
    return res;
}

int main(){
    int n;
    scanf("%d",&n);
    for(int i = 0; i < n; i++) scanf("%lld",&a[i]);
    vis[0] = 1, step[0] = 0, sum[0] = 0;
    pre[step[0]] = sum[0];
    int u = 0; 
    long long cycle = -1, start = -1;
    long long weight = 0;
    while(true){
        int v = (u+a[u])%n;
        if(vis[v]==1){
            cycle = step[u]+1-step[v];
            start = v;
            weight = sum[u]+a[u]-sum[v];
            break;
        }
        else{
            vis[v] = 1;
            step[v] = step[u]+1;
            sum[v] = sum[u]+a[u];
            pre[step[v]] = sum[v];
            u = v;
        }
    }
    //printf("start = %lld, cycle = %lld, weight = %lld\n",start,cycle,weight);
    int q;
    scanf("%d",&q);
    while(q--){
        long long k;
        scanf("%lld",&k);
        if(k<step[start]) printf("%lld\n",pre[k]);
        else{
            long long ans = 0;
            k -= step[start];
            ans += sum[start];
            long long cnt = k/cycle;
            //printf("ans = %lld, cnt = %lld\n",ans,cnt);
            ans += cnt*weight;
            long long left = k%cycle;
            ans += pre[step[start]+left]-pre[step[start]];
            printf("%lld\n",ans);
        }
    }
    
    return 0;
}
0