結果
問題 | No.1097 Remainder Operation |
ユーザー |
![]() |
提出日時 | 2020-06-26 22:20:21 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 67 ms / 2,000 ms |
コード長 | 1,755 bytes |
コンパイル時間 | 1,990 ms |
コンパイル使用メモリ | 197,652 KB |
最終ジャッジ日時 | 2025-01-11 11:44:05 |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 21 |
ソースコード
#include <bits/stdc++.h> #define rep(i, a) for (int i = (int)0; i < (int)a; ++i) #define rrep(i, a) for (int i = (int)a - 1; i >= 0; --i) #define REP(i, a, b) for (int i = (int)a; i < (int)b; ++i) #define RREP(i, a, b) for (int i = (int)a - 1; i >= b; --i) #define pb push_back #define eb emplace_back #define all(x) x.begin(), x.end() #define rall(x) x.rbegin(), x.rend() using ll = long long; constexpr ll mod = 1e9 + 7; constexpr ll INF = 1LL << 60; template <class T> inline bool chmin(T &a, T b) { if (a > b) { a = b; return true; } return false; } template <class T> inline bool chmax(T &a, T b) { if (a < b) { a = b; return true; } return false; } ll gcd(ll n, ll m) { ll tmp; while (m != 0) { tmp = n % m; n = m; m = tmp; } return n; } ll lcm(ll n, ll m) { return abs(n) / gcd(n, m) * abs(m); //gl=xy } using namespace std; void solve() { int n; cin>>n; vector<ll>a(n); rep(i,n)cin>>a[i]; ll x=0; vector<ll>used(n,-1); vector<ll>v; ll cnt=0; ll loop=0; ll start=0; v.eb(0); while(1){ x=x+a[x%n]; if(used[x%n]==-1)used[x%n]=cnt,v.eb(x); else { loop=cnt-used[x%n]; start=cnt+1; break; } ++cnt; } vector<ll>pre(start+1); rep(i,start){ pre[i+1]=pre[i]+a[v[i]%n]; } //cout<<loop<<endl; // rep(i,start+1){ // cout<<pre[i]<<"\n"; // } vector<ll>val(loop+1); rep(i,loop){ val[i+1]=val[i]+a[v[start-loop+i]%n]; } //rep(i,loop+1)cout<<val[i]<<endl; int q; cin>>q; while(q--){ ll k; cin>>k; if(k<=start){ cout<<pre[k]<<"\n"; } else{ ll x=(k-start)/loop*val.back(); x+=val[(k-start+loop)%loop]; x+=pre.back(); cout<<x<<"\n"; } } } int main() { ios::sync_with_stdio(false); cin.tie(0); cout << fixed << setprecision(15); solve(); return 0; }