一、现象
在刷力扣算法题 994. 腐烂的橘子 的时候,用python题解,发现多个for… for… if… 组合成一块,理解起来有点犯难,后面打印清楚了,鉴于此,便写篇博客记录一下
rotten = {(i, j) for i in range(row) for j in range(col) if grid[i][j] == 2} # rotten集合
fresh = {(i, j) for i in range(row) for j in range(col) if grid[i][j] == 1} # fresh集合
二、解决方案
打印出来看看结果
row = len(grid)
col = len(grid[0])
print('grid:', grid)
rotten = {(i, j) for i in range(row) for j in range(col) if grid[i][j] == 2} # rotten集合
print('rotten集合:', rotten)
fresh = {(i, j) for i in range(row) for j in range(col) if grid[i][j] == 1} # fresh集合
print('fresh集合:', fresh)
grid: [[2, 1, 1], [1, 1, 0], [0, 1, 1]]
rotten集合: {(0, 0)}
fresh集合: {(0, 1), (2, 1), (1, 1), (0, 2), (2, 2), (1, 0)}
操作如图所示:
这里解析:
rotten = {(i, j) for i in range(row) for j in range(col) if grid[i][j] == 2} # rotten集合
操作步骤如下:
1.for i in range(row): 循环row
2.for j in range(col): 循环col
3.if grid[i][j] == 2: 判断值为2输出
4.(i, j): 返回符合条件的结果
解决方案:带疑问,多print,勤动手,频思考