結果
問題 | No.1097 Remainder Operation |
ユーザー |
![]() |
提出日時 | 2021-04-15 19:29:51 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 223 ms / 2,000 ms |
コード長 | 1,238 bytes |
コンパイル時間 | 630 ms |
コンパイル使用メモリ | 72,052 KB |
実行使用メモリ | 6,272 KB |
最終ジャッジ日時 | 2024-07-02 04:10:08 |
合計ジャッジ時間 | 5,190 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 21 |
ソースコード
#include <iostream>#include <vector>using namespace std;using ll = long long int;int main(){int n;cin >> n;vector<int> a(n);for(auto &p: a) cin >> p;int q;cin >> q;vector<ll> k(q);for(auto &p: k) cin >> p;vector<int> order(n+10, -1);vector<ll> x(n+10, 0);ll loopstart, loopend, mod, prenum;for(int i = 1, r = 0; ; i++){order[r] = i;x[i] = x[i-1] + a[r];int prevr = r;r = x[i]%n;if(order[r] != -1){loopstart = order[r];loopend = order[prevr];prenum = order[r]-1;mod = loopend - loopstart + 1;//cerr << loopstart << " " << loopend << endl;break;}}vector<ll> loopsum(mod+1);for(int i = 0; i < mod; i++) loopsum[i] = x[i+loopstart]-x[loopstart-1];for(auto &p: k){if(p < loopstart) cout << x[p] << endl;else{ll ans = x[loopstart-1];ll loopcnt = (p-loopstart+1)/mod;ll loopind = (p-loopstart+1)%mod -1;ans += loopcnt * loopsum[mod-1];if(loopind != -1) ans += loopsum[loopind];cout << ans << endl;}}return 0;}