結果

問題 No.1097 Remainder Operation
ユーザー 👑 CleyLCleyL
提出日時 2022-10-15 15:11:55
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,079 bytes
コンパイル時間 887 ms
コンパイル使用メモリ 82,576 KB
最終ジャッジ日時 2025-02-08 06:34:18
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 10 WA * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <set>
#include <vector>
using namespace std;
int main(){
  int n;cin>>n;
  vector<long long> A(n);
  for(int i = 0; n > i; i++){
    cin>>A[i];
  }
  vector<long long> pre;
  vector<long long> circle;
  set<int> B;
  int nw = 0;
  B.insert(nw);
  while(true){
    nw = (nw+A[nw])%n;
    if(B.count(nw)){
      break;
    }
    B.insert(nw);
  }
  int nw2 = 0;
  while(nw2 != nw){
    if(nw2)pre.push_back(pre[pre.size()-1]+A[nw2]);
    else pre.push_back(A[nw2]);
    nw2 = (nw2+A[nw2])%n;
  }
  circle.push_back(A[nw2]);
  nw2 = (nw2+A[nw2])%n;
  while(nw2 != nw){
    circle.push_back(circle[circle.size()-1]+A[nw2]);
    nw2 = (nw2+A[nw2])%n;
  }

  int q;cin>>q;
  for(int i = 0; q > i; i++){
    int x;cin>>x;
    if(pre.size() >= x){
      cout << pre[x-1] << endl;
    }else{
      long long ans = 0;
      if(pre.size())ans += pre[pre.size()-1];
      ans += (x-pre.size())/circle.size()*circle[circle.size()-1];
      if((x-pre.size())%circle.size() != 0)ans += circle[(x-pre.size()-1)%circle.size()];
      cout << ans << endl;
    }
  }
}
0