关注
消消乐问题: 用一个感染函数,一个标志矩阵(初始值为1),把与初始点相同且连成一片的位置,对应在标志矩阵的位置改为0,然后与初始矩阵进行对应位置相乘的操作,则初始矩阵被消除的位置变为0,其他位置数字保持不变,然后执行下沉操作,对下沉完成的矩阵继续执行开始的消除操作 开始写的版本是贪心的,就是每次消除最大数量的点,但后来感觉不合适,不是最优解。后来改成每次从上到下,从左到右顺序‘感染’,只要凑够三个就删除 从上到下,从左到右顺序消除: import numpy as np
class returndata():#每个点给上一个点返回的信息(与本点相邻且相同的点有多少个?)
subnum = 0
def __init__(self,subnum):
self.subnum = subnum
#感染函数(给定flaglist矩阵,把能连城一片的点的位置,对应在flaglist中表为0)
def infect(i, j, number, checkerboard, flagList):
if i<0 or i >= len(checkerboard) or j<0 or j>= len(checkerboard[0]) or checkerboard[i][j] != number or \
flagList[i][j]==0 or number == 0:
return returndata(0)
flagList[i][j] = 0
sumsubnum = returndata(1)
#从上下左右四个方向继续“感染”
a = infect(i + 1, j, checkerboard[i][j], checkerboard, flagList)
b = infect(i - 1, j, number, checkerboard, flagList)
c = infect(i, j + 1, checkerboard[i][j], checkerboard, flagList)
d = infect(i, j - 1, number, checkerboard, flagList)
sumsubnum.subnum += a.subnum+b.subnum+c.subnum+d.subnum
return sumsubnum#返回值为 从此点出发,最多能删除的点的个数
#下沉函数,把不是0的数都下沉下去
def down(matrix):
for i in range(len(matrix)-1,0,-1):
for j in range(len(matrix[0])):
if matrix[i][j]==0:
for h in range(i-1,-1,-1):
if matrix[h][j]!=0:
matrix[i][j] = matrix[h][j]
matrix[h][j] = 0
return matrix
checkerboard = np.array([[3, 1, 2, 1, 1, ], \
[1, 1, 1, 1, 3], \
[1, 1, 1, 1, 1], \
[1, 1, 1, 1, 1], \
[3, 1, 2, 2, 2]])
row = checkerboard.shape[0] # 获取行数
col = checkerboard.shape[1] # 获取列数
flagList = [[1] * col for i in range(row)] # 创建标志矩阵,初始值为1,能删除的位置变为0
flagList = np.array(flagList)
#print(checkerboard)
for i in range(row):
for j in range(col):
flagList = [[1] * col for k in range(row)]#消除矩阵
flagList = np.array(flagList)
result = infect(i, j, checkerboard[i][j], checkerboard, flagList)
if result.subnum>=3:#如果相同点连通个数大于等于3,则进行消除操作
checkerboard = checkerboard*flagList
checkerboard = down(checkerboard)
print(checkerboard)
count = 0#最后剩下的不能删除的点的个数
for lst in checkerboard:
for x in lst:
if x!=0:
count+=1
print(count) 贪心版本: import numpy as np
class returndata():#返回的信息(与本点相邻的点有多少个?)
subnum = 0
def __init__(self,subnum):
self.subnum = subnum
def infect(i, j, number, checkerboard, flagList):#感染函数
if i<0 or i >= len(checkerboard) or j<0 or j>= len(checkerboard[0]) or checkerboard[i][j] != number or \
flagList[i][j]==0 or number == 0:
return returndata(0)
flagList[i][j] = 0
sumsubnum = returndata(1)
a = infect(i + 1, j, checkerboard[i][j], checkerboard, flagList)
b = infect(i - 1, j, number, checkerboard, flagList)
c = infect(i, j + 1, checkerboard[i][j], checkerboard, flagList)
d = infect(i, j - 1, number, checkerboard, flagList)
sumsubnum.subnum += a.subnum+b.subnum+c.subnum+d.subnum
return sumsubnum
def down(matrix):#把不是0的都沉下去
for i in range(len(matrix)-1,0,-1):
for j in range(len(matrix[0])):
if matrix[i][j]==0:
for h in range(i-1,-1,-1):
if matrix[h][j]!=0:
matrix[i][j] = matrix[h][j]
matrix[h][j] = 0
return matrix
checkerboard = np.array([[3, 1, 2, 1, 1, ], \
[1, 1, 1, 1, 3], \
[1, 1, 1, 1, 1], \
[1, 1, 1, 1, 1], \
[3, 1, 2, 2, 2]])
row = checkerboard.shape[0] # 获取行数
col = checkerboard.shape[1] # 获取列数
flagList = [[1] * col for i in range(row)]
flagList = np.array(flagList)
print(checkerboard)
dic = {}#key:可删除点的个数 value:对应个数的标志矩阵(0 1矩阵)
while True:
dic.clear()
for i in range(row):
for j in range(col):
flagList = [[1] * col for k in range(row)]#消除矩阵
flagList = np.array(flagList)
result = infect(i, j, checkerboard[i][j], checkerboard, flagList)
if result.subnum>=3:
dic[result.subnum] = flagList
if dic=={}:#结束条件(不能再消除的时候结束)
break
#if max(dic.keys())<3:
# break
maxdeletematrix = dic.get(max(dic.keys()))
checkerboard = checkerboard*maxdeletematrix
checkerboard = down(checkerboard)
print(checkerboard)
count = 0#最后剩下的不能删除的点的个数
for lst in checkerboard:
for x in lst:
if x!=0:
count+=1
print(count)
查看原帖
点赞 3
相关推荐
2024-12-26 20:46
复旦大学 C++ 点赞 评论 收藏
分享
![](https://static.nowcoder.com/fe/file/oss/1715049343797JOCFB.png)
点赞 评论 收藏
分享
牛客热帖
更多
- 1... 985计算机老学长掏心窝子:当年我踩过的坑,希望你们能绕开1.2W
- 2... 厌倦互联网了,开始逃离ing6669
- 3... 想要在大厂生存必须要学会提效3972
- 4... 【已挂】影石Insta360|嵌入式软件|日常实习一面1490
- 5... 杭州实在智能二面906
- 6... 我又又发现了一个使用满血版DeepSeek不卡顿的方法!(附春招企业汇总)824
- 7... CVTE - 嵌入式软件开发工程师(Linux方向)- 面经(12)788
- 8... 我是大一新生,本科就想找工作。我到底要不要打ACM?一文带你理清到底要不要打725
- 9... 腾讯hr-后台方向有过一面的吗718
- 10... 25届offer选择692
正在热议
更多
# 小厂实习有必要去吗 #
29950次浏览 203人参与
# 我的工作日记 #
51662次浏览 731人参与
# 国企和大厂硬件兄弟怎么选? #
112386次浏览 1635人参与
# 文科生还参加今年的春招吗 #
2451次浏览 23人参与
# 影石Insta360求职进展汇总 #
106657次浏览 952人参与
# 选择和努力,哪个更重要? #
38900次浏览 421人参与
# 中核求职进展汇总 #
12886次浏览 123人参与
# 如果再来一次,你还会学硬件吗 #
101783次浏览 1224人参与
# 如果公司降薪,你会跳槽吗? #
43464次浏览 335人参与
# 一人推荐一个值得去的通信/硬件公司 #
160624次浏览 1732人参与
# 机械人选offer,最看重什么? #
68308次浏览 431人参与
# 长光卫星求职进展汇总 #
27246次浏览 180人参与
# 秋招前后对offer的期望对比 #
223925次浏览 1675人参与
# 秋招盘点:机械人值得去的企业 #
60808次浏览 630人参与
# 大疆今年的机械笔试难吗? #
34971次浏览 406人参与
# 大疆的机械笔试比去年难吗 #
63779次浏览 575人参与
# 24届市场营销薪资爆料 #
9252次浏览 62人参与
# 你的秋招简历被谁挂了? #
215214次浏览 2393人参与
# 当下环境,你会继续卷互联网,还是看其他行业机会 #
67907次浏览 489人参与
# 我的国央企投递进展 #
35540次浏览 235人参与