結果

問題 No.1097 Remainder Operation
ユーザー HaarHaar
提出日時 2020-06-26 21:56:38
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 241 ms / 2,000 ms
コード長 1,285 bytes
コンパイル時間 5,078 ms
コンパイル使用メモリ 197,408 KB
最終ジャッジ日時 2025-01-11 11:23:14
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

#ifdef DEBUG
#include <Mylib/Debug/debug.cpp>
#else
#define dump(...)
#endif

int main(){
  int N;

  while(std::cin >> N){
    std::vector<int64_t> A(N);
    for(int i = 0; i < N; ++i) std::cin >> A[i];

    int64_t len = 0, loop = 0;
    std::vector<int> p(N, -1), v(N, -1);

    int X = 0;
    for(int i = 0; i < 2 * N; ++i){
      if(p[X] != -1){
        len = i;
        loop = i - p[X];
        break;
      }

      p[X] = i;
      v[i] = X;
      (X += A[X]) %= N;
    }


    int tail = len - loop;

    //std::cerr << len << " " << loop << " " << tail << "\n";

    std::vector<int64_t> tail_sum(tail + 1);
    for(int i = 0; i < tail; ++i) tail_sum[i+1] = tail_sum[i] + A[v[i]];

    dump(tail_sum);

    std::vector<int64_t> loop_sum(loop + 1);
    for(int i = 0; i < loop; ++i){
      loop_sum[i+1] = loop_sum[i] + A[v[tail + i]];
    }

    dump(loop_sum);

    

    
    

    int Q; std::cin >> Q;

    while(Q--){
      int64_t K; std::cin >> K;
      
      int64_t ans = 0;

      if(K <= tail){
        ans = tail_sum[K];
      }else{
        ans += tail_sum.back();
        K -= tail;

        ans += loop_sum.back() * (K / loop);
        ans += loop_sum[K % loop];
      }

      std::cout << ans << "\n";
    }
    
  }

  return 0;
}
0