压缩字符串

[fedono]

compress('a') // 'a' compress('aa') // 'a2' compress('aaa') // 'a3' compress('aaab') // 'a3b' compress('aaabb') // 'a3b2' compress('aaabba') // 'a3b2a'

/**
 * @param {string} str
 * @return {string}
 */
function compress(str) {
  const res = [];
  let count = 0;

  for (let i = 0; i < str.length; i++) {
    count++;
    if (str[i] !== str[i+1]) {
      res.push(str[i]);
      if (count > 1) res.push(count); 
      count = 0;
    }
  }
  return res.join('');
}