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

Chương trình tìm số chuỗi con chỉ có 1s 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 có tất cả các ký tự là 1. Câu trả lời có thể rất lớn nên kết quả trả về mod 10 ^ 9 + 7.

Vì vậy, nếu đầu vào là s =​​"1011010", thì đầu ra sẽ là 5 vì 1. bốn lần "1" 2. một lần "11"

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

  • m:=10 ^ 9 + 7

  • kết quả:=0

  • div:=chia chuỗi nhị phân bằng cách chia nhỏ nó bằng '0'

  • đối với mỗi x trong div, thực hiện

    • nếu x trống, thì hãy chuyển sang lần lặp tiếp theo

    • result:=result + thương số của (kích thước x * (kích thước x +1)) / 2

  • kết quả trả về mod m

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

Ví dụ

def solve(s):
   m = 10**9+7
   result = 0
   for x in s.split('0'):
      if not x: continue
      result += (len(x)*(len(x)+1)) // 2
   return result % m
s = "1011010"
print(solve(s))

Đầu vào

"1011010"

Đầu ra

5