排序

第 347 题

前 K 个高频元素open in new window

var topKFrequent = function (nums, k) {
  // 使用map统计各个元素出现的次数
  let map = new Map()
  for (let i = 0; i < nums.length; i++) {
    if (map.has(nums[i])) {
      map.set(nums[i], map.get(nums[i]) + 1)
    } else {
      map.set(nums[i], 1)
    }
  }
  // 按照出现次数将元素放在数组对应的位置
  // 例如出现3次的就放在数组的第3个位置
  // 因为可能有相同次数的，所以每个位置都用数组保存
  let arr = []
  map.forEach((value, key) => {
    if (arr[value]) {
      arr[value].push(key)
    } else {
      arr[value] = [key]
    }
  })
  // 定义返回结果
  let result = []
  let p = k
  // 倒叙遍历数组
  for (let j = arr.length; j >= 0 && p > 0; j--) {
    // 数组某个位置存在值时，表示有出现j次的元素，存入结果中
    if (arr[j] !== undefined) {
      result = result.concat(arr[j])
      p--
      // 达到条件返回
      if (result.length >= k) {
        return result
      }
    }
  }
  return result
}

第 451 题

根据字符出现频率排序open in new window

var frequencySort = function (s) {
  // 转换为数组，使用map统计各个元素出现的次数
  let nums = s.split('')
  let map = new Map()
  for (let i = 0; i < nums.length; i++) {
    if (map.has(nums[i])) {
      map.set(nums[i], map.get(nums[i]) + 1)
    } else {
      map.set(nums[i], 1)
    }
  }
  // 按照出现次数将元素放在数组对应的位置
  // 例如出现3次的就放在数组的第3个位置
  // 因为可能有相同次数的，所以每个位置都用数组保存
  let arr = []
  map.forEach((value, key) => {
    if (arr[value]) {
      arr[value].push(key)
    } else {
      arr[value] = [key]
    }
  })
  // 定义返回结果
  let result = ''
  for (let i = sArr.length; i >= 0; i--) {
    if (sArr[i]) {
      // 当前项有值时，将当前项里面的每一个字符都重复下标的次数存入结果
      sArr[i].forEach((item) => {
        result += item.repeat(i)
      })
    }
  }
  return result.join('')
}