結果

問題 No.1097 Remainder Operation
コンテスト
ユーザー Drice
提出日時 2020-06-27 00:41:19
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,485 bytes
コンパイル時間 263 ms
コンパイル使用メモリ 32,384 KB
実行使用メモリ 13,752 KB
最終ジャッジ日時 2024-07-05 02:58:25
合計ジャッジ時間 6,424 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 10 WA * 9 TLE * 1 -- * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<cstdio>
long long a[100005];
long long sum[100005];
int vis[100005];
int step[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;
    int u = 0; 
    long long cycle = -1, start = -1;
    long long weight = 0;
    while(true){
        int v = (u+a[u])%n;
        //printf("u = %d v = %d\n",u,v);
        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];
            u = v;
        }
    }
    //printf("start = %lld, cycle = %lld, weight = %lld\n",start,cycle,weight);
    int q;
    scanf("%d",&q);
    while(q--){
        int k;
        scanf("%d",&k);
        if(k<step[start]) printf("%lld\n",getSum(0,k,n));
        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 += getSum(start,left,n);
            printf("%lld\n",ans);
        }
    }
    
    return 0;
}
0