結果

問題 No.515 典型LCP
コンテスト
ユーザー myanta
提出日時 2017-05-09 09:17:39
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
MLE  
実行時間 -
コード長 2,045 bytes
コンパイル時間 678 ms
コンパイル使用メモリ 57,028 KB
実行使用メモリ 815,864 KB
最終ジャッジ日時 2024-09-14 19:56:43
合計ジャッジ時間 4,053 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 2
other MLE * 1 -- * 14
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:123:30: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  123 |                         scanf("%s", p);
      |                         ~~~~~^~~~~~~~~
main.cpp:137:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  137 |                 scanf("%d%lld%lld", &m, &x, &d);
      |                 ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

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