leetcode 1180. Count Substrings with Only One Distinct Letter

Seawolf

1. Given a string S, return the number of substrings that have only one distinct letter.
   
2. 
   
3. 
   
4. 
   
5. Example 1:
   
6. 
   
7. Input: S = "aaaba"
   
8. Output: 8
   
9. Explanation: The substrings with one distinct letter are "aaa", "aa", "a", "b".
   
10. "aaa" occurs 1 time.
    
11. "aa" occurs 2 times.
    
12. "a" occurs 4 times.
    
13. "b" occurs 1 time.
    
14. So the answer is 1 + 2 + 4 + 1 = 8.
    
15. Example 2:
    
16. 
    
17. Input: S = "aaaaaaaaaa"
    
18. Output: 55
    
19. 
    
20. 
    
21. Constraints:
    
22. 
    
23. 1 <= S.length <= 1000
    
24. S[i] consists of only lowercase English letters.

复制代码

1. class Solution {
   
2.     public int countLetters(String S) {
   
3.         if (S.length() == 0) return 0;
   
4.         if (S.length() == 1) return 1;
   
5.         int sum = 0;
   
6.         String remain = "";
   
7.         Set <String> set = new HashSet<>();
   
8.         String init = S.substring(0,1);
   
9.         set.add(init);
   
10.         for(int i = 0; i< S.length(); i++){
    
11.             
    
12.             if(S.substring(i,i+1).equals(init)) {
    
13.                
    
14.                 remain = remain + S.substring(i,i+1);
    
15.                 set.add(remain + S.substring(i,i+1));
    
16.             }
    
17.             else {
    
18.                 remain = "";
    
19.                 init = S.substring(i,i+1);
    
20.                
    
21.                 set.add(init);
    
22.             }
    
23.         }
    
24.         Iterator<String> iterator = set.iterator();
    
25.         while(iterator.hasNext()) {
    
26.             sum = sum + compare(S,iterator.next());
    
27.         }
    
28.         
    
29.         return sum;
    
30.     }
    
31.    
    
32.     public int compare(String S, String s){
    
33.         int sum = 0;
    
34.         for(int i = 0; i<= S.length() - s.length(); i++){
    
35.             if(S.substring(i,i+s.length()).equals(s)) sum ++;
    
36.             
    
37.         }
    
38.         return sum;
    
39.     }
    
40. }

复制代码