统计道路上的碰撞次数
在一条无限长的公路上有 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 <= 105directions[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
One More Thing
这道题其实还是一个脑筋急转弯。。。 去除掉左边向左的车,和右边向右的车之后,剩余车辆的数量 减去 停在原地的车的数量,就是碰撞总次数,因为剩下不是S的车都会碰撞。。。。。。
周赛题目确实挺有趣的,还能这么玩,学习了!