八阵图代码,目测可行

directions = [(1, 1), (1, 0), (1, -1), (0, -1), (-1, -1), (-1, 0), (-1, 1), (0, 1)]

class Solution:
    def __init__(self, row, col):
        self.flag_arr = [[0 for i in range(col)] for j in range(row)]
        self.stack = []
        self.stack_num = 0
        self.max = 0
        self.min = 100000
        self.mid = 0
    def bianli(self, arr, rows, cols): #广度优先遍历
        for i in range(rows):
            for j in range(cols):
                if self.flag_arr[i][j] == 0 and arr[i][j] != 0:
                    self.stack.append((i, j))
                    print((i, j))
                    self.stack_num += 1
                    self.mid += arr[i][j]
                    self.flag_arr[i][j] = 1

                    while self.stack_num:
                        local_x, local_y = self.stack.pop()
                        self.stack_num -= 1
                        for dir_x, dir_y in directions:
                            if 0 <= local_x + dir_x < rows \
                                and 0 <= local_y + dir_y < cols \
                                and self.flag_arr[local_x + dir_x][local_y + dir_y] == 0 \
                                and arr[local_x + dir_x][local_y + dir_y] != 0:
                                self.stack.append((local_x + dir_x, local_y + dir_y))
                                print((local_x + dir_x, local_y + dir_y))
                                self.stack_num += 1
                                self.mid += arr[local_x + dir_x][local_y + dir_y]
                                self.flag_arr[local_x + dir_x][local_y + dir_y] = 1
                if self.mid > self.max:
                    self.max = self.mid
                if self.mid <= self.min and self.mid != 0:
                    self.min = self.mid
                self.mid = 0


array = [
    [1, 1, 1, 1, 0, 0, 1, 1],
    [1, 1, 1, 1, 0, 0, 1, 1],
    [0, 0, 0, 0, 0, 0, 1, 1],
    [0, 0, 0, 0, 0, 0, 1, 1],
    [1, 1, 0, 0, 0, 0, 1, 0],
    [1, 1, 0, 0, 0, 0, 0, 1],
]

rows = 6
cols = 8

so = Solution(rows, cols)
so.bianli(array, rows, cols)
print(so.max)
print(so.min)


'''
if __name__ == "__main__":
    row = int(input("row:"))
    col = int(input("col:"))

    arr = [[0 for i in range(col)] for j in range(row)]
    #arr = [[0]*col]*row
    for i in range(row):
        string = input().strip().split()
        for j in range(col):
            arr[i][j] = string[j]
    so.bianli(array, rows, cols)
    print(so.max)
    print(so.min)
'''



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值