LeetCode 735: Asteroid Collision - スタックで衝突判定を美しく解く

Tue, 13 Jan 2026 19:30:00 +0900

問題概要

整数で表される小惑星の配列 asteroids が与えられる。各小惑星について：

絶対値：大きさ
符号：方向（正=右、負=左）
全て同じ速度で移動

衝突ルール：

小さい方が爆発
同じ大きさなら両方爆発
同じ方向に移動する小惑星は衝突しない

全ての衝突後の状態を返せ。

失敗した実装

from collections import deque

class Solution:
    def asteroidCollision(self, asteroids: List[int]) -> List[int]:
        stack = []
        for aster in asteroids:
            if len(stack) <= 0 or (aster <= 0) == (stack[-1] <= 0):
                stack.append(aster)
            else:
                while len(stack) > 0 and (aster <= 0) != (stack[-1] <= 0) and abs(aster) > abs(stack[-1]):
                    stack.pop()
                if len(stack) <= 0 or (stack[-1] <= 0) == (aster <= 0):
                    stack.append(aster)
                elif abs(aster) == abs(stack[-1]):
                    stack.pop()
        return stack

問題点

同じ条件判定 if len(stack) <= 0 or (stack[-1] <= 0) == (aster <= 0) が2箇所に重複
制御フローが複雑で読みにくい
(aster <= 0) != (stack[-1] <= 0) は「符号が異なる」を検出するが、衝突しないケースも含む

最適解

class Solution:
    def asteroidCollision(self, asteroids: List[int]) -> List[int]:
        stack = []
        
        for asteroid in asteroids:
            while stack and asteroid < 0 < stack[-1]:
                # 右向き vs 左向きの衝突が発生
                if abs(stack[-1]) < abs(asteroid):
                    # 右向きが小さい → 爆発して次の右向きとも衝突判定
                    stack.pop()
                    continue
                elif abs(stack[-1]) == abs(asteroid):
                    # 同じ大きさ → 両方爆発
                    stack.pop()
                break
            else:
                # 衝突しなかった or 左向きが勝った
                stack.append(asteroid)
        
        return stack

わかったこと

1. ループ条件の本質

asteroid < 0 < stack[-1]

これは (asteroid < 0) and (0 < stack[-1]) と同じ。つまり：

Stack on 怠惰技術ブログ

LeetCode 735: Asteroid Collision - スタックで衝突判定を美しく解く

問題概要

失敗した実装

問題点

最適解

わかったこと

1. ループ条件の本質