結果

問題 No.1097 Remainder Operation
ユーザー HaarHaar
提出日時 2020-06-26 21:56:38
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 225 ms / 2,000 ms
コード長 1,285 bytes
コンパイル時間 2,160 ms
コンパイル使用メモリ 204,680 KB
実行使用メモリ 5,632 KB
最終ジャッジ日時 2024-07-04 21:00:35
合計ジャッジ時間 6,574 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 23 ms
5,376 KB
testcase_08 AC 22 ms
5,376 KB
testcase_09 AC 22 ms
5,376 KB
testcase_10 AC 21 ms
5,376 KB
testcase_11 AC 22 ms
5,376 KB
testcase_12 AC 218 ms
5,376 KB
testcase_13 AC 216 ms
5,376 KB
testcase_14 AC 220 ms
5,376 KB
testcase_15 AC 213 ms
5,376 KB
testcase_16 AC 217 ms
5,376 KB
testcase_17 AC 225 ms
5,376 KB
testcase_18 AC 198 ms
5,632 KB
testcase_19 AC 202 ms
5,632 KB
testcase_20 AC 205 ms
5,504 KB
testcase_21 AC 212 ms
5,504 KB
testcase_22 AC 206 ms
5,504 KB
権限があれば一括ダウンロードができます

ソースコード

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