soj 3336 Diary(trie树)

@(K ACMer)

题意:
给你一个大字符串,是日记的编码.每个女生的名字对应一个长度为10的编码.求在日记中出现次数最多的编码所对应的名字

分析:
典型的trie树的应用,字符串计数.先把日记建立trie树,然后去查,每个字符串出现多少次即可.

#include <iostream>
#include <cstdio>
#include <cstring>
#include <set>
#include <map>
#include <stack>
#include <vector>
#include <string>
#include <queue>
#include <cstdlib>
#include <cmath>
#include <algorithm>
using namespace std;
const int mod = 835672545, INF = 0x3fffffff, maxn = 1e5 + 40;
char s[maxn * 10];
int n, nexts[12], slen;
struct node {
    char name[12], code[12];
    int num;
} v[maxn];

pair<int, int> ans;

#define MAX 26

typedef struct trienode
{
    int num;  // 该节点前缀 出现的次数
    struct trienode *next[MAX]; //该节点的后续节点
} trienode;

trienode memory[100000 * 5]; //先分配好内存。 malloc 较为费时
int mcnt = 0; // 内存计数器

//初始化一个节点。nCount计数为1， next都为null
trienode* create(void)
{
    trienode* tmp = &memory[mcnt++];  //节点指向一个已经分配好的内存节点之一.
    tmp -> num = 1;  //当前节点初始化个数为1
    for (int i = 0; i < MAX; i++)  //当前节点的所有儿子的指针初始化为null
        tmp->next[i] = NULL;
    return tmp;  //返回分配好的当前节点的指针
}

void insert(trienode* tmp, char* str)   //把字符串加入tire树中
{
    int i = 0, k;
    while (str[i]) {  //一个一个的插入字符
        k = str[i++] - 'a'; //当前字符 应该插入的位置
        if (tmp -> next[k])
            tmp -> next[k] -> num++;  //如果当前节点的该儿子已经存在,就把儿子的频率加1
        else
            tmp -> next[k] = create();  //如果该节点的该儿子不存在,就创建一个
        tmp = tmp -> next[k]; // 树往高深度进展
    }
}

int search(trienode* root, char * str)  //查找str为前缀的字符串,在树中的个数
{
    if (root == NULL)  //如果根节点为空,返回0
        return 0;
    trienode* tmp = root;  //当前节点
    int i = 0, k;
    while (str[i]) { //一直查找完整个字符串放可罢休
        k = str[i++] - 'a';  //当前节点对应的儿子的位置
        if (tmp -> next[k])  //如果儿子存在,就递归到儿子
            tmp = tmp->next[k];
        else
            return 0;  //前缀已经查完了,退出
    }
    return tmp -> num; //返回最后的那个字符,所在节点的num(也就是这个字符串在原串中的数量)
}

bool cmp(const node& a, const node& b) {
    return a.num > b.num;
}

int main(void) {
    int T;
    scanf("%d", &T);
    while (T--) {
        scanf("%s%d", s, &n);
        trienode* rt = create();
        ans.first = -1;
        ans.second = -1;
        slen = strlen(s);
        for (int i = 0; i <= slen - 10; i++) {
            char c = s[i + 10];
            s[i + 10] = '\0';
            insert(rt, s + i);
            s[i + 10] = c;
        }
        for (int i = 0; i < n; i++) {
            scanf("%s%s", v[i].name, v[i].code);
            v[i].num = search(rt, v[i].code);
            if (ans.first == -1 || ans.second == -1 || v[ans.second].num < v[i].num) {
                if (ans.first == -1) ans.first = i;
                else if (v[ans.first].num < v[i].num) ans.second = ans.first, ans.first = i;
                else ans.second = i;
            }
        }
        if (n >= 2 && v[ans.first].num == v[ans.second].num) puts("No-solution!");
        else printf("%s\n", v[ans.first].name);
    }
    return 0;
}