leetcode 算法题1180 (简单305) 统计只含单一字母的子串

leetcode 算法题1180 (简单305) 统计只含单一字母的子串

• 题目介绍

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

• 示例

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

输入： “aaaaaaaaaa”
输出： 55

• 提示

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

• 解法一

/**
 * @param {string} S
 * @return {number}
 */
var countLetters = function(S) {
    let i = count = 0, num = 1; nums = [];
    while(i < S.length - 1) {
      if(S[i++] === S[i]) {
        num++;
      } else {
        nums.push(num);
        num = 1;
      }
    }
    nums.push(num);
    i = 0;
    while(i < nums.length) {
      count += getNumber(nums[i++]);
    }
    return count;
};

const getNumber = n => {
  return n * (n + 1) / 2;
}

执行用时 : 68 ms, 在所有 JavaScript 提交中击败了100.00%的用户

内存消耗 : 34.8 MB, 在所有 JavaScript 提交中击败了100.00%的用户