Python NZEC on KFORK but AC using C++

general
,
1

I was trying to solve the KFORK problem. I have a solution in C++ that I got AC with, but my identical Python3 solution gets an NZEC. I’m new to using Python on CodeChef, so maybe there is something basic I am missing. Can someone please take a look at the following code and see if something is wrong?


def is_X_Y_valid(board, X, Y):
    return (X >= 0 and X < len(board) and
            Y >= 0 and Y < len(board[X]))


T = int(input())

for _ in range(0, T):
    N, M = (int(i) for i in input().split())

    board = [[False] * N for x in range(N)]
    xAttacked = [False] * N
    yAttacked = [False] * N
    mainDiagAttacked = [False] * (2 * N - 1)
    offDiagAttacked = [False] * (2 * N - 1)

    for _ in range(0, M):
        X, Y = (int(i) for i in input().split())
        X -= 1
        Y -= 1
        board[X][Y] = True
        xAttacked[X] = True
        yAttacked[Y] = True
        mainDiagAttacked[N - 1 + X - Y] = True
        offDiagAttacked[X + Y] = True

    forkedKnights = 0
    for x in range(0, N):
        for y in range(0, N):
            if board[x][y]:
                continue

            if (xAttacked[x] or yAttacked[y] or
                    mainDiagAttacked[N - 1 + x - y] or
                    offDiagAttacked[x + y]):
                continue

            attacking = 0
            offsets = [[[-2, 2], [-1, 1]], [[-1, 1], [-2, 2]]]
            for a, b in offsets:
                for i in a:
                    for j in b:
                        if is_X_Y_valid(board, x + i, y + j):
                            if board[x + i][y + j]:
                                attacking += 1

            if attacking >= 2:
                forkedKnights += 1

    print(forkedKnights)