码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode: Linked List Random Node

时间:2016-11-25 11:30:29      阅读:157      评论:0      收藏:0      [点我收藏+]

标签:更改   efi   lin   returns   anti   ati   not   http   called   

这题参照http://blog.jobbole.com/42550/


用的蓄水池算法,即更改ans的概率为1/(当前length)

 1 /**
 2  * Definition for singly-linked list.
 3  * public class ListNode {
 4  *     int val;
 5  *     ListNode next;
 6  *     ListNode(int x) { val = x; }
 7  * }
 8  */
 9 public class Solution {
10 
11     private ListNode head;
12     /** @param head The linked list‘s head.
13         Note that the head is guaranteed to be not null, so it contains at least one node. */
14     public Solution(ListNode head) {
15         this.head = head;
16     }
17     
18     /** Returns a random node‘s value. */
19     public int getRandom() {
20         ListNode p = head;
21         int ans = p.val;
22         for (int i = 1; p.next != null; i++) {
23             p = p.next;
24             if ((int)(Math.random() * (i + 1)) == i) 
25                 ans = p.val;
26         }
27         return ans;
28     }
29 }
30 
31 /**
32  * Your Solution object will be instantiated and called as such:
33  * Solution obj = new Solution(head);
34  * int param_1 = obj.getRandom();
35  */

 

LeetCode: Linked List Random Node

标签:更改   efi   lin   returns   anti   ati   not   http   called   

原文地址:http://www.cnblogs.com/yingzhongwen/p/6100576.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!