BZOJ 1398: Vijos1382寻找主人 Necklace 最小表示法

1398: Vijos1382寻找主人 Necklace

Time Limit: 5 Sec  Memory Limit: 128 MB
Submit: 326  Solved: 138
[Submit][Status][Discuss]

Description

给定两个项链的表示，判断他们是否可能是一条项链。

Input

输入文件只有两行，每行一个由0至9组成的字符串，描述一个项链的表示（保证项链的长度是相等的）。

Output

如果两条项链不可能同构，那么输出’No’，否则的话，第一行输出一个’Yes’

第二行输出该项链的字典序最小的表示。 设L = 项链长度，L <= 1000000。

Sample Input

2234342423
2423223434

Sample Output

Yes
2234342423

本来想写KMP

但要输出最小表示

那就写最小表示法吧。。。

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=10*x+ch-'0';ch=getchar();}
	return x*f;
}
inline void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=1000100;
int a[N<<1],b[N<<1];

int main()
{
	static char s[N];
	scanf("%s",s+1);int n=strlen(s+1);
	for(int i=1;i<=n;i++)a[i+n]=a[i]=s[i]-'0';
	scanf("%s",s+1);
	for(int i=1;i<=n;i++)b[i+n]=b[i]=s[i]-'0';
	int k=0,i=1,j=2;
	while(i<=n&&j<=n&&k<=n)
	{
		int t=a[i+k]-a[j+k];
		if(!t)k++;
		else {if(t<0)j+=k+1;else i+=k+1;k=0;if(i==j)i++;}
	}
	int ansa=min(i,j);
	k=0;i=1;j=2;
	while(i<=n&&j<=n&&k<=n)
	{
		int t=b[i+k]-b[j+k];
		if(!t)k++;
		else {if(t<0)j+=k+1;else i+=k+1;k=0;if(i==j)i++;}
	}
	int ansb=min(i,j);
	for(int i=0;i<n;i++)if(a[i+ansa]!=b[i+ansb]){puts("No");return 0;}
	puts("Yes");for(int i=0;i<n;i++)putchar(a[i+ansa]+'0');puts("");
	return 0;
}
/*
2234342423
2423223434

Yes
2234342423
*/