結果
問題 | No.1097 Remainder Operation |
ユーザー |
|
提出日時 | 2020-09-01 18:21:44 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 180 ms / 2,000 ms |
コード長 | 1,241 bytes |
コンパイル時間 | 2,228 ms |
コンパイル使用メモリ | 197,936 KB |
最終ジャッジ日時 | 2025-01-14 03:04:40 |
ジャッジサーバーID (参考情報) |
judge4 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 21 |
ソースコード
#include <bits/stdc++.h>using namespace std;#define REP(i,n) for(int i=0; i<(int)(n); i++)#define FOR(i,b,e) for (int i=(int)(b); i<(int)(e); i++)#define ALL(x) (x).begin(), (x).end()const double PI = acos(-1);int main() {ios_base::sync_with_stdio(0);cin.tie(0);int n;cin >> n;vector<int> a(n);REP (i, n) cin >> a[i];int len = -1;int s = -1;vector<int> vis(n, -1);int x = 0;for (int i = 0; ; i++) {if (vis[x] != -1) {s = vis[x];len = i - s;break;}vis[x] = i;x += a[x];x %= n;}vector<long long> pref;vector<long long> cyc;x = 0;for (int i = 0; i < s + len; i++) {if (i < s) {long long tmp = pref.size() ? pref.back() : 0;pref.push_back(tmp + a[x]);} else {long long tmp = cyc.size() ? cyc.back() : 0;cyc.push_back(tmp + a[x]);}x += a[x];x %= n;}int q;cin >> q;while (q--) {long long k;cin >> k;--k;long long ret = 0;if (k < s) {ret = pref[k];} else {if (pref.size())ret += pref.back();k -= s;ret += k / len * cyc.back();ret += cyc[k % len];}cout << ret << endl;}return 0;}