CodeupKMP算法：P2 统计单词数

问题 B: P2 统计单词数

时间限制: 1 Sec  内存限制: 128 MB
提交: 177  解决: 52
[提交][状态][讨论版][命题人:外部导入]

题目描述

统计单词数 (stat.cpp/c/pas) 
一般的文本编辑器都有查找单词的功能，该功能可以快速定位特定单词在文章中的位置，有的还能统计出特定单词在文章中的次数。

现在，请你编程实现这一功能，具体要求是：给定一个单词，请你输出它在给定的文章中出现的次数和第一次出现的位置。注意：匹配单词时，不区分大小写，但要求完全匹配，即单词必须与文章中某一独立残次在不区分大小写的情况下完全相同（参见样例1），如果给定单词仅是文章中某一单词的一部分则不算匹配（参见样例2）。

输入

输入文件共2行。

第1行为一个字符串，其中只包含字母，表示给定单词；

第2行为一个字符串，其中只可能包含字母和空格，表示给定的文章。

输出

只有1行，如果在文章中找到给定单词则输出两个整数，两个整数之间用一个空格隔开，分别是单词在文章中出现的次数和第一次出现的位置（即在文章中第一次出现时，单词首字母在文章中的位置，位置从0开始）；如果单词在文章中没有出现，则直接输出一个整数-1。

样例输入

输入样例1：
To
to be or not to be is a question
输入样例2：
to
Did the Ottoman Empire lose its power at that time

样例输出

输出样例1：
2 0
输出样例2：
-1

提示

1<=单词长度<=10。

1<=文章长度<=10,000,000。

NOIP2011 普及组 stat

---

#include<bits/stdc++.h>
using namespace std;
const int maxn=300;
int nextval[maxn],pos;
bool flag=false;
void Change(string &str){
	for(int i=0;i<str.length();i++){
		if(str[i]>='A' && str[i]<='Z') 
			str[i]+=32;
	}
}
void getNextval(string str){
	int j=-1;
	nextval[0]=-1;
	for(int i=1;i<str.length();i++){
		while(j!=-1&&str[i]!=str[j+1]){
			j=nextval[j];
		}
		if(str[i]==str[j+1]){
			j++;//令j指向原next[i]的位置 
		}
		if(j==-1||str[i+1]!=str[j+1]){
			nextval[i]=j;
		}else{
			nextval[i]=nextval[j];
		}
	}
} 
//KMP算法统计str2在str1中出现的次数 
int KMP(string str1,string str2){
	int ans=0,j=-1;
	int n=str1.length(),m=str2.length();
	for(int i=0;i<n;i++){
		if(j!=-1&&str1[i]!=str2[j+1]){
			j=nextval[j];
		}
		if(str1[i]==str2[j+1]) j++;
		if(j==m-1){
			if(i-m>=0&&i<n-1&&str1[i+1]==' ' && str1[i-m]==' '||i-m+1==0&&i<n-1&&str1[i+1]==' '||i-m>=0&&i==n-1&&str1[i-m]==' '||i-m+1==0&&i==n-1){
			//匹配串在str2的中部 || 匹配串在str2的头部 ||匹配串在str2的尾部 || 匹配串=str2	
				ans++;
				if(flag==false){
					pos=i-m+1;
					flag=true;
				}
			}
			j=nextval[j];
		}
	}
	return ans;	
}

int main(){
	int ans;
	string str1,str2;
	while(getline(cin,str1)){
		Change(str1);
		getNextval(str1);
		ans=0;
		getline(cin,str2);
		Change(str2);
		flag=false;
		ans+=KMP(str2,str1);
		if(ans==0) cout<<"-1\n";
		else cout<<ans<<" "<<pos<<endl;
	}
	return 0;
}