結果

問題 No.515 典型LCP
ユーザー myantamyanta
提出日時 2017-05-09 09:17:39
言語 C++11
(gcc 11.4.0)
結果
MLE  
実行時間 -
コード長 2,045 bytes
コンパイル時間 636 ms
コンパイル使用メモリ 60,404 KB
実行使用メモリ 814,028 KB
最終ジャッジ日時 2023-10-12 21:39:27
合計ジャッジ時間 4,812 ms
ジャッジサーバーID
(参考情報)
judge15 / judge14
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 MLE -
testcase_01 -- -
testcase_02 -- -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<vector>
#include<algorithm>


using namespace std;
using ll=long long;
using pcpi=pair<char*,int>;


vector<pcpi> s;
vector<int> lcp;
vector<int> idx;
vector<vector<int> > dp;


void swap(int& a, int& b)
{
	int c;
	c=a;
	a=b;
	b=c;
}


int calc_lcp(const pcpi& a, const pcpi& b)
{
	int k;
	const char *p, *q;

	p=a.first;
	q=b.first;

	int idx1, idx2;
	if(a.second>b.second)
	{
		idx1=a.second;
		idx2=b.second;
	}
	else
	{
		idx1=b.second;
		idx2=a.second;
	}
	if(dp[idx1].size()>idx2 && dp[idx1][idx2]>=0) return dp[idx1][idx2];

	for(k=0;p[k] && p[k]==q[k];k++)
		;
	return k;
}


bool cmp(const pcpi& a, const pcpi& b)
{
	int k;
	const char *p, *q;

	p=a.first;
	q=b.first;
	for(k=0;p[k] && p[k]==q[k];k++)
		;

	int idx1, idx2;
	if(a.second>b.second)
	{
		idx1=a.second;
		idx2=b.second;
	}
	else
	{
		idx1=b.second;
		idx2=a.second;
	}
	if(dp[idx1].size()<=idx2) dp[idx1].resize(idx2+1, -1);
	dp[idx1][idx2]=k;
	return p[k]<q[k];
}


void min_u(int& m, int v)
{
	if(m>v) m=v;
}


int get_lcp(int i, int j)
{
	int idx1=idx[i];
	int idx2=idx[j];
	if(idx1>idx2) swap(idx1, idx2);

	int ret=lcp[idx1];
	for(int i=idx1+1;i<idx2;i++)
	{
		min_u(ret, lcp[i]);
		if(!ret) break;
	}
	return ret;
}


int main(void)
{
	vector<char> buf;
	char *p;
	int i, j, k, m;
	ll x, result, n, d;

	while(scanf("%lld", &n)==1)
	{
		dp.clear();
		dp.resize(n);
		s.resize(n);
		lcp.resize(n);
		idx.resize(n);
		buf.resize(800000+100000);
		p=&buf[0];
		for(i=0;i<n;i++)
		{
			s[i]=make_pair(p, i);
			scanf("%s", p);
			p+=strlen(p)+1;
		}
		sort(s.begin(), s.end(), cmp);
		for(i=0;i+1<n;i++)
		{
			lcp[i]=calc_lcp(s[i], s[i+1]);
		}
		for(i=0;i<n;i++)
		{
			idx[s[i].second]=i;
		}

		result=0;
		scanf("%d%lld%lld", &m, &x, &d);
		for(k=0;k<m;k++)
		{
			i=(x/(n-1))+1;
			j=(x%(n-1))+1;
			if(i>j) swap(i, j);
			else j++;
			x=(x+d)%(n*(n-1));
			result+=get_lcp(i-1, j-1);
//			printf("k=%d  x=%d  i=%d  j=%d  %lld\n", k, x, i, j, result);
		}
		printf("%lld\n", result);
	}

	return 0;
}

0