結果

問題 No.1097 Remainder Operation
ユーザー Mayimg
提出日時 2020-07-03 15:52:33
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 162 ms / 2,000 ms
コード長 914 bytes
コンパイル時間 1,825 ms
コンパイル使用メモリ 201,044 KB
最終ジャッジ日時 2025-01-11 14:05:04
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #

#define _USE_MATH_DEFINES
#include <bits/stdc++.h>
using namespace std;
signed main() { 
  ios::sync_with_stdio(false); cin.tie(0);
  int n;
  cin >> n;
  vector<int> a(n);
  for (int i = 0; i < n; i++) {
    cin >> a[i];
  }
  vector<vector<int>> nxt(41, vector<int>(n));
  vector<vector<long long>> sum(41, vector<long long>(n));
  for (int i = 0; i < n; i++) {
    nxt[0][i] = (i + a[i]) % n;
    sum[0][i] = a[i];
  }
  for (int u = 1; u < 41; u++) {
    for (int i = 0; i < n; i++) {
      nxt[u][i] = nxt[u - 1][nxt[u - 1][i]];
      sum[u][i] = sum[u - 1][i] + sum[u - 1][nxt[u - 1][i]];
    }
  }
  int q;
  cin >> q;
  while (q--) {
    long long k;
    cin >> k;
    long long ans = 0;
    int cur = 0;
    for (int i = 40; i >= 0; i--) {
      if (k >= (1LL << i)) {
        ans += sum[i][cur];
        cur = nxt[i][cur];
        k -= (1LL << i);
      }
    }
    cout << ans << '\n';
  }
  return 0;
}
0