PTA R7-13 小明找前缀100000(假)

参考于书鸢1236的博客

题目背景
小明最近上课天天睡觉，于是啥都不会。
一天，老师终于点兵点将点到他回答问题，你能帮他渡过难关吗？
现在老师给了小明 $n$ 个由 0、1 构成的字符串，然后有 $m$ 次询问，
每次询问给出一个由 0、1 构成的字符串，
小明需要在最开始的 $n$ 个字符串中找出和当前字符串匹配的前缀最长的串。
如果有多个前缀最长的串，请找出字典序最小的那个。
如果小明找不出来，小明就无了，你能帮助小明吗？
(解释：字符串匹配的前缀最长的串：具有相同前缀且前缀中相同字符最多的串；字典序最小：按照字符 a~z 的顺序排，若前 $n$ 位(字符串长度较短的位数)字符都相同则按字符串长度从小到大排)
输入格式:
输入第一行是一个整数 $t$ ，表示有 $t$ 组测试实例。
对于每组输入：
输入第一行包含一个整数 $n$ ，代表接下来有 $n$ 行。
接下来 $n$ 行，每行1个字符串，字符串只由0、1构成。
接下来这一行有一个整数 $m$ ，表示询问次数。
接下来 $m$ 行，每行1个字符串，字符串只由0、1构成。
对于每一次询问，你需要求出匹配的前缀最长的字符串，如果有多个答案，请输出字典序最小的那个。
输出格式:
对于每组输入：
输出共 $m$ 行，每行一个字符串对应询问的答案。
输入样例:

2
5
0100
0101
011
10110
10111
1
10011011
4
01
1001
1011100
00110
2
100011
01100

输出样例:

10110
1001
01

题目大意:
在给出的 $n$ 个字符串中找出与询问的字符串前缀相同最长的字符串，若有多个答案输出字典序最小的那个。
解题思路:
利用题目只有 $01$ 两个字符，我们可以利用二叉树来进行存储(0左1右 或者 1左0右)。特别的是需要记录给出字符串的长度，换句话说就是要标记每个字符串的结束结点，因为最后我们要字典序小的那个字符串。存好了就要开始找前缀匹配的字符串了，按照需要寻找字符串中字符0和1顺序往下寻找，规则是：

• 按照顺序往下找，若还能往下匹配则继续往下找。
• 若字符串还未遍历完，但已经找不到不匹配了而且到达的结点不是某个字符串的结束结点，则将继续遍历找到一个字典序小的字符串。(尽量往 $0$ 子树走，走不了再往 $1$ 走，在这个过程中碰到某个字符串的结束标记就停止了)
• 若字符串遍历完了，但到达的结点不是某个字符串的结束结点，那么也要像上一条规则一样找出字典序小的那个字符串。

#include <bits/stdc++.h>
using namespace std;

typedef struct TNode* Tree;
struct TNode
{
    char Data;
    Tree L, R;
    bool Is;
};

void createTree(Tree& T, const string s, int idx)
{
    if(!T && idx<s.size())
    {
        T = new TNode;
        T->Data = s[idx++];
        T->L = T->R = nullptr;
        T->Is = false;
    }
    if(idx == s.size()) //标记该结点为字符串结束结点
    {
        T->Is = true;
        return;
    }
    // (T->L != nullptr) 表示结点不为空时为 1，即下标需要后移，判断下一位
    // 否则当前下标需要创建结点进行存储不进行后移
    if(s[idx] == '1') createTree(T->L, s, idx + (T->L != nullptr));
    else createTree(T->R, s, idx + (T->R != nullptr));
}

void getAns(Tree T, string& ans, const string s, int idx)
{
    if(T->Data != '-') ans += T->Data; //0或1才进行记录
    //若字符串遍历完(或者到某个字符就无法匹配)，且当前结点为某个字符串结束结点就退出
    if(idx >= s.size() && T->Is) return;
    else if(idx < s.size()) //进行字符匹配
    {
        if(s[idx] == '1' && T->L)
            getAns(T->L, ans, s, idx+1);
        else if(s[idx] == '0' && T->R)
            getAns(T->R, ans, s, idx+1);
        else if(T->Is) return; //往下不匹配后，若是结束结点就退出
        else idx = s.size()+1; //标记为不匹配
    }
    //不是结束结点，就开始尽量往 0 子树匹配，没 0 就往 1 匹配，在过程碰到结束结点就退出
    if(idx >= s.size() && !T->Is)
    {
        if(T->R) getAns(T->R, ans, s, idx);
        else if(T->L) getAns(T->L, ans, s, idx);
    }
}

int main()
{
    int t; cin>>t;
    while(t--)
    {
        Tree T = new TNode;
        T->Data = '-';
        T->L = T->R = nullptr;
        T->Is = false;
        int n; cin>>n;
        while(n--)
        {
            string s; cin>>s;
            //由字符串创建一颗树
            createTree(T, s, 0);
        }
        int m; cin>>m;
        while(m--)
        {
            string s, ans=""; cin>>s;
            getAns(T, ans, s, 0);
            cout<<ans<<'\n';
        }
    }
    return 0;
}