Computer >> Máy Tính >  >> Lập trình >> Python

Chương trình đếm số lượng ma trận con vuông trong ma trận nhị phân đã cho bằng Python

Giả sử chúng ta có một ma trận nhị phân 2D. Chúng ta phải tìm tổng số ma trận con vuông có trong ma trận, trong đó tất cả các phần tử là 1.

Vì vậy, nếu đầu vào giống như

0 1 1
0 1 1

thì đầu ra sẽ là 5, vì có một (2 × 2) hình vuông và bốn (1 × 1) hình vuông

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -

  • nếu thảm trống, thì
    • trả về 0
  • c:=0
  • đối với tôi trong phạm vi 0 đến số hàng của thảm, hãy thực hiện
    • đối với j trong phạm vi 0 đến số cột của thảm, thực hiện
      • nếu mat [i, j] là 1, thì
        • nếu tôi là 0 hoặc j là 0, thì
          • c:=c + 1
        • nếu không,
          • temp =(tối thiểu là (mat [i-1, j-1], mat [i, j-1] và mat [i-1, j]) + mat [i, j]
          • c:=c + temp
          • mat [i, j]:=temp
  • trả lại c

Ví dụ

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn -

def solve(mat):
   if mat == []:
      return 0
   c = 0

   for i in range(len(mat)):
      for j in range(len(mat[0])):
         if mat[i][j] == 1:
            if i == 0 or j == 0:
               c += 1
            else:
               temp = (min(mat[i - 1][j - 1], mat[i][j - 1], mat[i - 1][j]) + mat[i][j])
               c += temp
               mat[i][j] = temp
   return c

matrix = [
   [0, 1, 1],
   [0, 1, 1]
]
print(solve(matrix))

Đầu vào

[[2, 6],[3, 4],[4, 7],[5, 5]]

Đầu ra

5