题目
Output Format
Output GOOD SET if the set is valid.
Else, output BAD SET followed by the first string for which the condition fails.
Sample Input00
7
aab
defgab
abcde
aabcde
cedaaa
bbbbbbbbbb
jabjjjad
Sample Output00
BAD SET
aabcde
Sample Input01
4
aab
aac
aacghgh
aabghgh
Sample Output01
BAD SET
aacghgh
Explanation
aab is prefix of aabcde. So set is BAD SET and it fails at string aabcde.
解释
使用字典树,不用查询功能,因为本题只需找到第一个不符合的字符串,直接对插入功能进行改进。
设置一个全局ans数组存答案。字典树插入每个字符进入节点同时更新其flag,flag记录该节点被第i个字符串覆盖了,其flag值就是i。每个节点flag初始值均为-1,但其实具体被谁覆盖我们并不关心,只关心这个位置的节点是不是之前已经有字符串占据了,说明这条线路上有字符串存在。如果当前字符串到最后一位flag的值还不是-1,说明当前字符串比上面的某条字符串(第i条)短了,那么就是结果了。
if((node->exist != -1)||(*p == ‘\0’ && node->flag != -1)){
strcpy(ans, word);
return n;
}
这就是if右半边这样写的原因,例如abcabc,后遇到abc这种情况,这是容易落掉的情况。再讲一讲另外一种情况,现有条短的字符串,后来遇到长的字符串。例如abc,abcabc(例子不在多,在精巧)。
我们应该在第一个c的位置就意识到已经有结果了。在字典树中插入每个字符的最后一个字符是我们将其的exist的值更新为当前字符串的编号i,同样我们不关心它的值究竟为多少,总之这个节点要是exist的值不是初始值-1。说明这个节点已经是某条字符串(第i条)的结尾了,到了这无论当前字符串的字符是就此停止了还是还有更多的字符,都已经与上面的某条字符串重叠了。所以有了if左半边的写法。
找到结果之后记得把剩下的字符串读完,不插入字典树就好了,最后输出ans.
拓展
虽然这道题不关心是与第几条字符串重叠,但可以通过以下的代码确定是与第几条字符串重叠。
#include <cstdio>
#include <cstring>
#include <cstdlib>
typedef struct node{
int cont;
struct node *next[26];
int exist;
int flag;
}trieNode, *Trie;
char ans[100];
Trie buildtrieNode(){
Trie node = (Trie)malloc(sizeof(trieNode));
node->cont = 0;
node->exist = -1;
node->flag = -1;
memset(node->next, 0, sizeof(node->next));
return node;
}
int trieNode_insert(char *word, Trie root, int n){
Trie node = root;
int id;
char *p = word;
while(*p){
id = *p - 'a';
if(node->next[id] == NULL){
node->next[id] = buildtrieNode();
}
node = node->next[id];
p++;
node->cont++;
if((node->exist != -1)||(*p == '\0' && node->flag != -1)){
strcpy(ans, word);
return n;
}
node->flag = n;
}
node->exist = n;
return 0;
}
int main(){
int n, i;
char s[1000];
int cont = 0;
Trie root = buildtrieNode();
int flag = 0;
scanf("%d", &n);
for(i = 1; i <= n; i++){
scanf("%s", s);
if(!flag)
flag = trieNode_insert(s, root, i);
}
if(flag) printf("BAD SET\n%s", ans);
else printf("GOOD SET");
return 0;
}