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

Chương trình đếm các chuỗi con có tất cả các số 1 trong chuỗi nhị phân bằng Python

Giả sử chúng ta có một chuỗi nhị phân s. Chúng ta phải tìm số chuỗi con chỉ chứa "1" s. Nếu câu trả lời quá lớn, hãy sửa đổi kết quả thành 10 ^ 9 + 7.

Vì vậy, nếu đầu vào là s =​​"100111", thì đầu ra sẽ là 7, vì các chuỗi con chỉ chứa "1" s là ["1", "1", "1", "1", "11" , "11" và "111"]

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

  • a:=0
  • số lượng:=0
  • đối với tôi trong phạm vi từ 0 đến kích thước là s-1, hãy thực hiện
    • nếu s [i] giống "0" thì
      • a:=0
    • nếu không,
      • a:=a + 1
      • count:=count + a
  • số lượng trả lại

Ví dụ

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

def solve(s):
   a = 0
   count = 0
   for i in range(len(s)):
      if s[i] == "0":
         a = 0
      else:
         a += 1
         count += a
   return count

s = "100111"
print(solve(s))

Đầu vào

"100111"

Đầu ra

7