結果

問題 No.1097 Remainder Operation
ユーザー 👑 potato167
提出日時 2022-01-29 14:36:22
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 387 ms / 2,000 ms
コード長 1,636 bytes
コンパイル時間 2,431 ms
コンパイル使用メモリ 211,696 KB
最終ジャッジ日時 2025-01-27 17:49:43
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#define _GLIBCXX_DEBUG
using std::cout;
using std::cin;
using std::endl;

namespace po167{
template<class T,T(*op)(T,T)>
struct Doubling_op
{
	int _n;
	long long Log;
	std::vector<std::vector<int>> index_data;
	std::vector<std::vector<T>> value_data;
	Doubling_op(int n,long long limit,std::vector<int> index,std::vector<T> value):_n(n){
		Log=2;
		while(limit){
			limit/=2;
			Log++;
		}
		index_data.resize(Log);
		value_data.resize(Log);
		index_data[0]=index;
		value_data[0]=value;
		for(int i=1;i<Log;i++){
			index_data[i].resize(_n);
			value_data[i].resize(_n);
			for(int j=0;j<_n;j++){
				int tmp=index_data[i-1][j];
				index_data[i][j]=index_data[i-1][tmp];
				value_data[i][j]=op(value_data[i-1][j],value_data[i-1][tmp]);
			}
		}
	}
	std::pair<int,T> query(long long times,int start_index,T start_value){
		for(int i=0;i<Log;i++){
			if(times&1){
				start_value=op(start_value,value_data[i][start_index]);
				start_index=index_data[i][start_index];
			}
			times>>=1;
		}
		return std::make_pair(start_index,start_value);
	}
};
}

using doubleing_F=long long;
doubleing_F d_op(doubleing_F l,doubleing_F r){
	return l+r;
}

void solve();

//  rainy ~ 雨に打たれて ~
int main() {
	
	int t=1;
	//cin>>t;
	solve();
}

void solve(){
	int n;
	cin>>n;
	std::vector<int> q(n);
	std::vector<long long> p(n);
	for(int i=0;i<n;i++) cin>>p[i];
	for(int i=0;i<n;i++){
		q[i]=(i+p[i])%n;
	}
	long long L=1e13;
	po167::Doubling_op<doubleing_F,d_op> D(n,L,q,p);
	int Q;
	cin>>Q;
	for(int i=0;i<Q;i++){
		long long k;
		cin>>k;
		cout<<D.query(k,0,0).second<<"\n";
	}
	
}
0