#172340 (C++14) No.515 典型LCP

• 問題: No.515 典型LCP
• ユーザー: ciel
• 提出日時: 2017-05-06 16:27:44
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: TLE
• 実行時間: -
• コード長: 1,699 bytes
• コンパイル時間: 1,421 ms
• コンパイル使用メモリ: 92,476 KB
• 実行使用メモリ: 9,616 KB
• 最終ジャッジ日時: 2024-09-14 13:55:56
• 合計ジャッジ時間: 10,270 ms
• ジャッジサーバーID （参考情報）: judge1 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	TLE	-
testcase_01	TLE	-
testcase_02	WA	-
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	216 ms; 5,376 KB
testcase_06	AC	346 ms; 5,376 KB
testcase_07	AC	244 ms; 5,376 KB
testcase_08	AC	399 ms; 5,376 KB
testcase_09	AC	358 ms; 5,376 KB
testcase_10	AC	562 ms; 5,376 KB
testcase_11	AC	560 ms; 5,376 KB
testcase_12	AC	563 ms; 5,376 KB
testcase_13	WA	-
testcase_14	AC	24 ms; 5,376 KB
testcase_15	AC	151 ms; 5,376 KB
testcase_16	AC	146 ms; 5,376 KB

ソースコード

#pragma GCC optimize("O3")
#pragma GCC target("avx")
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
using namespace std;

#define INF (1<<29)
class RMQ{
	int size;
	vector<int> minval;
	vector<int> maxval;
	pair<int,int> _query(int a,int b,int k,int l, int r){
		if(r<=a||b<=l)return make_pair(INF,-INF);
		if(a<=l&&r<=b)return make_pair(minval[k],maxval[k]);
		else{
			pair<int,int> vl=_query(a,b,k*2+1,l,(l+r)/2);
			pair<int,int> vr=_query(a,b,k*2+2,(l+r)/2,r);
			return make_pair(min(vl.first,vr.first),max(vl.second,vr.second));
		}
	}

public:
	RMQ(int k){init(k);}
	void init(int k){
		for(size=1;size<k;)size<<=1;
		//size=1<<(32-__builtin_clz(k));
		minval=vector<int>(2*size-1,INF);
		maxval=vector<int>(2*size-1,-INF);
	}

	void update(int k,int t){
		k+=size-1;
		minval[k]=maxval[k]=t;
		while(k>0){
			k=(k-1)/2;
			minval[k]=min(minval[k*2+1],minval[k*2+2]);
			maxval[k]=max(maxval[k*2+1],maxval[k*2+2]);
		}
	}

	pair<int,int> query(int a,int b){
		return _query(a,b,0,0,size);
	}
};

int N,D,I,J;
long long X;
void iterate(){
	I=X/~-N;
	J=X%~-N;
	if(I>J)swap(I,J);
	else J++;
	X=(X+D)%((long long)N*~-N);
}
int calc(const string &s,const string &t){
	int l=min(s.size(),t.size()),i=0;
	for(;i<l&&s[i]==t[i];i++);
	return i;
}
int main(){
	int M,R=0;
	cin>>N;
	vector<pair<string,int>>v(N);
	vector<int>rev(N);
	for(int i=0;i<N;i++)cin>>v[i].first,v[i].second=i;
	cin>>M>>X>>D;
	sort(v.begin(),v.end());
	for(int i=0;i<N;i++)rev[v[i].second]=i;
	RMQ rmq(N);
	for(int i=0;i<N-1;i++)rmq.update(i+1,calc(v[i].first,v[i+1].first));
	for(;M--;){
		iterate();
		int i=rev[I],j=rev[J];
		if(i>j)swap(i,j);
		R+=rmq.query(i+1,j+1).first;
	}
	cout<<R<<endl;
}