【Leetcode】1180. Count Substrings with Only One Distinct Letter

記錄演算法發表於2021-01-04

題目地址:

https://leetcode.com/problems/count-substrings-with-only-one-distinct-letter/

給定一個字串 s s s,問其有多少個只含同一個字元的子串。

對於一個長 l l l的且只含一種字元的字串,其子串個數應該是 ∑ i = 1 l i = ( 1 + l ) l 2 \sum _{i=1}^{l}i=\frac{(1+l)l}{2} i=1li=2(1+l)l(這可以理解為列舉子串開頭字元然後累加)。所以只需要每次擷取出 s s s中只含同一個字元的子串然後累加即可。程式碼如下:

public class Solution {
    public int countLetters(String S) {
        int res = 0;
        
        for (int i = 0; i < S.length(); i++) {
            int j = i;
            while (j < S.length() && S.charAt(j) == S.charAt(i)) {
                j++;
            }
            res += (1 + j - i) * (j - i) / 2;
            
            i = j - 1;
        }
        
        return res;
    }
}

時間複雜度 O ( l s ) O(l_s) O(ls),空間 O ( 1 ) O(1) O(1)

相關文章