在本地 idea 中自动生成牛客刷题列表

前言

前几天使用牛客 “题库” 刷题时，尽管牛客本身的代码提示足够智能，但发现还是更喜欢本地 idea 进行编译调试。
尤其是带有 链表 ListNode 或者二叉树 TreeNode 的题目。本地开发调试会更加方便。
同时，将所有刷题记录都保存在一个工程中，也更方便后续的复习查看。
但是，如果在本地一个个新建题目类和写测试类，又显得特别的累赘繁琐。

有没有办法 自动生成 这部分内容呢？

思路分析

以剑指offer 这个专题为例。https://www.nowcoder.com/exam/oj/ta?tpId=13
通过 F12 查看可以发现，题目列表的 JSON 都保存在如下请求中：
https://www.nowcoder.com/api/questiontraining/coding/getTopicQuestion?pageSize=100&title=&topicId=265&page=1&_=166****285152

在请求结果 JSON 中，可以看到，题目难度保存在 difficultyArray 中，而题目列表保存在 questions 中。
从而，我们可以分析该 JSON，来获取生成相应的 Java 类。
为了后续生成 测试类方便，我们可以同时生成对应的题目 Markdown 文档，最终从 Markdown 中，生成对应的测试类。

生成 Java 类

查看其中一个题目的 JSON。如下所示：

      {
        "difficulty": 2,
        "topicUrl": "/ta/coding-interviews-all",
        "tpId": 265,
        "questionId": 1375279,
        "questionTitle": "数组中重复的数字",
        "acceptRate": 54.5677,
        "questionNo": "JZ3",
        "isNew": false,
        "questionUUid": "6fe361ede7e54db1b84adc81d09d8524",
        "tqId": 39207,
        "tags": [
          {
            "name": "数组",
            "id": 578
          }
        ]
      }

我们可以在 questionTitle 中得到题目名称，从 questionUUid 得到题目的 UUID，进而得到题目的 URL。
如 "6fe361ede7e54db1b84adc81d09d8524" 这个 UUID，可以链接到对应的题目：https://www.nowcoder.com/practice/6fe361ede7e54db1b84adc81d09d8524

生成类名

可以看到，questionTitle 是中文，我们如果想作为 Java 类名的话，需要转换为英文。
查询可以发现开源工具 https://github.com/houbb/pinyin 可以将中文转换为对应的拼音，从而我们可以转换得到类名。
编写对应的转换方法，如下所示：

    public static String toPinyin(String info) {
        if (StringUtils.isBlank(info)) {
            return null;
        }
        if (info.contains("（")) {
            info = info.split("（")[0];
        }
        if (info.contains("(")) {
            info = info.split("\\(")[0];
        }
        String pinyin = PinyinHelper.toPinyin(info, PinyinStyleEnum.NORMAL);
        List<String> list = new LinkedList<>();
        for (String t : pinyin.split(" ")) {
            list.add(StringUtils.capitalize(t));
        }
        return String.join("", list).replaceAll("[^a-zA-Z0-9]", "");
    }

可以发现，questionTitle 的中文中，存在部分括号、和标点，这部分我们进行过滤去除。
同时，为了检索方便，我们将 questionNo 也放到类名中。
为了排序方便，JZ3 我们也转换为 JZ03，加0补齐。
最终生成的类名如：JZ03_ShuZuZhongChongFuDeShuZi

获取题目列表

通过 JSON 遍历，得到所有题目的 QuestionInfo 对象 。

    public static List<QuestionInfo> getQuestionInfoList(String html) {
        JSONObject contentJson = JSONObject.parseObject(html);
        JSONObject dataJson = contentJson.getJSONObject("data");
        JSONArray questionsArray = dataJson.getJSONArray("questions");
        JSONArray difficultyArray = dataJson.getJSONArray("difficultyArray");
        Map<Integer, String> difficultyMap = new HashMap<>();
        for (int i = 0; i < difficultyArray.size(); i++) {
            JSONObject jsonObject = difficultyArray.getJSONObject(i);
            difficultyMap.put(jsonObject.getInteger("id"), jsonObject.getString("title"));
        }
        List<QuestionInfo> questionInfos = JSONObject.parseArray(questionsArray.toString(), QuestionInfo.class);
        for (QuestionInfo info : questionInfos) {
            info.setDifficultyInfo(difficultyMap.getOrDefault(info.getDifficulty(), String.valueOf(info.getDifficulty())));
        }
        return questionInfos;
    }

    public static List<QuestionInfo> getQuestionInfoList() {
        String content = LoadTestCaseData.loadInfo(CODE_PATH);
        List<QuestionInfo> questionInfos = getQuestionInfoList(content);
        return questionInfos;
    }

在上述方法中，我们将题目的“难度”也获取了出来。

生成Java 类

对每个题目，生成Java类：

    public static void createJavaName(QuestionInfo questionInfo, String type) {
        try {
            File file = new File(getJavaPath(questionInfo));
            if (!file.getParentFile().exists()) {
                file.getParentFile().mkdirs();
            }
            if (!file.exists()) {
                file.createNewFile();
            } else {
                System.out.println(questionInfo.getJavaName() + " is exists.");
                return;
            }
            FileOutputStream out = new FileOutputStream(file, true);
            StringBuffer sb = new StringBuffer();
            sb.append("package com.jueee.nowcoder." + type + ";\n\n\n");
            sb.append(getClassAnnotation(questionInfo));
            sb.append("public class " + questionInfo.getJavaName() + " {\n\n\n\n}");
            out.write(sb.toString().getBytes("utf-8"));
            out.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
        return;
    }

其中，getClassAnnotation 方法表示注释，由于测试类中也有使用，我们将其独立出来。

    public static String getClassAnnotation(QuestionInfo questionInfo) {
        StringBuffer buffer = new StringBuffer();
        buffer.append("/**\n");
        buffer.append(" * " + questionInfo.getQuestionNo() + "：" + questionInfo.getQuestionTitle() + "\n");
        buffer.append(" * 链接：https://www.nowcoder.com/practice/" + questionInfo.getQuestionUUid() + "\n");
        buffer.append(" * 难度：" + questionInfo.getDifficultyInfo() + "\n");
        buffer.append(" * 描述：" + questionInfo.getMarkDownName() + ".md\n");
        buffer.append(" * " + questionInfo.getTagInfo() + "\n");
        buffer.append(" */\n");
        return buffer.toString();
    }

生成Markdown 文件

同理，我们生成MarkDown的文档。

    public static void createMarkDown(QuestionInfo questionInfo) {
        try {
            File file = new File(getMarkPath(questionInfo));
            if (!file.getParentFile().exists()) {
                file.getParentFile().mkdirs();
            }
            if (!file.exists()) {
                file.createNewFile();
            } else {
                System.out.println(questionInfo.getMarkDownName() + " is exists.");
                return;
            }
            FileOutputStream out = new FileOutputStream(file, true);
            StringBuffer buffer = new StringBuffer();
            buffer.append("## " + questionInfo.getQuestionTitle() + "  \r");
            buffer.append("链接：[" + questionInfo.getQuestionNo() + "](" + questionInfo.getTitleUrl() + ") \r");
            buffer.append("难度：**" + questionInfo.getDifficultyInfo() + "**  \r");
            buffer.append("实现：" + questionInfo.getJavaName() + ".java  \r");
            buffer.append("" + questionInfo.getTagInfo() + "  \r");
            buffer.append("  \r\r");
            out.write(buffer.toString().getBytes("utf-8"));
            out.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

最终效果

类文件和MarkDown文件：

对于每个Java类，内容如下所示：

对于每个MarkDown文件，内容如下所示：

关于测试类和测试方法的生成，会比较复制，我们下次再聊。

如果有其他问题，可以在本贴下面留言交流。