week15作业（字符串）

A - ZJM 与霍格沃兹（必做）

问题描述

ZJM 为了准备霍格沃兹的期末考试，决心背魔咒词典，一举拿下咒语翻译题
题库格式：[魔咒] 对应功能
背完题库后，ZJM 开始刷题，现共有 N 道题，每道题给出一个字符串，可能是 [魔咒]，也可能是对应功能
ZJM 需要识别这个题目给出的是 [魔咒] 还是对应功能，并写出转换的结果，如果在魔咒词典里找不到，输出 “what?”

Input

首先列出魔咒词典中不超过100000条不同的咒语，每条格式为：

[魔咒] 对应功能

其中“魔咒”和“对应功能”分别为长度不超过20和80的字符串，字符串中保证不包含字符“[”和“]”，且“]”和后面的字符串之间有且仅有一个空格。魔咒词典最后一行以“@END@”结束，这一行不属于词典中的词条。
词典之后的一行包含正整数N（<=1000），随后是N个测试用例。每个测试用例占一行，或者给出“[魔咒]”，或者给出“对应功能”。

Output

每个测试用例的输出占一行，输出魔咒对应的功能，或者功能对应的魔咒。如果在词典中查不到，就输出“what?”

Example

Input

[expelliarmus] the disarming charm
[rictusempra] send a jet of silver light to hit the enemy
[tarantallegra] control the movement of one's legs
[serpensortia] shoot a snake out of the end of one's wand
[lumos] light the wand
[obliviate] the memory charm
[expecto patronum] send a Patronus to the dementors
[accio] the summoning charm
@END@
4
[lumos]
the summoning charm
[arha]
take me to the sky

Output

light the wand
accio
what?
what?

解题思路

这个题因为空间的限制不能直接使用map<string，string>，因此需要通过hash函数来计算每一个字符串的hash值，然后建立hash值到index的映射。hash值我使用的是ll进行存储。seed选取的是31，mod采用的是1e9 + 7（初始时mod设置错误导致一直wrong）。
另外就是字符串的读入，这个还是比较好解决的。

代码

#include<iostream> 
#include<stdio.h>
#include<map>
#include<string>
#include<string.h>
#include<vector>
#include<cmath>
using namespace std;


map<long long int, long long int> s1, s2;
vector<string> p1, p2;


long long int Hash(string s){
    long long int hash=0;
    for(int i=0;i<s.size();i++)
		hash=hash*31+s[i];
    return (hash&0x7fffffff)%1000000007;
}

int main()
{
	string l1, l2;
	char c;
	long long int num1, num2, num;
	num = 0;
	while(true)
	{
		cin>>c;
		if(c=='[')
		{
			getline(cin, l1, ']'); 
			getline(cin, l2, ' '); 
			getline(cin, l2, '\n');
			l1 = '['+l1+']';
			num1 = Hash(l1), num2 = Hash(l2);
			s1[num1] = num, s2[num2] = num;
			p1.push_back(l1), p2.push_back(l2);
			num++;
		}
		else
		{
			getline(cin, l2, '\n');
			break;
		}		
	}
	cin>>num;
	getchar();
	for(int i=0;i<num;i++)
	{
		getline(cin, l1, '\n');
		num1 = Hash(l1);
		if(l1[0]=='[')
		{
			if(s1.find(num1)!=s1.end())
			{
				num2 = s1[num1];
				cout<<p2[num2]<<endl;
			}
			else
				cout<<"what?"<<endl;
		}
		else
		{
			if(s2.find(num1)!=s2.end())
			{
				num2 = s2[num1];
				for(int i=1;i<p1[num2].size()-1;i++)
					cout<<p1[num2][i];
				cout<<endl;
			}
			else
				cout<<"what?"<<endl;
		}
	}
}

B - ZJM 与生日礼物（选做）

问题描述

ZJM 收到了 Q老师 送来的生日礼物，但是被 Q老师 加密了。只有 ZJM 能够回答对 Q老师 的问题，Q老师 才会把密码告诉 ZJM。

Q老师 给了 ZJM 一些仅有 01 组成的二进制编码串, 他问 ZJM：是否存在一个串是另一个串的前缀.

Input

多组数据。每组数据中包含多个仅有01组成的字符串，以一个9作为该组数据结束的标志。

Output

对于第 k 组数据(从1开始标号)，如果不存在一个字符串使另一个的前缀，输出"Set k is immediately decodable"，否则输出"Set k is not immediately decodable"。
每组数据的输出单独一行

Example

Input

01
10
0010
0000
9
01
10
010
0000
9

Output

Set 1 is immediately decodable
Set 2 is not immediately decodable

解题思路

这是一道基本的字典树题目，我们判断是否有前缀可以在插入的时候直接判断，假设当前字符串是S，有两种情况。

1.之前插入串是S的前缀
2.是之前插入串的前缀

对于第一种情况，我们只需要在插入S的过程中，判断路径上是否有结束的位置，如果有，则之前插入的串有S的前缀，否则没有。

对于第二种情况，在S插入到最后一个位置的时候，判断最后一个字母是否已经存在，如果存在，那么S就是其他串的前缀，否则没有。
只要存在上述任一种情况就是输出Set k is not immediately decodable，因此我加了一个全局变量用于进行标记。

注意：要分清对应输出，我初始时就把输出看反了，导致一直wrong，直到最后才看出来

代码

#include<iostream> 
#include<stdio.h>
#include<string>
#include<string.h>
#include<math.h>
using namespace std;

struct Trie{
	static const int N = 1010, charset = 30;
	int tot, root, child[N][charset], flag[N];
	Trie() {
		memset(child, -1, sizeof child);
		memset(flag, 0, sizeof flag);
        root=tot=0;
    }
	void clear() {
		memset(child, -1,sizeof child);
		memset(flag, 0, sizeof flag);
		root=tot=0;
	}
	int insert(string str) {
		int now = root, len = str.size(), status = 0;
		for (int i=0; i<len; i++){
            int x=str[i]-'0';
            if(child[now][x]==-1){
                child[now][x]=++tot;
                flag[now]=0;
            }
            else if(i==len-1 || flag[child[now][x]]) 
				status = 1;
            now=child[now][x];
        }
		flag[now]=1;
		return status;
	}
};

int main()
{
	string s;
	int k = 1, status = 0;
	Trie m;
	while(cin>>s)
	{
		if(s[0]=='9')
		{
			if(!status)
				printf("Set %d is immediately decodable\n", k);
			else
				printf("Set %d is not immediately decodable\n", k);
			m.clear(), status = 0;
			k++;
		}
		else
			status = max(status, m.insert(s));				
	}
}

C - ZJM 与纸条（选做）

问题描述

ZJM 的女朋友是一个书法家，喜欢写一些好看的英文书法。有一天 ZJM 拿到了她写的纸条，纸条上的字暗示了 ZJM 的女朋友 想给 ZJM 送生日礼物。ZJM 想知道自己收到的礼物是不是就是她送的，于是想看看自己收到的礼物在纸条中出现了多少次。

Input

第一行输入一个整数代表数据的组数

每组数据第一行一个字符串 P 代表 ZJM 想要的礼物, 包含英语字符 {‘A’, ‘B’, ‘C’, …, ‘Z’}, 并且字符串长度满足 1 ≤ |P| ≤ 10,000 (|P| 代表字符串 P 的长度).

接下来一行一个字符串 S 代表 ZJM 女朋友的纸条, 也包含英语字符 {‘A’, ‘B’, ‘C’, …, ‘Z’}, 满足 |P| ≤ |S| ≤ 1,000,000.

Output

输出一行一个整数代表 P 在 S中出现的次数.

Example

Input

3
BAPC
BAPC
AZA
AZAZAZA
VERDI
AVERDXIVYERDIAN

Output

1
3
0

解题思路

KMP算法
作用
判断字符串P是否为字符串S的子串即字符串匹配问题

暴力解法
记n=len(S),m=len§，枚举i=0,1,…,n-m，将S[i,i+m]与P比较，复杂度为O(nm)

KMP优化
上述暴力算法中每次比较失败，后移一位继续比较，KMP跳过那些绝不可能成功的字符串比较，尽量减少比较的趟数。

核心：next数组
Next数组
• “跳过不可能成功的尝试” • Next 数组定义
• Next[i] 的值为使得 P[0…i] 这一个子串的 K-真前缀 等于 K-真后缀的最大的K

每次匹配失败后，若再P[r]匹配失败，则对于P[0…r-1]这一段前next[r-1]个字符一定与后next[r-1]个字符相同，则可用长度next[r-1]的前缀替代当前比较的后缀，让P[next[r-1]]这个字符对准刚刚匹配失败的地方进行下一次匹配。

代码

#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;

char ptr[10005];
char str[1000005];
int Next[10005];

void getnext(char*pstr)
{
	int len = strlen(pstr);
	Next[0]=0;
	for(int i=1,j=0;i<len;i++)
	{
		while(j&&pstr[i]!=pstr[j])
			j=Next[j-1];
		if(pstr[i]==pstr[j])
			j++;
		Next[i]=j;
	}
}
int KMP(char*str, char*ptr)
{
	int len1=strlen(str);
	int len2=strlen(ptr);
	int cnt=0;
	getnext(ptr);
	for(int i=0,j=0;i<len1;i++)
	{
		while(j&&str[i]!=ptr[j]) 
			j=Next[j-1];
		if(str[i]==ptr[j])
			j++;
		if(j==len2)
		{
			cnt++;
			j=Next[j-1];
		}
	}
	return cnt;
}
int main() {
	int n, num;
	scanf("%d",&n);
	while(n--)
	{
		scanf("%s",ptr);
		scanf("%s",str);
		num=KMP(str,ptr);
		printf("%d\n",num);
	}
	return 0;
}