codeforces #336 E. Marbles （字符串hash或者kmp匹配）

最新推荐文章于 2025-02-13 20:01:46 发布

原创最新推荐文章于 2025-02-13 20:01:46 发布 · 955 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#字符串hash #kmp

ACM-数据结构专栏收录该内容

87 篇文章

订阅专栏

博客探讨了Codeforces竞赛中关于E. Marbles问题的解决方法，主要涉及如何运用字符串Hash和KMP匹配算法来分析两个小球在通道中的移动情况。虽然初始思路困难，但通过阅读分享的博客文章，可以找到解决问题的灵感。

题目：http://codeforces.com/contest/608/problem/E

题意：有两个通道，在每个通道的起点分别放一个小球，然后问是否存在一系列指令使得两个小球最后都在终点。

若小球碰壁，则保持原来的位置不变。

分析：完全没思路，，，看的这篇博客，http://blog.csdn.net/snowy_smile/article/details/50402283

Hash代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
typedef unsigned long long ULL;
const LL INF = 1E9+9;
const int MI = ~0u>>1;

int main()
{
	int n;
	string s1,s2;
	cin>>n>>s1>>s2;
	n--;
	for(int i=0;i<n;i++)
	{
		switch(s2[i])
		{
			case 'N':s2[i]='S';break;
			case 'S':s2[i]='N';break;
			case 'W':s2[i]='E';break;
			case 'E':s2[i]='W';break;
		}
	}
	ULL v1=0,v2=0;
	const ULL seed=131;
	ULL wt=1;
	for(int i=n-1;i>=0;i--,wt*=131)
	{
		v1=v1+(s1[i]-'a')*wt;
		v2=v2*seed+s2[i]-'a';
		if(v1==v2)
		{
			printf("NO");
			return 0;
		}
	}
	printf("YES");
	return 0;
}

KMP代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
typedef unsigned long long ULL;
const LL INF = 1E9+9;
const int MI = ~0u>>1;

const int maxn = 1111111;
int fail[maxn];

void getnext(char s[])
{
	fail[0]=-1;
	int len=strlen(s);
	int k=-1,j=0;
	while(j<len)
	{
		if(k==-1 || s[k]==s[j])
		{
			k++;
			j++;
			if(s[k]==s[j])
				fail[j]=fail[k];
			else
				fail[j]=k;
		}
		else
			k=fail[k];
	}
}

int main()
{
	int n;
	char s1[maxn],s2[maxn];
	scanf("%d%s%s",&n,s1,s2);
	n--;
	for(int i=0;i<n;i++)
	{
		switch(s2[i])
		{
			case 'N':s2[i]='S';break;
			case 'S':s2[i]='N';break;
			case 'W':s2[i]='E';break;
			case 'E':s2[i]='W';break;
		}
	}
	for(int i=0,j=n-1;i<j;i++,j--)
		swap(s2[i],s2[j]);
	getnext(s2);
	for(int i=0,k=0;i<n;i++,k++)
	{
		while(k!=-1 && s1[i]!=s2[k])
			k=fail[k];
		if(k>=0 && i==n-1)
		{
			printf("NO");
			return 0;
		}
	}
	printf("YES");
	return 0;
}