hdu1247(Hat’s Words)

张志航

于 2024-08-15 02:21:44 发布

阅读量10

点赞数

文章标签：算法数据结构

我以为像a、aa这样的输入应该是没有输出的，结果还是要输出aa。

建树的时候就是常规建树，不过查找的时候要做一些变形：对于一个单词，从第一位检查有没有单词是它的前缀，如果有的话，再去检查它的后半部分是不是一个独立的单词，要满足这两次查找才能输出。

题意：给一些单词（以字典序输入），找出那些可以分成另外的两个单词的单词，以字典序输出；

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
using namespace std;
typedef struct Node
{
    int flag;
    struct Node *next[26];
}Node,*Tree;
char a[50001][30];
void Creat(Tree &T)
{
    T=(Node *)malloc(sizeof(Node));
    T->flag=0;
    for(int i=0;i<26;i++)
    {
        T->next[i]=NULL;
    }
}
void insert(Tree &T,char *s)
{
    Tree p=T;
    int t;
    int l=strlen(s);
    for(int i=0;i<l;i++)
    {
        t=s[i]-'a';
        if(p->next[t]==NULL)
        {
            Creat(p->next[t]);
        }
        p=p->next[t];
    }
    p->flag=1;
}
int search(Tree T,char *s)
{
    int t;
    Tree p=T;
    int l=strlen(s);
    for(int i=0;i<l;i++)
    {
        t=s[i]-'a';
        if(p->next[t]==NULL)
            return 0;
        p=p->next[t];
    }
    if(p->flag) return 1;
    else return 0;
}
void D(Tree p)
{
    for(int i=0;i<26;i++)
    {
        if(p->next[i]!=NULL)
            D(p->next[i]);
    }
    free(p);
}
int main()
{
     char s1[30];
     char s2[30];
     Tree T;
     int kk=0;
     Creat(T);
     while(gets(a[kk])&&strlen(a[kk]))
     {
         insert(T,a[kk]);
        kk++;
     }
     int i,j;
     for(i=0;i<kk;i++)
     {
         int l=strlen(a[i]);
         for(j=0;j<l;j++)
         {
             memset(s1,'\0',sizeof(s1));
             memset(s2,'\0',sizeof(s2));
             int kkk=0;
             int jjj=0;
             for(int kk=0;kk<=j;kk++)
                s1[kkk++]=a[i][kk];
              for(int jj=j+1;jj<l;jj++)
              {
                  s2[jjj++]=a[i][jj];
              }
              int qq=search(T,s1);
              int ww=search(T,s2);
              if(qq&&ww)
              {
                  printf("%s\n",a[i]);
                  break;
              }
              
         }
     }
     D(T);
    return 0;
}