結果
問題 | No.1097 Remainder Operation |
ユーザー |
![]() |
提出日時 | 2020-06-26 22:48:48 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 465 ms / 2,000 ms |
コード長 | 1,875 bytes |
コンパイル時間 | 1,941 ms |
コンパイル使用メモリ | 183,164 KB |
実行使用メモリ | 84,736 KB |
最終ジャッジ日時 | 2024-07-04 22:50:27 |
合計ジャッジ時間 | 7,875 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 21 |
ソースコード
#pragma GCC optimize("O3")#include <bits/stdc++.h>#define ll long long#define rep(i,n) for(ll i=0;i<(n);i++)#define pll pair<ll,ll>#define pq priority_queue#define pb push_back#define eb emplace_back#define fi first#define se second#define ios ios_base::sync_with_stdio(0),cin.tie(0),cout.tie(0);#define lb(c,x) distance(c.begin(),lower_bound(all(c),x))#define ub(c,x) distance(c.begin(),upper_bound(all(c),x))using namespace std;template<class T> inline bool chmax(T& a,T b){if(a<b){a=b;return 1;}return 0;}template<class T> inline bool chmin(T& a,T b){if(a>b){a=b;return 1;}return 0;}const ll INF=1e18;int main(){ll n;cin >> n;vector<ll> a(n),to(n);vector<vector<pll>> lca(50,vector<pll>(n));rep(i,n){cin >> a[i];}for(ll i=0;i<n;i++){ll p=i+a[i];to[i]=p;}for(ll i=0;i<50;i++){if(i==0){rep(j,n){lca[i][j].fi=to[j]%n;lca[i][j].se=a[j];}continue;}rep(j,n){lca[i][j].fi=lca[i-1][lca[i-1][j].fi].fi;lca[i][j].se=lca[i-1][j].se+lca[i-1][lca[i-1][j].fi].se;}}ll q;cin >> q;vector<ll> db(50);ll v=1;rep(i,50){db[i]=v;v*=2;}/*rep(i,50){rep(j,n){if(j) cout << " ";cout << lca[i][j].se;}cout << endl;}*/rep(i,q){ll k;cin >> k;vector<ll> t(50);ll num=49;while(k){if(k>=db[num]){k-=db[num];t[num]=1;}num--;}ll ans=0;ll r=0;rep(j,50){if(t[j]){ans+=lca[j][r].se;r=lca[j][r].fi;}}cout << ans << endl;}}