题目解析：SQL代码补全｜ 豆包MarsCode AI 刷题

题目分析与解题思路

在开发SQL编辑器时，实现自动补全功能是非常常见的需求之一。其基本功能是：当用户输入字符片段时，程序能够根据已知的SQL关键字和数据库相关名称，自动提示所有以该字符片段开头的候选词，并按字典序输出。如果只有一个符合条件的词，则直接输出该词；如果没有符合条件的词，则返回“-1”。

问题的关键点：

1. 候选词的匹配：通过用户输入的前缀，查找所有以该前缀开头的候选词。
2. 去重：候选词中可能会有重复项，需进行去重处理。
3. 排序：输出的候选词需要按字典序排序。
4. 特殊情况处理：若没有匹配项或输入数据为空时，需要返回“-1”。

解题思路

1. 数据结构选择：为了实现自动补全功能，我们需要存储所有的SQL关键字和数据库字段。在Java中，Set（集合）是一个非常合适的选择。Set集合能自动去重，避免重复的候选词。此外，Set的查找效率为常数时间O(1)，能够有效提高匹配速度。

2. 匹配操作：对于每个输入的字符片段，我们需要遍历所有给定的SQL关键字和数据库名称，检查每个词是否以该片段为前缀。Java提供了String.startsWith()方法，这个方法非常适合做前缀匹配。通过该方法，我们可以快速地找到符合条件的候选词。

3. 去重与排序：由于候选词的集合是存储在Set中，因此不需要担心重复问题。当所有符合条件的候选词都被添加到Set中时，我们将其转换为List，并使用Collections.sort()对其进行字典序排序。

4. 返回结果：最后，我们需要根据候选词的数量来决定返回什么结果：

   • 如果候选词只有一个，直接返回该词；
   • 如果候选词有多个，将其用逗号连接并返回；
   • 如果没有任何候选词，返回"-1"表示没有匹配。

代码分析

1. solution函数：

   • 输入参数：num表示数据长度，data是包含所有SQL关键字和数据库字段的数组，input是用户输入的字符片段。

   • 主要操作：

     • 创建Set集合candidatesSet来存储候选词，利用Set去重功能，避免重复。
     • 遍历data数组，检查每个元素是否以input为前缀，符合条件的词添加到candidatesSet中。
     • 将Set集合转换为List并进行排序，确保候选词按字典序排列。
     • 如果候选词只有一个，返回该词；如果有多个，使用逗号连接返回；如果没有候选词，返回"-1"。