HDU 1075 What Are You Talking About // 字符串映射，Trie

题目描述

HDU 1075 What Are You Talking About

解题思路

题目大意：
给一些单词及其翻译，然后再输入一个句子，对其进行翻译，若句子中的某个单词不存在翻译，则原样输出，其中标点符号也是原样输出。

建Trie树跑一遍，然后需要注意的是:【句子中某个单词A 是以 字典中某个单词B的前缀出现，但是字典里却没有单词A】这种情况.比如说：
STRAT
earth fnnvk
END
START
i fiiwj fnnvk! dif fnn
END
其中fnn是单词中fnnvk的前缀，但是字典中却没有fnn这个单词！！

参考代码

//**********************************************
//  Author: @xmzyt1996
//  Date:   2015-10-15
//  Name:   HDU 1075.cpp
//**********************************************
#include <cstdio>
#include <cmath>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <string>
#include <bitset>
#include <vector>
#include <stack>
#include <queue>
#include <map>

using namespace std;

const int MAX_N = 3010;

struct Trie {
    Trie* next[26];
    int flag;
    char word[15];
}node[MAX_N*300]; 

int num;

void init() {
    num = 0;
    memset(node, 0, sizeof(node));
}

int getid(char c) { return c - 'a'; }

void insert(char* s, char* p) {
    Trie* head = &node[0];
    while (*s) {
        int id = getid(*s++);
        if ((head->next[id]) == NULL)
            head->next[id] = &node[++num];
        head = head->next[id];
    }
    head->flag = 1;
    strcpy(head->word, p);
}

void search(char* s) {
    Trie* head = &node[0];
    char* p = s;
    while (*s) {
        int id = getid(*s++);
        if ((head->next[id]) == NULL) { // 没找到就原样输出
            printf("%s", p);
            return ;
        }
        head = head->next[id];
    }
    printf("%s", (head->flag) ? head->word : p);
    // 一开始我写的是 printf("%s", head->word); 显然就是没考虑上面提到的情况
}

char s[MAX_N], s1[15], s2[15];

int main() {
    init();
    while(gets(s) && strcmp(s, "END")) {
        if (strcmp(s, "START") == 0) continue;
        sscanf(s, "%s %s", s1, s2);
        insert(s2, s1);
    }
    while(gets(s) && strcmp(s, "END")) {
        if (strcmp(s, "START") == 0) continue;
        int len = strlen(s), k = 0;
        s[len] = ' ', s[len+1] = 0;
        for (int i = 0; i < len; ++i) { // 将句子分解为一个个单词
            if(isalpha(s[i])) { 
                s1[k++] = s[i];
                if (!isalpha(s[i+1])) {
                    s1[k] = 0;
                    search(s1);
                    k = 0;
                }
            }else putchar(s[i]);
        }
        putchar('\n');
    }
    return 0;
}