Program to find total number of strings, that contains one unique characters in Python

Suppose we have a string s of lowercase letters, we have to find the total number of substrings that contain one unique character.

So, if the input is like "xxyy", then the output will be 6 as substrings are [x, x, xx, y, y, yy]

To solve this, we will follow these steps −

• total := 0
• previous := blank string
• for each character c in s, do
  • if c is not same as previous, then
    • previous := c
    • temp := 1
  • otherwise,
    • temp := temp + 1
  • total := total + temp
• return total

Let us see the following implementation to get better understanding −

Example

 Live Demo

class Solution:    def solve(self, s):       total = 0       previous = ''       for c in s:          if c != previous:             previous = c             in_a_row = 1          else:             in_a_row += 1             total += in_a_row       return total ob = Solution() print(ob.solve("xxyy"))

Input

"xxyy"

Output

6