P1628 合并序列 题解

题目描述

有 $N$ 个单词和字符串 $T$，按字典序输出以字符串 $T$ 为前缀的所有单词。

输入格式

输入文件第一行包含一个正整数 $N$；

接下来 $N$ 行，每行一个单词，长度不超过 $100$；

最后一行包含字符串 $T$。

输出格式

按字典序升序输出答案。

样例

样例输入

6
na
no
ki
ki
ka
ku
k

样例输出

ka
ki
ki
ku

数据范围与提示

对于 $60\%$ 的数据，满足 $1\le N\le 1000$；

对于 $100\%$ 的数据，满足 $1\le N\le 100000$ 且所有字符均为小写字母。

题目传送门（题目源自洛谷）

思路及部分实现

首先考虑用什么数据类型储存 $n$ 个字符串和 $k$，由于题目要求输出以 $k$ 为前缀的字符串，众所不太周知 $string$ 类型有一个函数 $find$ 可以查找某个字符串是否存在，如果存在返回值为 $0$，否则返回值为 $1$。简直就是为这道题而存在的函数啊！代码实现如下：

for(int i=1;i<=n;i++) 
	if(a[i].find(k)==0)
		//如果能找到 k就输出
		cout<<a[i]<<endl;

如果你不嫌麻烦的话，也可以自己手写 $find$ 函数。代码实现如下：

bool find(string str){
	int len=k.size();
	if(len<str.size()) return true;
	for(int i=0;i<len;i++)
		if(k[i]!=str[i]) return true;
	return false;
}
for(int i=1;i<=n;i++) 
	if(find(a[i])==false)
		cout<<a[i]<<endl;

就在你以为题目已经完成的时候，出题人却露出了险恶的笑容。

按字典序升序输出答案。

解决字典序升序的问题对 $string$ 类型来说很简单，直接用函数 $sort$ 就可以完美解决，且不影响后面的操作。代码实现如下：

sort(a+1,a+n+1);

完整代码

第一种写法，即使用函数库查找 $k$。

#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
	int n;
	string k,a[100001];
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		cin>>a[i];
	cin>>k;
	sort(a+1,a+n+1);
	for(int i=1;i<=n;i++) 
		if(a[i].find(k)==false)
			cout<<a[i]<<endl;
	return 0;
}

第二种写法，即手写 $find$ 函数。

#include<iostream>
#include<algorithm>
using namespace std;
bool find(string str){
	int len=k.size();
	if(len<str.size()) return true;
	for(int i=0;i<len;i++)
		if(k[i]!=str[i]) return true;
	return false;
}
int main()
{
	int n;
	string k,a[100001];
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		cin>>a[i];
	cin>>k;
	sort(a+1,a+n+1);
	for(int i=1;i<=n;i++) 
		if(find(a[i])==false)
			cout<<a[i]<<endl;
	return 0;
}