力扣 382. 链表随机节点

题目

给你一个单链表，随机选择链表的一个节点，并返回相应的节点值。每个节点 被选中的概率一样 。

实现 Solution 类：

Solution(ListNode head) 使用整数数组初始化对象。
int getRandom() 从链表中随机选择一个节点并返回该节点的值。链表中所有节点被选中的概率相等。

示例

输入
[“Solution”, “getRandom”, “getRandom”, “getRandom”, “getRandom”, “getRandom”]
[[[1, 2, 3]], [], [], [], [], []]
输出
[null, 1, 3, 2, 2, 3]

解释
Solution solution = new Solution([1, 2, 3]);
solution.getRandom(); // 返回 1
solution.getRandom(); // 返回 3
solution.getRandom(); // 返回 2
solution.getRandom(); // 返回 2
solution.getRandom(); // 返回 3
// getRandom() 方法应随机返回 1、2、3中的一个，每个元素被返回的概率相等。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/linked-list-random-node
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：模拟

1、记录所有链表元素。
2、利用随机函数返回索引。

Java实现

class Solution {
    Random r = new Random();
    List<Integer> res = new ArrayList<>();
    public Solution(ListNode head) {
        while (head != null) {
            res.add(head.val);
            head = head.next;
        }
    }
    
    public int getRandom() {
        return res.get(r.nextInt(res.size()));
    }
}

方法2：水塘抽样

Java实现

class Solution {
    ListNode head;
    Random random = new Random();

    public Solution(ListNode head) {
        this.head = head;
    }
    
    public int getRandom() {
        int i = 1, res = -1;
        ListNode node = head;
        
        while (node != null) {
            if (random.nextInt(i) == 0) res = node.val;
            node = node.next;
            i++;
        }
        return res;
    }
}