Soohan-Kim / Leetcode

0 stars 0 forks source link

[HW 4] Asteroid Collision #735. #49

Open Soohan-Kim opened 3 weeks ago

Soohan-Kim commented 3 weeks ago

https://leetcode.com/problems/asteroid-collision/description/

Soohan-Kim commented 3 weeks ago
class Solution:
    def asteroidCollision(self, asteroids: List[int]) -> List[int]:
        i = 0
        while i < len(asteroids) - 1:
            if (asteroids[i] > 0 and asteroids[i+1] < 0):
                if abs(asteroids[i]) == abs(asteroids[i+1]):
                    asteroids.pop(i)
                    asteroids.pop(i)
                    i = max(0, i - 1)
                else:
                    asteroids.pop(i) if abs(asteroids[i]) < abs(asteroids[i+1]) else asteroids.pop(i+1)
                    i = max(0, i - 1)
            else:
                i += 1
        return asteroids