結果
問題 | No.1097 Remainder Operation |
ユーザー |
![]() |
提出日時 | 2020-06-26 23:05:00 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 240 ms / 2,000 ms |
コード長 | 1,221 bytes |
コンパイル時間 | 1,826 ms |
コンパイル使用メモリ | 179,564 KB |
実行使用メモリ | 11,568 KB |
最終ジャッジ日時 | 2024-07-04 23:29:47 |
合計ジャッジ時間 | 6,358 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 21 |
ソースコード
#include <bits/stdc++.h>using namespace std;#define REP(i,a,n) for(int i=(a); i<(int)(n); i++)#define rep(i,n) REP(i,0,n)#define FOR(it,c) for(__typeof((c).begin()) it=(c).begin(); it!=(c).end(); ++it)#define ALLOF(c) (c).begin(), (c).end()typedef long long ll;typedef unsigned long long ull;int main(){int N;cin >> N;vector<int> v;rep(i,N){int a;cin >> a;v.push_back(a);}vector<int> w;map<int,int> memo;int X = 0;while(true){X %= N;if(memo.count(X) != 0) break;memo[X] = w.size();w.push_back(X);X += v[X];}vector<int> w2;vector<ll> sum1, sum2;ll base = 0;rep(i,w.size()){base += v[w[i]];sum1.push_back(base);}base = 0;int st = (w[w.size()-1] + v[w[w.size()-1]])%N;REP(i,memo[st],w.size()){w2.push_back(w[i]);base += v[w[i]];sum2.push_back(base);}int Q;cin >> Q;rep(q,Q){ll K;cin >> K;K--;if(K < w.size()){cout << sum1[K] << endl;}else{ll sum = sum1[sum1.size()-1];K -= sum1.size();sum += (K/sum2.size()) * sum2[sum2.size()-1];sum += sum2[K%(sum2.size())];cout << sum << endl;}}return 0;}