QUESTION
easy
题目描述
我们想要使用一棵四叉树来储存一个 N x N
的布尔值网络。网络中每一格的值只会是真或假。树的根结点代表整个网络。对于每个结点, 它将被分等成四个孩子结点直到这个区域内的值都是相同的.
每个结点还有另外两个布尔变量: isLeaf
和 val
。isLeaf
当这个节点是一个叶子结点时为真。val 变量储存叶子结点所代表的区域的值。
你的任务是使用一个四叉树表示给定的网络。下面的例子将有助于你理解这个问题:
给定下面这个 8 x 8
网络,我们将这样建立一个对应的四叉树:
对应的四叉树应该像下面这样,每个结点由一对 (isLeaf, val)
所代表.
对于非叶子结点,val
可以是任意的,所以使用 *
代替。
说明
- N 将小于
1000
且确保是2
的整次幂。 - 如果你想了解更多关于四叉树的知识,你可以参考这个 wiki 页面。
SOLUTION
这道题没什么好说的,用递归很简单,主要看耐心
方法一
我这里是保存了,左右上下坐标的范围
class Solution {
public:
Node* construct(vector<vector<int>>& grid) {
if(grid.size() == 0) return NULL;
return helper(grid, 0, grid.size()-1, 0, grid.size()-1);
}
Node* helper(vector<vector<int>>&grid, int up, int down, int l, int r){
if(l == r) return new Node((bool)grid[up][l], true, NULL, NULL, NULL, NULL);
Node *node = new Node(false, false, NULL, NULL, NULL, NULL);
int flag = grid[up][l];
for(int i = up; i <= down; i++){
for(int j = l; j <= r; j++){
if(grid[i][j] != flag){
node->topLeft = helper(grid, up, (up+down)/2, l, (l+r)/2);
node->topRight = helper(grid, up, (up+down)/2, (l+r)/2+1, r);
node->bottomLeft = helper(grid, (up+down)/2+1, down, l, (l+r)/2);
node->bottomRight = helper(grid, (up+down)/2+1, down, (l+r)/2+1, r);
return node;
}
}
}
node->val = flag;
node->isLeaf = true;
return node;
}
};
方法二
实际上 N x N
矩阵且 N 为 2
的次幂,可以更简单(转自博客)
class Solution {
public:
Node* construct(vector<vector<int>>& grid) {
return build(grid, 0, 0, grid.size());
}
Node* build(vector<vector<int>>& grid, int x, int y, int len) {
if (len <= 0) return NULL;
for (int i = x; i < x + len; ++i) {
for (int j = y; j < y + len; ++j) {
if (grid[i][j] != grid[x][y]) {
return new Node(true, false,
build(grid, x, y, len / 2),
build(grid, x, y + len / 2, len / 2),
build(grid, x + len/ 2, y, len / 2),
build(grid, x + len / 2, y + len / 2, len / 2));
}
}
}
return new Node(grid[x][y] == 1, true, NULL, NULL, NULL, NULL);
}
};