如何快速判断正对角线、反对角线、某一行、某一列元素是否相等

在二维数组或矩阵中快速判断正对角线、反对角线、特定行或列的元素是否相等，通常可以按照以下步骤操作：

1. **正对角线**（从左上到右下）：
- 对每个元素 `(i, j)`，检查它是否满足 `i == j` 的条件。
- 如果满足，比较该位置的值和其他对应位置的值是否相等。

2. **反对角线**（从左下到右上）：
- 对每个元素 `(i, j)`，检查它是否满足 `i + j == N - 1` 的条件，其中 `N` 是矩阵的大小。
- 同样比较该位置的值与其他对应位置的值是否相等。

3. **特定行**：
- 遍历指定行的所有元素，逐个与同一列的其他行元素比较。
- 使用循环和索引来获取特定行的元素并与列中对应的其他行元素做比较。

4. **特定列**：
- 类似于行的操作，遍历指定列的所有元素，逐个与所在行的其他列元素比较。

为了简化代码，你可以创建一个辅助函数，接受矩阵、行/列数以及目标元素作为参数，然后使用循环处理这些条件并返回结果。

def check_line(matrix, row_col_index, line_type):
    # matrix: 二维列表
    N = len(matrix)
    
    if line_type == "diagonal":
        for i in range(N):
            if (line_type == 'main') and (i != i):  # 主对角线检查 i == j
                continue
            elif (line_type == 'anti') and (i + i != N - 1):  # 反对角线检查 i + j == N - 1
                continue
            if matrix[i][i] != matrix[i][j]:  # 正反对角线元素比较
                return False
        return True
    elif line_type == "row" or "column":  # 行/列比较
        target_value = matrix[row_col_index][0]  # 获取指定位置的值
        for i in range(1, N):
            if matrix[row_col_index][i] != target_value:
                return False
        return True

# 使用示例
matrix = [[1, 2], [3, 4]]
row_col_index = 0  # 指定行/列索引
line_type = "row"  # 或者 "column"
result = check_line(matrix, row_col_index, line_type)