mirror of
https://github.com/emirpasic/gods
synced 2024-11-06 15:20:25 +00:00
122 lines
3.5 KiB
Go
122 lines
3.5 KiB
Go
/*
|
|
Copyright (c) 2015, Emir Pasic
|
|
All rights reserved.
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions are met:
|
|
|
|
* Redistributions of source code must retain the above copyright notice, this
|
|
list of conditions and the following disclaimer.
|
|
|
|
* Redistributions in binary form must reproduce the above copyright notice,
|
|
this list of conditions and the following disclaimer in the documentation
|
|
and/or other materials provided with the distribution.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
|
|
FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
|
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
|
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
|
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
// Implementation of stack using linked list.
|
|
// Used by red-black tree during in-order traversal.
|
|
// Structure is not thread safe.
|
|
// References: http://en.wikipedia.org/wiki/Stack_%28abstract_data_type%29
|
|
|
|
package linkedliststack
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/emirpasic/gods/stacks"
|
|
"strings"
|
|
)
|
|
|
|
func assertInterfaceImplementation() {
|
|
var _ stacks.Interface = (*Stack)(nil)
|
|
}
|
|
|
|
type Stack struct {
|
|
top *element
|
|
size int
|
|
}
|
|
|
|
type element struct {
|
|
value interface{}
|
|
next *element
|
|
}
|
|
|
|
// Instantiates a new empty stack
|
|
func New() *Stack {
|
|
return &Stack{}
|
|
}
|
|
|
|
// Pushes a value onto the top of the stack
|
|
func (stack *Stack) Push(value interface{}) {
|
|
stack.top = &element{value, stack.top}
|
|
stack.size += 1
|
|
}
|
|
|
|
// Pops (removes) top element on stack and returns it, or nil if stack is empty.
|
|
// Second return parameter is true, unless the stack was empty and there was nothing to pop.
|
|
func (stack *Stack) Pop() (value interface{}, ok bool) {
|
|
if stack.size > 0 {
|
|
value, stack.top = stack.top.value, stack.top.next
|
|
stack.size -= 1
|
|
return value, true
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
// Returns top element on the stack without removing it, or nil if stack is empty.
|
|
// Second return parameter is true, unless the stack was empty and there was nothing to peek.
|
|
func (stack *Stack) Peek() (value interface{}, ok bool) {
|
|
if stack.size > 0 {
|
|
return stack.top.value, true
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
// Returns true if stack does not contain any elements.
|
|
func (stack *Stack) Empty() bool {
|
|
return stack.size == 0
|
|
}
|
|
|
|
// Returns number of elements within the stack.
|
|
func (stack *Stack) Size() int {
|
|
return stack.size
|
|
}
|
|
|
|
// Removes all elements from the stack.
|
|
func (stack *Stack) Clear() {
|
|
stack.top = nil
|
|
stack.size = 0
|
|
}
|
|
|
|
// Returns all elements in the stack (LIFO order).
|
|
func (stack *Stack) Values() []interface{} {
|
|
elements := make([]interface{}, stack.size, stack.size)
|
|
element := stack.top
|
|
for index := 0; element != nil; index++ {
|
|
elements[index] = element.value
|
|
element = element.next
|
|
}
|
|
|
|
return elements
|
|
}
|
|
|
|
func (stack *Stack) String() string {
|
|
str := "LinkedListStack\n"
|
|
values := []string{}
|
|
for _, value := range stack.Values() {
|
|
values = append(values, fmt.Sprintf("%v", value))
|
|
}
|
|
str += strings.Join(values, ", ")
|
|
return str
|
|
}
|