HDU：1251 统计难题

统计难题

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 131070/65535 K (Java/Others)
Total Submission(s): 35579    Accepted Submission(s): 13301

Problem Description

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

 

Input

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

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

 

Output

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

 

Sample Input

  
  

   
   banana
band
bee
absolute
acm

ba
b
band
abc
  
  

 

Sample Output

  
  

   
   2
3
1
0
  
  

 

解题思路：

最开始的时候，没有用map写，直接用strstr串查找函数求解的，以为只要串查找函数返回的地址与目标串的地址一样，就让以该字符串为前缀的单词数量++，

但是这样求解的时候，每次输入一个单词，要在提供的每一个字符串中查找，果断超时了，所以做题头脑还是不能太简单的。

比较方便的方法是用stl中的map来求解这道题目。map<string,int>

最初输入banana时str是  b a n a n a \0

映射：

banana 1

然后将逐渐缩短字符串的长度，b a n a n \0

映射

banan 1

在通过改变\0的位置缩短字符串长度，b a n a \0......

方法1：

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

char str[20];
int main()
{
    map<string,int>Map;//定义一个映射
    while(gets(str))//接收一串字符串
    {
        int len = strlen(str);//测串长
        if(len == 0)//串长为0，结束循环
            break;
        for(int i = len; i > 0; i--)//从字符串的最后一位开始，每一次左移一个字符的位置，将该字符位置赋为‘\0‘,形成新的字符串
        {
            str[i] = '\0';
            Map[str]++;
        }
    }
    while(~scanf("%s",str))
    {
        printf("%d\n",Map[str]);//最后输出str中对应的int即可
    }
    return 0;
}

方法2：用字典树做。

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

using namespace std;

const int allSon = 26;
const int maxn = 500000;
char word[15];
int node[maxn][allSon];  ///字典树节点
int num[maxn],id;
void insertWord() {
    int p = 0;  //根节点是0
    int i = 0;
    while(word[i] != '\0') {
        int lowerCase = word[i]-'a';
        if(node[p][lowerCase] == -1) {
            memset(node[id],-1,sizeof(node[id]));
            num[id] = 0;
            node[p][lowerCase] = id++;
        }
        p = node[p][lowerCase];
        num[p]++;
        i++;
    }
}
int findWord() {
    int p = 0,i = 0;
    while(word[i] != '\0') {
        int lowerCase = word[i]-'a';
        if(node[p][lowerCase] == -1) {
            return 0;
        }
        p = node[p][lowerCase];
        i++;
    }
    return num[p];
}
int main() {
    memset(node[0],-1,sizeof(node[0]));
    num[0] = 0;
    id = 1;
    while(gets(word)) {
        int len = strlen(word);
        if(len == 0) {
            break;
        }
        else {
            insertWord();
        }
    }
    while(~scanf("%s",word)) {
        int ans = findWord();
        printf("%d\n",ans);
    }
    return 0;
}