OJ P1361噩梦

这题是KMP算法模板。先总结下KMP：

对于a，b两个字符串，如何用O（n）的时间求出b是否是a的字串：

对比着看：

一般算法  O(m*n）：

      对a有指针i，对b有指针j。i不断自增的同时不断使a【i..i+j】与b比较。

传说中的KMP：

      首先应先预处理出p数组记录当指针为j时p【j】记录b串中前j个字符中前j'个字符与后j'个字符相等（即对于指针j，b【1..j’】==b【j’..j】）

如何处理等会再说。。。先假设以处理好p数组

      此时对a串按KMP搜索b，方法是用指针i扫一遍a串，同时先给j赋初值为-1（因为字符串下标开始是0）后不断向后比较。但当a【i】≠b【j】时，j退到p【j】，然后j++，i++  ……

 

 然后这题大概就完了。。。照着模板稍稍一改就A了。

 

#include<iostream>
#include<iomanip>
#include<cstring>
#include<string>
#include<algorithm>
#include<cstdio>
using namespace std;
int m,n;
string a,b;
int p[210][210];
long long summ=0;
struct fearthings
{
	string fth;
	int WV;	
}fthings[210];
void init()
{
	cin>>n>>m;
	for(int i=1;i<=m;i++)
		cin>>fthings[i].fth;
	for(int i=1;i<=m;i++)
		cin>>fthings[i].WV;
}//读入数据

void deal_p(int i)//预处理所有p数组，用i参数控制行（第i个p）
{
	
	p[i][0]=-1;  //cpp的string 的第一个元素下标是0，所以边界设为-1；
	int j=-1;
	int lenn=fthings[i].fth.size();
	for(int i2=1;i2<lenn;i2++)
	{
		while ((j>=0)&&(fthings[i].fth[j+1]!=fthings[i].fth[i2]))   j=p[i][j];
		if (fthings[i].fth[j+1]==fthings[i].fth[i2])  j++;
		p[i][i2]=j;
	}
}

void KMP(int bv,int line,string b)
{
	int j=-1;  //边界因为string的第1个元素下标是0
	for(int i=0;i<a.size();i++)
	{
		while ((j>-1)&&(b[j+1]!=a[i]))   j=p[line][j];
		if (b[j+1]==a[i])  j++;
		if (j==(b.size()-1)) //找到匹配，当然可以继续找下一个匹配
		{
			int mul=b.size();
			summ+=bv*(i-mul+2);//注意这里：原代码模板此处求的是位置，但我们这题要求的是权值（即第一个字母出现的第一个位置）
			j=p[line][j];//最后的j:=P[j]是为了让程序继续做下去，因为我们有可能找到多处匹配
		}
	}			
}

int main()
{
	//freopen("aa.in","r",stdin);
	init();
	
	memset(p,0,sizeof(p));//先初始化整个p数组
	for(int i=1;i<=m;i++)
		deal_p(i);
	

	for(int i=1;i<=n;i++)
	{
		cin>>a;//每次读入一天晚上的梦
		for(int i2=1;i2<=m;i2++)//对所有害怕的事依次枚举做KMP
		{
			b=fthings[i2].fth;
			KMP(fthings[i2].WV,i2,b);			
		}
	}
	cout<<summ<<endl;//输出黑暗总和
return 0;
}