ขั้นตอนการเขียนโค้ด · การเข้ารหัสกริดทวิภาคด้วยควอดทรี

def encode_quadtree(grid):
    grid = tuple(tuple(row) for row in grid)
    n = len(grid)
    if not 1 <= n <= 16 or n & (n - 1) or any(len(row) != n for row in grid):
        raise ValueError("grid.shape")
    if any(type(v) is not int or v not in (0, 1) for row in grid for v in row):
        raise ValueError("cell.binary")
    def visit(x, y, size):
        value = grid[y][x]
        if all(grid[r][c] == value for r in range(y, y + size) for c in range(x, x + size)):
            return value
        half = size // 2
        return (visit(x, y, half), visit(x + half, y, half),
                visit(x, y + half, half), visit(x + half, y + half, half))
    return visit(0, 0, n)