Leetcode_735_行星碰撞_栈

原创于 2022-07-13 11:32:26 发布 · 144 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #职场和发展

daily_algorithm 同时被 2 个专栏收录

230 篇文章

订阅专栏

数据结构

52 篇文章

订阅专栏

本文解析了一种使用栈数据结构解决空间中 asteroid collision 问题的解决方案。通过遍历数组，判断每个小行星与栈顶元素的关系并进行相应操作，最后返回剩余小行星的顺序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最难的部分是抽象出栈这个数据结构

class Solution {
    public int[] asteroidCollision(int[] asteroids) {
        Deque<Integer> stack = new ArrayDeque<>();
        int len = asteroids.length;
        for (int i = 0; i < len; i++) {
            int asteroid = asteroids[i];
            if (stack.isEmpty()) {
                stack.add(asteroid);
            } else {
                int top = stack.peek();
                if (asteroid > 0 || top < 0 && asteroid < 0) {
                    stack.push(asteroid);
                } else if (Math.abs(asteroid) == Math.abs(top)) {
                    stack.pop();
                } else if (Math.abs(asteroid) > Math.abs(top)) {
                    i--;
                    stack.pop();
                }
            }
        }
        int size = stack.size();
        int[] ans = new int[size];
        for (int i = size - 1; i >= 0; i--) {
            ans[i] = stack.pop();
        }
        return ans;
    }
}