活用巧用数据结构

最新推荐文章于 2021-03-03 20:26:19 发布

elims小熊

最新推荐文章于 2021-03-03 20:26:19 发布

阅读量516

点赞数

分类专栏： acm

本文链接：https://blog.csdn.net/a249900679/article/details/50901063

版权

acm 专栏收录该内容

23 篇文章 0 订阅

订阅专栏

C - 活用,巧用数据结构

Time Limit:2000MS Memory Limit:65535KB 64bit IO Format:%I64d & %I64u

Submit Status

Description

Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).

Input

输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.

注意:本题只有一组测试数据,处理到文件结束.

Output

对于每个提问,给出以该字符串为前缀的单词的数量.

Sample Input

       
       
        
        banana
band
bee
absolute
acm

ba
b
band
abc

Sample Output

PS:一开始看到这题时，想到的就是用树，其实就是简单的字典树

代码如下：

PS:代码中的注释是我测试中用的，请忽略

#include<iostream>
#include<string>
using namespace std;

struct T
{
	int sum;
	T *t[26];
	T()
	{
		sum = 0;
		for (int i = 0; i < 26; ++i)
		{
			t[i] = NULL;
		}
	}
}*TNode;

void insert(T *&root, char *s)
{
	int i = 0;
	TNode = root;
	while (s[i] != '\0')
	{
		//cout << s[i] << "   ";
		int j = s[i] - 'a';
		if (TNode->t[j] == NULL)
		{
			//cout << "建新的" << endl;
			TNode->t[j] = new T();
		}
		TNode = TNode->t[j];
		TNode->sum++;
		//cout << "继续下去sum为：" << TNode->sum << endl;
		i++;
	}
}

int find(T *&root, char *s)
{
	int i = 0;
	TNode = root;
	while (s[i] != '\0')
	{
		int j = s[i] - 'a';
		if (TNode->t[j] == NULL)
		{
			return 0;
		}
		else
		{
			TNode = TNode->t[j];
		}
		i++;
	}
	return TNode->sum;
}


int main()
{
	char str[11];
	T *root = new T();
	while (gets_s(str) && str[0])
	{
		insert(root, str);
	}
	while (gets_s(str) && str[0])
	{
		cout << find(root, str) << endl;
	}

	return 0;
}