小行星的碰撞

2023-12-13 05:35| 来源: 网络整理| 查看: 265

给定一个整数数组，代表一行小行星。对于每个小行星，绝对值表示其大小，符号表示其方向（正意义右，负意义左）。每个小行星以相同的速度移动。返回所有碰撞发生后小行星的状态。如果两颗小行星相遇，则较小的小行星会爆炸。如果两者的大小相同，则两者都会爆炸。沿同一方向移动的两颗小行星永远不会相遇。

asteroid-collision

样例

样例 1:

输入: [5, 10, -5] 输出: [5, 10] 解释: 10 和 -5 碰撞得 10. 而 5 和 10 永远不会碰撞.

样例 2:

输入: [10, 2, -5] 输出: [10] 解释: 2 和 -5 碰撞后得到 -5. 然后 10 和 -5 碰撞剩下10.

思路 1、使用 stack 来解题 2、循环数组 item 1、如果大于 0 直接入栈 2、如果小于 0 遍历非空 stach 如果栈顶大于 0 并且栈顶小于item 的绝对值，则出栈 3、如果栈为空或者栈顶小于 0 入栈 4、如果 item 跟栈顶相加为 0 则出栈

public class Solution { /** * @param asteroids: a list of integers * @return: return a list of integers */ public int[] asteroidCollision(int[] asteroids) { // write your code here Stack stack = new Stack(); for (int item : asteroids) { if (item > 0) { stack.push(item); } else { while (!stack.isEmpty() && stack.peek() > 0 && stack.peek() < Math.abs(item)) { stack.pop(); } if (stack.isEmpty() || stack.peek() < 0) { stack.push(item); } else if (item + stack.peek() == 0) { stack.pop(); } } } int[] result = new int[stack.size()]; for (int i = stack.size() - 1; i >= 0; i--) { result[i] = stack.pop(); } return result; } }

源码地址: https://github.com/xingfu0809/Java-LintCode

【本文地址】

小行星的碰撞

小行星的碰撞

今日新闻

推荐新闻