結果
問題 |
No.1097 Remainder Operation
|
ユーザー |
|
提出日時 | 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 |
ソースコード
#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; }