Hat’s Words (分成两个字符串考虑)

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

思路：

把字符串分成两个部分，看这两个部分是不是也在字典树上。如果在，就输出这个字符串。 

这题我只想说静态内存真是容易爆！

#include <stdio.h>
#include <string.h>
#include <string>
#include <iostream>
#include <stdlib.h>
#include <algorithm>

using namespace std;
const int MAX_NODE = 100000 + 10;
const int CHARSET = 26;

int trie[MAX_NODE][CHARSET]={0};
int color[MAX_NODE]={0};
int vis[MAX_NODE];
char s[50001][300];
int k = 1;

void insert(char *w)
{
    int len = strlen(w);
    int p = 0;
    for (int i=0;i<len;i++)
    {
        int c = w[i]-'a';
        if (!trie[p][c])
        {
            trie[p][c] = k;
            k++;
        }
        p = trie[p][c];
        vis[p]++;
    }
    color[p] = 1;
}


int search(char *s)
{
    int len = strlen(s);
    int p = 0;
    for (int i=0;i<len;i++)
    {
        int c = s[i]-'a';
        if (!trie[p][c])
            return 0;
        p = trie[p][c];
    }
    return color[p] == 1;
}


int main()
{
#ifndef ONLINE_JUDGE
    freopen("../in.txt","r",stdin);
#endif
    int t = 0;
    while (scanf("%s",s[t])!=EOF)
    {
        insert(s[t]);
        t++;
    }
    for (int i=0;i<t;i++)
    {
        for (int j=1;j<strlen(s[i]);j++)
        {
            char str1[1000]={''};
            char str2[1000]={''};
            strncpy(str1,s[i],j);
            strncpy(str2,s[i]+j,strlen(s[i])-j);
            if(search(str1) && search(str2))
            {
                cout << s[i] << endl;
                break;
            }
        }
    }
    return 0;
}