日渐头秃的代码日记 -- 3 LC第285周赛第二题

156 阅读2分钟

统计道路上的碰撞次数

在一条无限长的公路上有 n 辆汽车正在行驶。汽车按从左到右的顺序按从 0 到 n - 1 编号,每辆车都在一个 独特的 位置。

给你一个下标从 0 开始的字符串 directions ,长度为 n 。directions[i] 可以是 'L''R' 或 'S' 分别表示第 i 辆车是向  、向  或者 停留 在当前位置。每辆车移动时 速度相同 。

碰撞次数可以按下述方式计算:

  • 当两辆移动方向 相反 的车相撞时,碰撞次数加 2 。
  • 当一辆移动的车和一辆静止的车相撞时,碰撞次数加 1 。

碰撞发生后,涉及的车辆将无法继续移动并停留在碰撞位置。除此之外,汽车不能改变它们的状态或移动方向。

返回在这条道路上发生的 碰撞总次数 。

 

示例 1:

输入: directions = "RLRSLL"
输出: 5
解释:
将会在道路上发生的碰撞列出如下:
- 车 0 和车 1 会互相碰撞。由于它们按相反方向移动,碰撞数量变为 0 + 2 = 2 。
- 车 2 和车 3 会互相碰撞。由于 3 是静止的,碰撞数量变为 2 + 1 = 3 。
- 车 3 和车 4 会互相碰撞。由于 3 是静止的,碰撞数量变为 3 + 1 = 4 。
- 车 4 和车 5 会互相碰撞。在车 4 和车 3 碰撞之后,车 4 会待在碰撞位置,接着和车 5 碰撞。碰撞数量变为 4 + 1 = 5 。
因此,将会在道路上发生的碰撞总次数是 5 。

示例 2:

输入: directions = "LLRR"
输出: 0
解释:
不存在会发生碰撞的车辆。因此,将会在道路上发生的碰撞总次数是 0 。

 

提示:

  • 1 <= directions.length <= 105
  • directions[i] 的值为 'L''R' 或 'S'

解法与思路

这道题使用栈进行路况模拟就可以,从左边开始,发现后一辆车的状态与栈顶会碰撞,则弹出栈顶,并把碰撞后的结果入栈,对碰撞次数进行 +1 或者 +2 处理。 对了,也要多RR, RRR 这种连续向右碰撞进行记录。

class Solution:
    def countCollisions(self, directions: str) -> int:
        stack = []
        collisions = 0
        length = len(directions)
        if length == 1:
            return 0
        directions += "O"
        rr = 0
        for i in range(length):
            if not stack:
                stack.append(directions[i])
            if i >= 1:
                d = stack[-1]
                n = directions[i]
                if d + n == "RR":
                    rr += 1
                    stack.append(directions[i])
                elif d + n == "RS":
                    collisions += 1
                    stack.pop()
                    stack.append("S")
                    stack.append("S")
                    collisions += rr
                    rr = 0
                elif d + n == "RL":
                    collisions += 2
                    stack.pop()
                    stack.append("S")
                    stack.append("S")
                    collisions += rr
                    rr = 0
                elif d + n == "SL":
                    collisions += 1
                    stack.pop()
                    stack.append("S")
                    stack.append("S")
                else:
                    stack.append(directions[i])
        return collisions

image.png

One More Thing

这道题其实还是一个脑筋急转弯。。。 去除掉左边向左的车,和右边向右的车之后,剩余车辆的数量 减去 停在原地的车的数量,就是碰撞总次数,因为剩下不是S的车都会碰撞。。。。。。

周赛题目确实挺有趣的,还能这么玩,学习了!