LeetCode 427. 建立四叉树

427. 建立四叉树

题目来源：427. 建立四叉树

2022.04.29 每日一题

LeetCode 题解持续更新中GitHub仓库地址 CSDN博客地址

今天的题目一看，有点吓人，上来就给弄了个四叉树。但是实际仔细看一下题目和样例，就会发现，其实还好，就是 递归 的改编版。

我们需要做的就是判断一个范围之中的值是否都相等，如果不相等，就将当前范围划分为 四份，得到了新的范围，然后通过这四个范围，继续进行递归探寻。

class Solution {
public:
    Node *construct(vector<vector<int>> &grid) {
        return dfs(0, 0, grid.size(), grid.size(), grid);
    }

    // 创建递归函数
    Node *dfs(int r1, int c1, int r2, int c2, vector<vector<int>> &g) {
        // 创建变量 tag 判断 从 点(r1,c1) 到 点(r2,c2) 中是否所有的数字都是一样的
        bool tag = true;
        // 遍历 点(r1,c1) 到 点(r2,c2)这个范围之中的元素
        for (int i = r1; i < r2; i++) {
            for (int j = c1; j < c2; j++)
                // 如果发现了不相等
                if (g[r1][c1] != g[i][j]) {
                    // 将 tag 置反
                    tag = false;
                    // 并且退出遍历
                    break;
                }
            // 继续退出遍历
            if (!tag) break;
        }
        // 根据题意，如果范围内的数字均相同( tag 为 true)，val 的值与 范围内的值相同，这里需要注意一下， val 是布尔类型
        // 并且将 isLeaf 置为 True
        if (tag) return new Node(g[r1][c1] == 1, true);
        // 如果不满足要求，就将 val 随意置为一个值，将 isLeaf 置为 false，四个节点分别置为新的四叉树节点。
        return new Node(true, false, dfs(r1, c1, (r1 + r2) / 2, (c1 + c2) / 2, g), dfs(r1, (c1 + c2) / 2, (r1 + r2) / 2, c2, g), dfs((r1 + r2) / 2, c1, r2, (c1 + c2) / 2, g), dfs((r1 + r2) / 2, (c1 + c2) / 2, r2, c2, g));
    }
};

public class Solution {
    public Node construct(int[][] grid) {
        return dfs(0, 0, grid.length, grid.length, grid);
    }

    // 创建递归函数
    Node dfs(int r1, int c1, int r2, int c2, int[][] g) {
        // 创建变量 tag 判断 从 点(r1,c1) 到 点(r2,c2) 中是否所有的数字都是一样的
        boolean tag = true;
        // 遍历 点(r1,c1) 到 点(r2,c2)这个范围之中的元素
        for (int i = r1; i < r2; i++) {
            for (int j = c1; j < c2; j++)
                // 如果发现了不相等
                if (g[r1][c1] != g[i][j]) {
                    // 将 tag 置反
                    tag = false;
                    // 并且退出遍历
                    break;
                }
            // 继续退出遍历
            if (!tag) break;
        }
        // 根据题意，如果范围内的数字均相同( tag 为 true)，val 的值与 范围内的值相同，这里需要注意一下， val 是布尔类型
        // 并且将 isLeaf 置为 True
        if (tag) return new Node(g[r1][c1] == 1, true);
        // 如果不满足要求，就将 val 随意置为一个值，将 isLeaf 置为 false，四个节点分别置为新的四叉树节点。
        return new Node(true, false, dfs(r1, c1, (r1 + r2) / 2, (c1 + c2) / 2, g), dfs(r1, (c1 + c2) / 2, (r1 + r2) / 2, c2, g), dfs((r1 + r2) / 2, c1, r2, (c1 + c2) / 2, g), dfs((r1 + r2) / 2, (c1 + c2) / 2, r2, c2, g));
    }
}