题意:给你一个单词表 输出能由另外两个单词组成的单词
先判断一个单词的前缀是否为单词表中的单词,再判断后缀是否为单词表中的单词;
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<limits.h>
#include<ctype.h>
#include<math.h>
#include<string>
#include<iostream>
#include<algorithm>
using namespace std;
#include<queue>
#include<stack>
#include<vector>
#include<deque>
#include<set>
#include<map>
//N为最大点数
#define M 150
#define IN freopen ("in.txt" , "r" , stdin);
#define OUT freopen ("out.txt" , "w" , stdout);
//M为最大边数
typedef long long LL;
const int MAXN = 2999;//点数的最大值
const int MAXM = 2222;//边数的最大值
const int INF = 0x3f3f3f3f;
const int maxnode = 1000000;
const int sigma_size =30;
struct trie
{int ch[maxnode][sigma_size];int val[maxnode];int sz; // 结点总数trie(){sz=1; //初始化memset(ch[0],0,sizeof(ch[0]));}int idx(char c){return c-'a';}
//插入字符串s,附加信息为v。注意v必须!=0,因为0表示 本结点不是单词结点void insert(char *s){int u=0,n=strlen(s);for(int i=0; i<n; i++){int c=idx(s[i]);if(!ch[u][c])//结点不存在{memset(ch[sz],0,sizeof(ch[sz]));val[sz]=0; //附加信息为0ch[u][c]=sz++;//新建结点}u=ch[u][c];//往下走}val[u]++;//字符串的最后一个字符的附加信息为v}int find(char *s){int u=0,n=strlen(s);for(int i=0; i<n; i++){int c=idx(s[i]);if(!ch[u][c])//结点不存在{return 0;}u=ch[u][c];//往下走}return val[u];}
} tre;
int main()
{//IN;char s[30010][20];int i=0;while(scanf("%s",s[i++])!=EOF){tre.insert(s[i-1]);}for(int j=0; j<i; j++){int c=tre.idx(s[j][0]);int ac=tre.ch[0][c];for(int k=0; s[j][k]!='\0'; k++){if(tre.val[ac])//若前缀是{if(tre.find(&s[j][k+1]))//判断后缀{printf("%s\n",s[j]);break;}}c=tre.idx(s[j][k+1]);if(tre.ch[ac][c]==0)//若前缀不在树上则breakbreak;ac=tre.ch[ac][c];}}return 0;
}