[4코3파] #310. Leetcode 1759. Count Number of Homogenous Substrings

[4코3파] 1명의 스위프트 개발자가 도망가고 4명의 코틀린 개발자, 3명의 파이썬 개발자의 코딩 테스트 스터디 : 4코3파

START :

[3코1파] 2023.01.04~ (310일차)
[4코1파] 2023.01.13~ (302일차)
[4코3파] 2023.10.01 ~ (40일차)

Today :

2023.11.09 [310일차]

Leetcode 1759. Count Number of Homogenous Substrings

[]

https://leetcode.com/problems/count-number-of-homogenous-substrings/description/

문제 설명

문자열에서 동일한 문자로 이루어진 부분 문자열의 개수를 세서 반환한다.

내 코드

 class Solution:
    def countHomogenous(self, s: str) -> int:
        cur = s[0]
        curCnt = 1
        ans = 0

        for i in range(1, len(s)):
            if s[i] == cur:
                curCnt +=1
            else:
                ans += ((1+curCnt)*curCnt)//2
                cur = s[i]
                curCnt = 1

        ans += ((1+curCnt) * curCnt )//2

        return ans % (10**9 + 7)
            

증빙

여담

데일리 문제... 웩