結果

問題 No.1097 Remainder Operation
コンテスト
ユーザー chocorusk
提出日時 2020-06-27 05:27:41
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 277 ms / 2,000 ms
コード長 1,446 bytes
コンパイル時間 1,399 ms
コンパイル使用メモリ 133,248 KB
最終ジャッジ日時 2025-01-11 12:33:52
ジャッジサーバーID
（参考情報） 		judge5 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 21
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:53:9: warning: ‘x1’ may be used uninitialized [-Wmaybe-uninitialized]
   53 |         if(v[i]==x1){
      |         ^~
main.cpp:41:9: note: ‘x1’ was declared here
   41 |     int x1, l;
      |         ^~
main.cpp:58:20: warning: ‘l’ may be used uninitialized [-Wmaybe-uninitialized]
   58 |     vector<ll> s1(l+1), s2(m+1);
      |                   ~^~
main.cpp:41:13: note: ‘l’ was declared here
   41 |     int x1, l;
      |             ^

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
#include <fstream>
#include <utility>
#include <functional>
#include <time.h>
#include <stack>
#include <array>
#include <list>
#define popcount __builtin_popcount
using namespace std;
typedef long long int ll;
typedef pair<int, int> P;

int main()
{
    int n; cin>>n;
    int a[100010];
    for(int i=0; i<n; i++){
        cin>>a[i];
    }
    int x=0;
    vector<int> v;
    set<int> st;
    //v.push_back(0), st.insert(0);
    int x1, l;
    for(int i=0; i<=n; i++){
        if(st.find(x)!=st.end()){
            x1=x;
            break;
        }
        v.push_back(x);
        st.insert(x);
        x+=a[x];
        x%=n;
    }
    for(int i=0; i<v.size(); i++){
        if(v[i]==x1){
            l=i; break;
        }
    }
    int m=v.size()-l;
    vector<ll> s1(l+1), s2(m+1);
    for(int i=0; i<l; i++) s1[i+1]=s1[i]+a[v[i]];
    for(int i=0; i<m; i++) s2[i+1]=s2[i]+a[v[l+i]];
    int q; cin>>q;
    while(q--){
        ll k; cin>>k;
        if(k<=l){
            cout<<s1[k]<<endl;
        }else{
            k-=l;
            cout<<s1.back()+(k-1)/m*s2.back()+s2[1+(k-1)%m]<<endl;
        }
    }
    return 0;
}
0