#735551 (C++17) No.1097 Remainder Operation

• 問題: No.1097 Remainder Operation
• ユーザー: 👑 potato167
• 提出日時: 2022-01-29 14:36:22
• 言語: C++17 (gcc 13.3.0 + boost 1.87.0)
• 結果: AC
• 実行時間: 352 ms / 2,000 ms
• コード長: 1,636 bytes
• コンパイル時間: 2,183 ms
• コンパイル使用メモリ: 210,268 KB
• 実行使用メモリ: 59,648 KB
• 最終ジャッジ日時: 2025-01-02 01:47:25
• 合計ジャッジ時間: 9,052 ms
• ジャッジサーバーID （参考情報）: judge3 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 6,820 KB
testcase_01	AC	1 ms; 6,816 KB
testcase_02	AC	2 ms; 6,820 KB
testcase_03	AC	2 ms; 6,816 KB
testcase_04	AC	2 ms; 6,820 KB
testcase_05	AC	1 ms; 6,820 KB
testcase_06	AC	2 ms; 6,820 KB
testcase_07	AC	28 ms; 8,704 KB
testcase_08	AC	28 ms; 8,704 KB
testcase_09	AC	28 ms; 8,704 KB
testcase_10	AC	30 ms; 8,704 KB
testcase_11	AC	28 ms; 8,704 KB
testcase_12	AC	337 ms; 59,520 KB
testcase_13	AC	335 ms; 59,520 KB
testcase_14	AC	350 ms; 59,648 KB
testcase_15	AC	352 ms; 59,520 KB
testcase_16	AC	343 ms; 59,520 KB
testcase_17	AC	295 ms; 59,648 KB
testcase_18	AC	259 ms; 59,648 KB
testcase_19	AC	273 ms; 59,520 KB
testcase_20	AC	271 ms; 59,520 KB
testcase_21	AC	327 ms; 59,520 KB
testcase_22	AC	345 ms; 59,648 KB

ソースコード

#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";
	}
	
}