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

• 問題: No.515 典型LCP
• ユーザー: ciel
• 提出日時: 2017-05-06 16:56:04
• 言語: C++14 (gcc 13.2.0 + boost 1.83.0)
• 結果: TLE (最新)
  AC
  (最初)
• 実行時間: -
• コード長: 1,538 bytes
• コンパイル時間: 2,114 ms
• コンパイル使用メモリ: 103,264 KB
• 実行使用メモリ: 8,208 KB
• 最終ジャッジ日時: 2023-10-12 15:09:50
• 合計ジャッジ時間: 13,466 ms
• ジャッジサーバーID （参考情報）: judge14 / judge13

入力	結果	実行時間; 実行使用メモリ
testcase_00	TLE	-
testcase_01	TLE	-
testcase_02	AC	856 ms; 4,436 KB
testcase_03	AC	1 ms; 4,352 KB
testcase_04	AC	2 ms; 4,352 KB
testcase_05	AC	257 ms; 4,356 KB
testcase_06	AC	429 ms; 4,352 KB
testcase_07	AC	271 ms; 4,360 KB
testcase_08	AC	522 ms; 4,352 KB
testcase_09	AC	498 ms; 4,352 KB
testcase_10	AC	770 ms; 4,352 KB
testcase_11	AC	775 ms; 4,376 KB
testcase_12	AC	774 ms; 4,352 KB
testcase_13	AC	389 ms; 4,348 KB
testcase_14	AC	12 ms; 4,352 KB
testcase_15	AC	146 ms; 4,352 KB
testcase_16	AC	139 ms; 4,348 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;
	int _query(int a,int b,int k,int l, int r){
		if(r<=a||b<=l)return INF;
		if(a<=l&&r<=b)return minval[k];
		else{
			int vl=_query(a,b,k*2+1,l,(l+r)/2);
			int vr=_query(a,b,k*2+2,(l+r)/2,r);
			return min(vl,vr);
		}
	}

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);
	}

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

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

int N,I,J;
long long X,D;
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(){
	cin.tie(0);
	ios::sync_with_stdio(false);
	int M;
	long long R=0;
	cin>>N;
	vector<string>v(N);
	vector<int>tbl(N),rev(N);
	for(int i=0;i<N;i++)cin>>v[i],tbl[i]=i;
	cin>>M>>X>>D;
	sort(tbl.begin(),tbl.end(),[&](int i,int j)->bool{
		return v[i]<v[j];
	});
	for(int i=0;i<N;i++)rev[tbl[i]]=i;
	RMQ rmq(N);
	for(int i=0;i<N-1;i++)rmq.update(i+1,calc(v[tbl[i]],v[tbl[i+1]]));
	for(;M--;){
		iterate();
		int i=rev[I],j=rev[J];
		if(i>j)swap(i,j);
		R+=rmq.query(i+1,j+1);
	}
	cout<<R<<endl;
}