2391. 收集垃圾的最少总时间

收集垃圾的最少总时间

• 难度: medium
• 原始链接: https://leetcode.cn/problems/minimum-amount-of-time-to-collect-garbage
• 标签: 一次遍历

解法一: 一次遍历

go

func garbageCollection(garbage []string, travel []int) int {
  n := len(garbage)
  ans := 0
  cur := 0
  countMap := map[byte]int{
    'M': 0,
    'P': 0,
    'G': 0,
  }
  travel = append(travel, 0)
  for i := 0; i < n; i++ {
    m := len(garbage[i])
    ans += m
    for j := 0; j < m; j++ {
      countMap[garbage[i][j]] = cur
    }
    cur += travel[i]
  }
  return ans + countMap['M'] + countMap['P'] + countMap['G']
}