算法笔记：带障碍的铺砖问题

最新推荐文章于 2024-07-29 19:17:56 发布

FrenchOldDriver

最新推荐文章于 2024-07-29 19:17:56 发布

阅读量1k

点赞数

分类专栏：算法文章标签：算法 python 数据结构

本文链接：https://blog.csdn.net/OldDriver1995/article/details/106399653

版权

算法专栏收录该内容

8 篇文章

订阅专栏

问题描述：给定一个长方形地面，地面上有石头，现在要用长度为1x2的砖去铺满没有障碍的区域，砖可以横着或者竖着，但不能切割。最简单的例子，比如地面大小是2x2，没有障碍，那就用两块1x2的砖即可铺满。而假如是3x3的地面，正好中心点有一个障碍，剩下四个方向各放一块砖也能铺满。那么当地面很大的时候呢？

这道题一开始想了很久，一直想用带有回溯的方式去搜索所有可能性。但非常复杂，水平有限没写出来。后来想到了一个01染色的方法，用两种颜色去染空白区域，比如黑白，一黑一白即为一块砖，并且同样颜色的砖块不能相邻。也就是说白色四周（上下左右）只能是黑，反之亦然。

看下面几个例子，画图的时候黑色区域表示障碍，0和1分别是两种颜色。这个题肯定是不能根据空白区域的大小来判断的，下面两个图，空白区域都是6，但是右边这个你怎么都没办法用完整的砖块铺满。

在这里插入图片描述

当染色结束后，只需要统计01两种颜色的个数是否想等即可。

然而仅仅统计01个数是否相等就能判断了吗，答案肯定是不能的。如下图，当空白区域被障碍分割时，该方法会出现问题

在这里插入图片描述
但可以发现，不连通的也是可以判断的，只需要分开判断即可（如上图），虽然总数相等，但每个区域内部01是不等的。

这种分开判断的思想参考了《leetcode 200 岛屿数量》这道题的部分思想。即我的搜索算法执行次数等于分开的空白区域的个数。

并且这种朝四个方向染色的思想，跟岛屿数量那道题也挺像。

代码实现：

import numpy as np
dx=[1,0,0,-1]
dy=[0,1,-1,0]
# down, right,left, up
total1=0 
total2=0 
P=np.array(
  [[1,0,0,0,0,1,1,0],
   [1,1,1,1,1,1,0,0],
   [0,0,0,0,1,0,0,0],
   [1,0,0,1,1,0,1,0]] )*2

N = len(P)
M = len(P[0])
st = (P==2) #visited
st_sum = (P==2).sum()
def paint(i, j, k):
    global total1
    global total2
    global st_sum
    if i<0 or i>=N or j<0 or j>=M or P[i][j]==2 or st[i][j]==True: 
        return    
    P[i][j]=k
    st[i][j]=True
    st_sum+=1
    if P[i][j]==0:
        total1+=1
        paint(i+1, j, 1)
        paint(i-1, j, 1)
        paint(i, j+1, 1)
        paint(i, j-1, 1)
    else:
        total2+=1
        paint(i+1, j, 0)
        paint(i-1, j, 0)
        paint(i, j+1, 0)
        paint(i, j-1, 0)
res=True
for i in range(N):
    for j in range(M):
        if st[i][j]==False:
            paint(i, j, P[i][j])
            sub = (total1==total2)
            print("current space: ", sub)
            res = res and sub
            total1=0
            total2=0
            if st_sum==N*M:
                break        
print(P)
print("final result: ", res)