HDU_1251 统计难题 【字典树||map】

题目链接

题目描述

就是在给定的单词中求前缀的个数

解题思路

字典树问题，统计前缀的个数。亲试用map也可以过。
不知道为什么，做字典树额题的时候，用g++ 一般都会超内存，改成c++就可以过。感觉这道题很有邪性，静态的动态数组，不是时间超限就是答案错误。改成静态的就过了。不过也是通过这道题尝试了动态的字典树构建，也是种收获吧。再有就是读回车结束，也是很少遇到，上来就是个措手不及，好在最后悠然解决。

代码部分

动态字典树

#include <iostream>
#include <sstream>
#include <stdlib.h>
using namespace std;
const int en = 26;
const int maxn = 15;
struct Trie
{
    int cnt;
    Trie *nex[en];
};
int top;
inline int idx(char c)
{
    return c - 'a';
}
inline Trie * CreatTrie()
{
    Trie *p = (Trie *)(malloc)(sizeof(Trie));
    p -> cnt = 0;
    for(int i = 0; i< en; ++ i) p -> nex[i] = NULL;
    return p;
}
inline Trie * Init()
{
    top = 0;
    return CreatTrie();
}
void Insert(Trie *root, string s)
{
    Trie *p = root;
    for(int i = 0; i < s.size(); ++ i)
    {
        int temp = idx(s[i]);
        if(p -> nex[temp] == NULL) p -> nex[temp] = CreatTrie();
        p = p -> nex[temp];
        p -> cnt ++;
    }
    //p -> cnt --;
}
int Search(Trie *root, string s)
{
    Trie *p = root;
    for(int i = 0; i < s.size(); ++ i)
    {
        int temp = idx(s[i]);
        if(p -> nex[temp] == NULL) return 0;
        p = p -> nex[temp];
    }
    return p -> cnt;
}
int main()
{
    ios::sync_with_stdio(false);
    string s;
    Trie *root = Init();
    while(getline(cin, s, '\n') && s.size() != 0) Insert(root, s);
    while(cin >> s) cout << Search(root, s) << endl;
    return 0;
}

---

map

#include <iostream>
#include <cstring>
#include <string.h>
#include <map>
using namespace std;
int main()
{
    string c;
    int i;
    map <string, int> m;
    while(getline(cin, c, '\n') && c.size() != 0)
    {
        for(i = c.size(); i > 0; -- i)
            m[c.substr(0,i)] ++;
    }
    string s;
    while(cin >> s)
    {
        cout << m[s] << endl;
    }
    return 0;
}

---

静态字典树

代码不过不了，不知道出了什么状况，如果有巨巨知道求指正。

#include <bits/stdc++.h>
using namespace std;
const int en = 26;
const int maxn = 1e3;
struct Trie
{
    int cnt;
    Trie *nex[en];
} T[maxn];
int top;
inline int idx(char c)
{
    return c - 'a';
}
inline Trie * CreatTrie()
{
    Trie *p = &T[top ++];
    p -> cnt = 0;
    for(int i = 0; i< en; ++ i) p -> nex[i] = NULL;
    return p;
}
inline Trie * Init()
{
    top = 0;
    return CreatTrie();
}
void Insert(Trie *root, string s)
{
    Trie *p = root;
    for(int i = 0; i < s.size(); ++ i)
    {
        int temp = idx(s[i]);
        if(p -> nex[temp] == NULL) p -> nex[temp] = CreatTrie();
        p = p -> nex[temp];
        p -> cnt ++;
    }
    //p -> cnt --;
}
int Search(Trie *root, string s)
{
    Trie *p = root;
    for(int i = 0; i < s.size(); ++ i)
    {
        int temp = idx(s[i]);
        if(p -> nex[temp] == NULL) return 0;
        p = p -> nex[temp];
    }
    return p -> cnt;
}
int main()
{
    ios::sync_with_stdio(false);
    string s;
    Trie *root = Init();
    while(getline(cin, s, '\n') && s.size() != 0) Insert(root, s);
    while(cin >> s) cout << Search(root, s) << endl;
    return 0;
}

另外学习到了其他读回车结束的代码

while(1)
{
    gets(s);
    if(strcmp(s,"")==0)
        break;
}