[HDOJ1247]Hat’s Words

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1247

Hat’s Words

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 10982    Accepted Submission(s): 3937

Problem Description

A hat’s word is a word in the dictionary that is the concatenation of exactly two other words in the dictionary.
You are to find all the hat’s words in a dictionary.

 

Input

Standard input consists of a number of lowercase words, one per line, in alphabetical order. There will be no more than 50,000 words.
Only one case.

 

Output

Your output should contain all the hat’s words, one per line, in alphabetical order.

 

Sample Input

a ahat hat hatword hziee word

 

Sample Output

ahat hatword

 

 

在所有单词中找被其他两个单词表示的单词，全部丢到字典树里维护，暴力枚举每个单词，再将每个单词「芙兰达」判断是否在字典树中维护即可。注意如果找到符合情况的要及时跳出内层的枚举。

 

 

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <cmath>
#include <queue>
#include <map>
#include <stack>
#include <list>
#include <vector>

using namespace std;

char str[66666][81];

typedef struct Node {
    Node *next[26];
    int cnt;
    Node() {
        cnt = 0;
        for(int i = 0; i < 26; i++) {
            next[i] = NULL;
        }
    }
}Node;

void insert(Node *p, char *str) {
    for(int i = 0; str[i]; i++) {
        int t = str[i] - 'a';
        if(p->next[t] == NULL) {
            p->next[t] = new Node();
        }
        p = p->next[t];
    }
    p->cnt++;   //is a word
}

int find(Node *p, char *str) {
    for(int i = 0; str[i]; i++) {
        int t = str[i] - 'a';
        p = p->next[t];
        if(!p) {
            return 0;
        }
    }
    if(p->cnt >= 1) {
        return 1;
    }
    return 0;
    // return p->cnt;
}

int main() {
    int n = 0;
    Node *root = new Node();
    // freopen("in", "r", stdin);
    while(gets(str[n]) && strlen(str[n])) {
        insert(root, str[n]);
        n++;
    }
    char a[81];
    char b[81];
    for(int i = 0; i < n; i++) {
        int len = strlen(str[i]);
        for(int j = 1; j < len; j++) {
            memset(a, 0, sizeof(a));
            memset(b, 0, sizeof(b));
            strncpy(a, str[i], j);
            strncpy(b, str[i] + j, len - j);
            if(find(root, a) && find(root, b)) {
                printf("%s
", str[i]);
                break;
            }
        }
    }
    return 0;
}