結果
問題 |
No.1097 Remainder Operation
|
ユーザー |
![]() |
提出日時 | 2024-03-06 01:10:07 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 192 ms / 2,000 ms |
コード長 | 997 bytes |
コンパイル時間 | 4,847 ms |
コンパイル使用メモリ | 196,408 KB |
最終ジャッジ日時 | 2025-02-20 01:10:26 |
ジャッジサーバーID (参考情報) |
judge2 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 21 |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; int main(){ cin.tie(nullptr); ios_base::sync_with_stdio(false); ll N, val=0, cnt=1, loop, start; cin >> N; vector<ll> A(N), ord(N, -1), vals; for (int i=0; i<N; i++) cin >> A[i]; while(ord[val] == -1){ ord[val] = cnt; vals.push_back(val); val = (A[val]+val) % N; cnt++; } start = ord[val]; loop = cnt - start; ll M = cnt-1; vector<ll> S(M+1); for (int i=1; i<=M; i++){ S[i] = S[i-1] + A[vals[i-1]]; } ll Q, K, q, r, loop_sum, line_sum; cin >> Q; loop_sum = S[M]-S[start-1]; line_sum = S[start-1]; while(Q--){ cin >> K; if(K < start){ cout << S[K] << endl; }else{ //start->K q = (K-start+1) / loop; r = (K-start+1) % loop; cout << S[start-1] + q*loop_sum + S[r+start-1]-line_sum << endl; } } return 0; }