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