go: add «735. Asteroid Collision»
Signed-off-by: Matej Focko <me@mfocko.xyz>
This commit is contained in:
parent
23cd6ac490
commit
dca74915b4
1 changed files with 33 additions and 0 deletions
33
go/asteroid-collision.go
Normal file
33
go/asteroid-collision.go
Normal file
|
@ -0,0 +1,33 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import as "github.com/emirpasic/gods/v2/stacks/arraystack"
|
||||||
|
|
||||||
|
func asteroidCollision(asteroids []int) []int {
|
||||||
|
st := as.New[int]()
|
||||||
|
|
||||||
|
for _, asteroid := range asteroids {
|
||||||
|
alive := true
|
||||||
|
|
||||||
|
for top, ok := st.Peek(); ok && asteroid < 0 && top > 0; top, ok = st.Peek() {
|
||||||
|
if top < -asteroid {
|
||||||
|
st.Pop()
|
||||||
|
continue
|
||||||
|
} else if top == -asteroid {
|
||||||
|
st.Pop()
|
||||||
|
}
|
||||||
|
|
||||||
|
alive = false
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
if alive {
|
||||||
|
st.Push(asteroid)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
result := make([]int, st.Size())
|
||||||
|
for i := st.Size() - 1; i >= 0; i-- {
|
||||||
|
result[i], _ = st.Pop()
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
Loading…
Reference in a new issue