LWC 60:735. Asteroid Collision

LWC 60:735. Asteroid Collision

传送门:735. Asteroid Collision

Problem:

We are given an array asteroids of integers representing asteroids in a row.

For each asteroid, the absolute value represents its size, and the sign represents its direction (positive meaning right, negative meaning left). Each asteroid moves at the same speed.

Find out the state of the asteroids after all collisions. If two asteroids meet, the smaller one will explode. If both are the same size, both will explode. Two asteroids moving in the same direction will never meet.

Example 1:

Input:
asteroids = [5, 10, -5]
Output: [5, 10]
Explanation:
The 10 and -5 collide resulting in 10. The 5 and 10 never collide.

Example 2:

Input:
asteroids = [8, -8]
Output: []
Explanation:
The 8 and -8 collide exploding each other.

Example 3:

Input:
asteroids = [10, 2, -5]
Output: [10]
Explanation:
The 2 and -5 collide resulting in -5. The 10 and -5 collide resulting in 10.

Example 4:

Input:
asteroids = [-2, -1, 1, 2]
Output: [-2, -1, 1, 2]
Explanation:
The -2 and -1 are moving left, while the 1 and 2 are moving right.
Asteroids moving the same direction never meet, so no asteroids will meet each other.

Note:

  • The length of asteroids will be at most 10000.
  • Each asteroid will be a non-zero integer in the range [-1000, 1000].

思路:
问题的归简,相碰的行星一定是→←这个方向,所以当遇到前后两行星属于这种方向时,则缩小规模,如果质量相等则都消失,否则质量较小的消失。采用栈的形式始终维持新鲜的最右位置和加入的行星比较即可。

Java版:

List{Integer} 转 int[] : list.stream().mapToInt(i -> i).toArray();

    public int[] asteroidCollision(int[] asteroids) {
        int n = asteroids.length;
        Stack<Integer> stack = new Stack<>();

        for (int i = 0; i < n; ++i) {

            if (stack.isEmpty()) {
                stack.push(asteroids[i]);
                continue;
            }

            int prev = stack.peek();
            boolean pf = prev >= 0;

            int curr =  asteroids[i];
            boolean cf = curr >= 0;

            if (pf == true && cf == false) {
                if (Math.abs(prev) > Math.abs(curr)) {

                }
                else if (Math.abs(prev) < Math.abs(curr)) {
                    stack.pop();
                    i = i - 1;
                }
                else {
                    stack.pop();
                }
            }
            else {
                stack.push(curr);
            }
        }

        return stack.stream().mapToInt(i -> i).toArray();
    }

Python 版本:

    def asteroidCollision(self, asteroids):
        """
        :type asteroids: List[int]
        :rtype: List[int]
        """
        stack = []
        i = 0
        n = len(asteroids)
        while i < n:
            if stack == []:
                stack.append(asteroids[i])
                i = i + 1
                continue

            prev = stack[-1]
            curr = asteroids[i]
            if prev >= 0 and curr < 0:
                if abs(prev) > abs(curr):
                    pass
                elif abs(prev) < abs(curr):
                    stack.pop()
                    i = i - 1
                else:
                    stack.pop()
            else:
                stack.append(curr)
            i = i + 1
        return stack
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值