add sol

ductnn · Dec 13, 2023 · f2745f2 · f2745f2
1 parent f188183
commit f2745f2
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/leetcode/leetcode75/Stack/735.AsteroidCollision/asteroidCollision.go b/leetcode/leetcode75/Stack/735.AsteroidCollision/asteroidCollision.go
@@ -0,0 +1,29 @@
+package main
+
+import (
+	"fmt"
+)
+
+func asteroidCollision(asteroids []int) []int {
+	result := []int{}
+	for _, v := range asteroids {
+		if v > 0 {
+			result = append(result, v)
+		} else {
+			for len(result) > 0 && result[len(result)-1] > 0 && result[len(result)-1] < -v {
+				result = result[:len(result)-1]
+			}
+			if len(result) > 0 && result[len(result)-1] == -v {
+				result = result[:len(result)-1]
+			} else if len(result) == 0 || result[len(result)-1] < 0 {
+				result = append(result, v)
+			}
+		}
+	}
+	return result
+}
+
+func main() {
+	asteroids := []int{5, 10, -5}
+	fmt.Println(asteroidCollision(asteroids))
+}