#503205 (C++14) No.1097 Remainder Operation

• 問題: No.1097 Remainder Operation
• ユーザー: snow39
• 提出日時: 2020-06-26 21:57:12
• 言語: C++14 (gcc 13.2.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 38 ms / 2,000 ms
• コード長: 1,223 bytes
• コンパイル時間: 799 ms
• コンパイル使用メモリ: 97,508 KB
• 実行使用メモリ: 6,560 KB
• 最終ジャッジ日時: 2023-09-18 04:24:37
• 合計ジャッジ時間: 3,342 ms
• ジャッジサーバーID （参考情報）: judge12 / judge15

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 4,384 KB
testcase_01	AC	2 ms; 4,376 KB
testcase_02	AC	2 ms; 4,380 KB
testcase_03	AC	1 ms; 4,376 KB
testcase_04	AC	2 ms; 4,376 KB
testcase_05	AC	2 ms; 4,384 KB
testcase_06	AC	2 ms; 4,380 KB
testcase_07	AC	5 ms; 4,376 KB
testcase_08	AC	5 ms; 4,376 KB
testcase_09	AC	5 ms; 4,380 KB
testcase_10	AC	5 ms; 4,380 KB
testcase_11	AC	5 ms; 4,376 KB
testcase_12	AC	37 ms; 4,812 KB
testcase_13	AC	37 ms; 5,212 KB
testcase_14	AC	38 ms; 5,216 KB
testcase_15	AC	37 ms; 5,224 KB
testcase_16	AC	37 ms; 4,800 KB
testcase_17	AC	38 ms; 4,932 KB
testcase_18	AC	36 ms; 6,428 KB
testcase_19	AC	36 ms; 6,316 KB
testcase_20	AC	36 ms; 6,424 KB
testcase_21	AC	36 ms; 6,244 KB
testcase_22	AC	36 ms; 6,560 KB

ソースコード

#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <cmath>
#include <map>
#include <queue>
#include <iomanip>
#include <set>
#include <tuple>
#define mkp make_pair
#define mkt make_tuple
#define rep(i,n) for(int i = 0; i < (n); ++i)
#define all(v) v.begin(),v.end()
using namespace std;
typedef long long ll;
const ll MOD=1e9+7;
template<class T> void chmin(T &a,const T &b){if(a>b) a=b;}
template<class T> void chmax(T &a,const T &b){if(a<b) a=b;}

int main(){
  cin.tie(0);
  ios::sync_with_stdio(false);

  int N;
  cin>>N;
  vector<ll> A(N);
  rep(i,N) cin>>A[i];
  int Q;
  cin>>Q;
  vector<ll> K(Q);
  rep(i,Q) cin>>K[i];

  vector<int> used(N,0);
  vector<int> idxs;
  ll x=0;
  while(1){
    int r=x%N;
    if(used[r]) break;
    idxs.push_back(r);
    used[r]=1;
    x+=A[r];
  }

  int P=idxs.size();
  vector<ll> sum(P+1,0);
  int st=0;
  for(int i=0;i<P;i++){
    if(idxs[i]==x%N){
      st=i;
      break;
    }
  }
  int T=P-st;

  rep(i,P) sum[i+1]=sum[i]+A[idxs[i]];
  rep(q,Q){
    ll tar=K[q];
    if(tar<=st){
      cout<<sum[tar]<<"\n";
      continue;    
    }

    tar-=st;
    ll res=(sum[P]-sum[st])*(tar/T)+sum[st+tar%T];
    cout<<res<<"\n";
  }

  return 0;
}