棋子翻转

[编程题]棋子翻转

热度指数：3022 时间限制：C/C++ 1秒，其他语言2秒空间限制：C/C++ 256M，其他语言512M
算法知识视频讲解

在 4x4 的棋盘上摆满了黑白棋子，黑白两色棋子的位置和数目随机，其中0代表白色，1代表黑色；左上角坐标为 (1,1) ，右下角坐标为 (4,4) 。

现在依次有一些翻转操作，要对以给定翻转坐标(x,y)（也即第x行第y列）为中心的上下左右四个棋子的颜色进行翻转。

给定两个数组 A 和 f ，分别代表初始棋盘和哪些要进行翻转的位置(x,y) ，请返回经过所有翻转操作后的棋盘。

例如输入[[0,0,1,1],[1,0,1,0],[0,1,1,0],[0,0,1,0]],[[2,2],[3,3],[4,4]]时，初始键盘如下图所示：

对应的输出为[[0,1,1,1],[0,0,1,0],[0,1,1,0],[0,0,1,0]]，如下图所示：

示例1

输入

[[0,0,1,1],[1,0,1,0],[0,1,1,0],[0,0,1,0]],[[2,2],[3,3],[4,4]]

输出

[[0,1,1,1],[0,0,1,0],[0,1,1,0],[0,0,1,0]]

算法知识视频讲解

Python3

liangzhengjie

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param A int整型二维数组 
# @param f int整型二维数组 
# @return int整型二维数组
#
class Solution:
    def flipChess(self , A: List[List[int]], f: List[List[int]]) -> List[List[int]]:
        # write code here
        dp = [[0 for i in range(len(A[0]))] for j in range(len(A))]

        for arr in f:
            x = arr[0] - 1
            y = arr[1] - 1

            if x-1 >= 0:
                dp[x-1][y] = (dp[x-1][y]+1)%2

            if x+1 < len(A):
                dp[x+1][y] = (dp[x+1][y]+1)%2

            if y-1 >= 0:
                dp[x][y-1] = (dp[x][y-1]+1)%2

            if y+1 < len(A[0]):
                dp[x][y+1] = (dp[x][y+1]+1)%2

        for i in range(len(dp)):
            for j in range(len(dp[0])):
                if dp[i][j] == 1:
                    A[i][j] = 0 if A[i][j] == 1 else 1

        return A

发表于 2024-04-29 20:33:27 回复(0)

Python3

牛客343459580号

不确定题目是否有被重复翻转的块

# @param A int整型二维数组 
# @param f int整型二维数组 
# @return int整型二维数组
#
class Solution:
    def flipChess(self , A: List[List[int]], f: List[List[int]]) -> List[List[int]]:
        # write code here
        save = [[0]*len(A) for i in range(0,len(A))]
        for i in f:
            m,n = i[0],i[1]
            m -= 1
            n -= 1
            if(0<=m-1<len(A) and 0<=n<len(A)):
                save[m-1][n] += 1
            if(0<=m+1<len(A) and 0<=n<len(A)):
                save[m+1][n] += 1
            if(0<=m<len(A) and 0<=n-1<len(A)):
                save[m][n-1] += 1
            if(0<=m<len(A) and 0<=n+1<len(A)):
                save[m][n+1] += 1
        for i in range(0,len(A)):
            for j in range(0,len(A)):
                if(save[i][j]%2 == 1):
                    if(A[i][j] == 1):
                        A[i][j] = 0
                    else:
                        A[i][j] = 1
        return A

发表于 2022-02-18 17:09:25 回复(0)