Golang Leetcode 382. Linked List Random Node.go

思路

主要是如何获得随机的效果

code

type ListNode struct {
	Val  int
	Next *ListNode
}

type Solution struct {
	count int
	head  *ListNode
	pool  []int
}

/** @param head The linked list's head.
  Note that the head is guaranteed to be not null, so it contains at least one node. */
func Constructor(head *ListNode) Solution {
	var s Solution
	s.head = head
	return s
}

/** Returns a random node's value. */
func (this *Solution) GetRandom() int {
	r := rand.Intn(this.count + 1)
	this.count++
	if r == this.count-1 && this.head != nil {
		t := this.head
		this.head = this.head.Next
		this.pool = append(this.pool, t.Val)
		return t.Val
	} else {
		return this.pool[r%len(this.pool)]
	}

}

更多内容请移步我的repo：https://github.com/anakin/golang-leetcode