lqiu03

# Example test inside a notebook cell
n = 3
A = [
    [0, 1, 2],
    [3, 4, 5],
    [6, 7, 8]
]
B = [
    [1, 0, 0],
    [0, 1, 0],