focus/list/list.go

package list

type Node struct {
	next  *Node
	value interface{}
}

func (node *Node) Value() interface{} {
	return node.value
}

type List struct {
	head *Node
	size uint
}

func New() *List {
	return &List{}
}

func (l *List) Size() uint {
	return l.size
}

func (l *List) Push(v interface{}) {
	l.size++
	if l.head == nil {
		l.head = &Node{value: v}
		return
	}
	l.head = &Node{value: v, next: l.head}
}

func (l *List) PushNode(n *Node) {
	l.size++
	if l.head == nil {
		l.head = n
		return
	}

	n.next = l.head
	l.head = n
}

func (l *List) Pop() (result interface{}, found bool) {
	if n, ok := l.PopNode(); ok {
		return n.value, ok
	}
	return nil, false
}

func (l *List) PopNode() (result *Node, found bool) {
	if l.head == nil {
		return nil, false
	}

	result = l.head
	found = true
	l.head = result.next
	result.next = nil
	l.size--
	return
}

func (l *List) Remove(idx uint) (result *Node, found bool) {
	if l.size == 0 {
		return nil, false
	}

	if idx == 0 {
		result = l.head
		found = true
		l.head = result.next
		result.next = nil
		l.size--
		return
	}

	for cur := l.head; cur.next != nil; cur = cur.next {
		if idx == 1 {
			l.size--
			result = cur.next
			found = true
			cur.next = result.next
			result.next = nil
			return
		}
		idx--
	}

	return nil, false
}

func (l *List) Values() (result []interface{}) {
	l.Traversal(func(cur *Node) bool {
		result = append(result, cur.value)
		return true
	})
	return
}

func (l *List) Traversal(every func(*Node) bool) {
	for cur := l.head; cur != nil; cur = cur.next {
		if !every(cur) {
			break
		}
	}
}