[Swift]LeetCode1180. 统计只含单一字母的子串 ｜ Count Substrings with Only One Distinct Letter

★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★
➤微信公众号：山青咏芝（shanqingyongzhi）
➤博客园地址：山青咏芝（www.zengqiang.org）
➤GitHub地址：https://github.com/strengthen/LeetCode
➤原文地址：https://www.cnblogs.com/strengthen/p/11484244.html
➤如果链接不是山青咏芝的博客园地址，则可能是爬取作者的文章。
➤原文已修改更新！强烈建议点击原文地址阅读！支持作者！支持原创！
★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★

热烈欢迎，请直接点击！！！

进入博主App Store主页，下载使用各个作品！！！

注：博主将坚持每月上线一个新app！！！

Given a string S, return the number of substrings that have only one distinct letter.

 

Example 1:

Input: S = "aaaba"
Output: 8
Explanation: The substrings with one distinct letter are "aaa", "aa", "a", "b".
"aaa" occurs 1 time.
"aa" occurs 2 times.
"a" occurs 4 times.
"b" occurs 1 time.
So the answer is 1 + 2 + 4 + 1 = 8.

Example 2:

Input: S = "aaaaaaaaaa"
Output: 55

 

Constraints:

• 1 <= S.length <= 1000
• S[i] consists of only lowercase English letters.

---

 

给你一个字符串 S，返回只含 单一字母 的子串个数。

示例 1：

输入： "aaaba"
输出： 8
解释： 
只含单一字母的子串分别是 "aaa"， "aa"， "a"， "b"。
"aaa" 出现 1 次。
"aa" 出现 2 次。
"a" 出现 4 次。
"b" 出现 1 次。
所以答案是 1 + 2 + 4 + 1 = 8。

示例 2:

输入： "aaaaaaaaaa"
输出： 55

 

提示：

1. 1 <= S.length <= 1000
2. S[i] 仅由小写英文字母组成。

---

 

Runtime: 4 ms

Memory Usage: 21 MB

class Solution {
    func countLetters(_ S: String) -> Int {
        var arr:[Character] = Array(S)
        arr.append("#")
        var k:Int = 0
        var c:Character = "#"
        var ans:Int = 0
        for i in 0..<arr.count
        {
            if arr[i] == c
            {
                k += 1
            }
            else
            {
                if c != "#"
                {
                    ans += k * (k + 1) / 2
                }
                c = arr[i]
                k = 1
            }
        }
        return ans
    }
}