码迷,mamicode.com
首页 > 其他好文 > 详细

统计难题

时间:2016-04-28 20:12:11      阅读:238      评论:0      收藏:0      [点我收藏+]

标签:

【题目描述】

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

【输入描述】

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

【输出描述】

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

【输入样例】

banana
band
bee
absolute
acm
ba
b
band
abc

【输出样例】

2
3
1
0
源代码:

#include<iostream>
using namespace std;
string n;
int s,num(0);
struct treetype
{
    char t; 
    int next,deep,same,son;
}tree[1000001];
void x1(string s) 
{
    int m=s.size()-1,k(0);
    while (1)
    {
        if (tree[k].deep==m) 
          break; //增点完毕。
        if (tree[k].next==-1) //没有兄弟,直接增加儿子。
        {
            num++;
            tree[num].deep=tree[k].deep+1;
            tree[num].t=s[tree[num].deep];
            tree[num].same=-1;
            tree[num].next=-1;
            tree[num].son=1;
            tree[k].next=num; //终于有儿子了。
            k=num;
        }
        else //有兄弟,遍历兄弟。
        {
            k=tree[k].next;
            while (tree[k].t!=s[tree[k].deep]&&tree[k].same!=-1) //找与自己相同的兄弟。
              k=tree[k].same;
            if (tree[k].t==s[tree[k].deep]) //找到了,当兄弟的儿子。
              tree[k].son++;
            if (tree[k].t!=s[tree[k].deep]) //没找到,还当那个谁的儿子。
            {
                num++;
                tree[num].deep=tree[k].deep;
                tree[num].t=s[tree[num].deep];
                tree[num].same=-1;
                tree[num].next=-1;
                tree[num].son=1;
                tree[k].same=num; //又来了个兄弟。
                k=num;
            }
        }
    }
}
int x2(string s)
{
    int m=s.size()-1,k(0);
    while (1)
    {
        if (tree[k].deep==m) //前缀存在,返回儿子们的数量值。
          return tree[k].son;
        if (tree[k].next==-1) //没前缀,自然为0。
          return 0;
        k=tree[k].next; 
        while (tree[k].t!=s[tree[k].deep]&&tree[k].same!=-1) //找与自己相同的兄弟。
          k=tree[k].same;
        if (tree[k].t!=s[tree[k].deep]) //没找到与自己相同的兄弟,自然也为0。
          return 0;
    }
}
int main() //字典树。
{
    tree[0].deep=0;
    tree[0].next=-1;
    tree[0].same=-1;
    tree[0].son=1; //已增加的点必定是它自己的儿子。
    cin>>s;
    for (int a=1;a<=s;a++)
    {
        cin>>n;
        n=" "+n; //使字符串首位为1。
        x1(n);
    }
    cin>>s;
    for (int a=1;a<=s;a++)
    {
        cin>>n;
        n=" "+n;
        printf("%d\n",x2(n));
    }
    return 0;
}

统计难题

标签:

原文地址:http://www.cnblogs.com/Ackermann/p/5443967.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!