Sunday, February 10, 2019

427. Construct Quad Tree

427Construct Quad Tree
We want to use quad trees to store an N x N boolean grid. Each cell in the grid can only be true or false. The root node represents the whole grid. For each node, it will be subdivided into four children nodes until the values in the region it represents are all the same.
Each node has another two boolean attributes : isLeaf and valisLeaf is true if and only if the node is a leaf node. The valattribute for a leaf node contains the value of the region it represents.
Your task is to use a quad tree to represent a given grid. The following example may help you understand the problem better:
Given the 8 x 8 grid below, we want to construct the corresponding quad tree:
It can be divided according to the definition above:

The corresponding quad tree should be as following, where each node is represented as a (isLeaf, val) pair.
For the non-leaf nodes, val can be arbitrary, so it is represented as *.
Note:
  1. N is less than 1000 and guaranteened to be a power of 2.
  2. If you want to know more about the quad tree, you can refer to its wiki.
------------------------
Solution#1, 很直接,扫到如果有不一样的,就重新递归。如果都一样,建一个node返回
复杂度应该是O(n^2), 类似1/2 + 1/4 + 1/8那种
/*
// Definition for a QuadTree node.
class Node {
    public boolean val;
    public boolean isLeaf;
    public Node topLeft;
    public Node topRight;
    public Node bottomLeft;
    public Node bottomRight;

    public Node() {}

    public Node(boolean _val,boolean _isLeaf,Node _topLeft,Node _topRight,Node _bottomLeft,Node _bottomRight) {
        val = _val;
        isLeaf = _isLeaf;
        topLeft = _topLeft;
        topRight = _topRight;
        bottomLeft = _bottomLeft;
        bottomRight = _bottomRight;
    }
};
*/
class Solution {
    public Node construct(int[][] grid) {
        return helper(grid,0,0,grid.length - 1,grid.length - 1);
    }
    
    private Node helper(int[][] grid, int r1, int c1, int r2, int c2) {
        if (r1 > r2 || c1 > c2)  return null;
        
        boolean isLeaf = true;
        int val = grid[r1][c1];
        for (int i = r1; i <= r2; i++) {
            for (int j = c1; j <= c2; j++) {
                if (val != grid[i][j]) {
                    isLeaf = false;
                    break;
                }
            }
        }
        
        if (isLeaf) return new Node(val == 1, true, null,null,null,null);
        
        int midR = (r1 + r2) / 2;
        int midC = (c1 + c2) / 2;        
        return new Node(false, false,
                       helper(grid,r1,c1,midR,midC),
                       helper(grid,r1,midC + 1,midR,c2),
                       helper(grid,midR + 1,c1,r2,midC),
                       helper(grid,midR + 1,midC + 1,r2,c2));
    }
}

另一个方法,先递归
https://leetcode.com/problems/construct-quad-tree/discuss/154565/Java-recursive-solution

No comments:

Post a Comment