Description

題目鏈接

Solution

這裏用類似hash的方法將判斷2個矩陣是否相同的時間降為O(m)，總時間復雜度為O(m³)

對於一行字符串s[i]，它的hash值為∑(j*s[i][j])%mo,(1≤j≤m),mo為一個大質數

理論上來說，可能會出現2個不同字符串hash值相等，但cf上測是沒有

Code

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;

const int mo=997;
int n,m,AA[2010],BB[210][2010];
char A[2010][210],B[210][2010],tmp[210][210];

int main(){
	scanf("%d%d\n",&n,&m);
	for(int i=1;i<=n;++i){
		scanf("%s\n",A[i]+1);
		for(int j=1;j<=m;++j) AA[i]=(AA[i]+(A[i][j]-97+1)*j)%mo;
	}
	for(int i=1;i<=m;++i){
		scanf("%s\n",B[i]+1);
		for(int j=1;j+m-1<=n;++j){
			int r=j+m-1;
			for(int k=j,kk=1;k<=r;++k,++kk) BB[i][j]=(BB[i][j]+(B[i][k]-97+1)*kk)%mo;
		}
	}
	for(int q=1;q+m-1<=n;++q)
		for(int g=1;g+m-1<=n;++g){
			bool b=1;int tmp=0;
			for(int i=1,j=q;i<=m;++i,++j) if(AA[j]!=BB[i][g]){b=0;break;}
			if(b){
				printf("%d %d\n",q,g);
				return 0;
			}
			
		}
	return 0;
}

[Codeforces958A2]Death Stars (medium)(字符串+hash)