Select a random number from stream, with O(1) space Last Updated : 20 Apr, 2023 Comments Improve Suggest changes Like Article Like Report Given a stream of numbers, generate a random number from the stream. You are allowed to use only O(1) space and the input is in the form of a stream, so can't store the previously seen numbers. So how do we generate a random number from the whole stream such that the probability of picking any number is 1/n. with O(1) extra space? This problem is a variation of Reservoir Sampling. Here the value of k is 1.1) Initialize 'count' as 0, 'count' is used to store count of numbers seen so far in stream. 2) For each number 'x' from stream, do following .....a) Increment 'count' by 1. .....b) If count is 1, set result as x, and return result. .....c) Generate a random number from 0 to 'count-1'. Let the generated random number be i. .....d) If i is equal to 'count - 1', update the result as x. C++ // An efficient C++ program to randomly select // a number from stream of numbers. #include <bits/stdc++.h> #include <time.h> using namespace std; // A function to randomly select a item // from stream[0], stream[1], .. stream[i-1] int selectRandom(int x) { static int res; // The resultant random number static int count = 0; // Count of numbers visited // so far in stream count++; // increment count of numbers seen so far // If this is the first element from stream, // return it if (count == 1) res = x; else { // Generate a random number from 0 to count - 1 int i = rand() % count; // Replace the prev random number with // new number with 1/count probability if (i == count - 1) res = x; } return res; } // Driver Code int main() { int stream[] = {1, 2, 3, 4}; int n = sizeof(stream) / sizeof(stream[0]); // Use a different seed value for every run. srand(time(NULL)); for (int i = 0; i < n; ++i) cout << "Random number from first " << i + 1 << " numbers is " << selectRandom(stream[i]) << endl; return 0; } // This is code is contributed by rathbhupendra C // An efficient C program to randomly select a number from stream of numbers. #include <stdio.h> #include <stdlib.h> #include <time.h> // A function to randomly select a item from stream[0], stream[1], .. stream[i-1] int selectRandom(int x) { static int res; // The resultant random number static int count = 0; //Count of numbers visited so far in stream count++; // increment count of numbers seen so far // If this is the first element from stream, return it if (count == 1) res = x; else { // Generate a random number from 0 to count - 1 int i = rand() % count; // Replace the prev random number with new number with 1/count probability if (i == count - 1) res = x; } return res; } // Driver program to test above function. int main() { int stream[] = {1, 2, 3, 4}; int n = sizeof(stream)/sizeof(stream[0]); // Use a different seed value for every run. srand(time(NULL)); for (int i = 0; i < n; ++i) printf("Random number from first %d numbers is %d \n", i+1, selectRandom(stream[i])); return 0; } Java //An efficient Java program to randomly select a number from stream of numbers. import java.util.Random; public class GFG { static int res = 0; // The resultant random number static int count = 0; //Count of numbers visited so far in stream //A method to randomly select a item from stream[0], stream[1], .. stream[i-1] static int selectRandom(int x) { count++; // increment count of numbers seen so far // If this is the first element from stream, return it if (count == 1) res = x; else { // Generate a random number from 0 to count - 1 Random r = new Random(); int i = r.nextInt(count); // Replace the prev random number with new number with 1/count probability if(i == count - 1) res = x; } return res; } // Driver program to test above function. public static void main(String[] args) { int stream[] = {1, 2, 3, 4}; int n = stream.length; for(int i = 0; i < n; i++) System.out.println("Random number from first " + (i+1) + " numbers is " + selectRandom(stream[i])); } } //This code is contributed by Sumit Ghosh Python3 # An efficient python3 program # to randomly select a number # from stream of numbers. import random # A function to randomly select a item # from stream[0], stream[1], .. stream[i-1] # The resultant random number res=0 # Count of numbers visited # so far in stream count=0 def selectRandom(x): global res global count # increment count of numbers # seen so far count += 1; # If this is the first element # from stream, return it if (count == 1): res = x; else: # Generate a random number # from 0 to count - 1 i = random.randrange(count); # Replace the prev random number # with new number with 1/count # probability if (i == count - 1): res = x; return res; # Driver Code stream = [1, 2, 3, 4]; n = len(stream); # Use a different seed value # for every run. for i in range (n): print("Random number from first", (i + 1), "numbers is", selectRandom(stream[i])); # This code is contributed by mits C# // An efficient C# program to randomly // select a number from stream of numbers. using System; class GFG { // The resultant random number static int res = 0; // Count of numbers visited // so far in stream static int count = 0; // A method to randomly select // a item from stream[0], // stream[1], .. stream[i-1] static int selectRandom(int x) { // increment count of // numbers seen so far count++; // If this is the first // element from stream, // return it if (count == 1) res = x; else { // Generate a random number // from 0 to count - 1 Random r = new Random(); int i = r.Next(count); // Replace the prev random // number with new number // with 1/count probability if(i == count - 1) res = x; } return res; } // Driver Code public static void Main() { int[] stream = {1, 2, 3, 4}; int n = stream.Length; for(int i = 0; i < n; i++) Console.WriteLine("Random number from " + "first {0} numbers is {1}" , i + 1, selectRandom(stream[i])); } } // This code is contributed by mits PHP <?php // An efficient php program to randomly // select a number from stream of numbers. // A function to randomly select a item // from stream[0], stream[1], .. stream[i-1] function selectRandom($x) { // The resultant random number $res; // Count of numbers visited so far // in stream $count = 0; // increment count of numbers seen // so far $count++; // If this is the first element // from stream, return it if ($count == 1) $res = $x; else { // Generate a random number from // 0 to count - 1 $i = rand() % $count; // Replace the prev random number // with new number with 1/count // probability if (i == $count - 1) $res = $x; } return $res; } // Driver program to test above function. $stream = array(1, 2, 3, 4); $n = sizeof($stream)/sizeof($stream[0]); // Use a different seed value for // every run. srand(time(NULL)); for ($i = 0; $i < $n; ++$i) echo "Random number from first ", $i+1, "numbers is " , selectRandom($stream[$i]), "\n" ; // This code is contributed by nitin mittal. ?> JavaScript <script> //An efficient Javascript program to randomly select a number from stream of numbers. let res = 0; // The resultant random number let count = 0; //Count of numbers visited so far in stream //A method to randomly select a item from stream[0], stream[1], .. stream[i-1] function selectRandom(x) { count++; // increment count of numbers seen so far // If this is the first element from stream, return it if (count == 1) res = x; else { // Generate a random number from 0 to count - 1 let i = Math.floor(Math.random()*(count)); // Replace the prev random number with new number with 1/count probability if(i == count - 1) res = x; } return res; } // Driver program to test above function. let stream=[1, 2, 3, 4]; let n = stream.length; for(let i = 0; i < n; i++) document.write("Random number from first " + (i+1) + " numbers is " + selectRandom(stream[i])+"<br>"); // This code is contributed by avanitrachhadiya2155 </script> OutputRandom number from first 1 numbers is 1 Random number from first 2 numbers is 1 Random number from first 3 numbers is 1 Random number from first 4 numbers is 4 Time Complexity: O(n)Auxiliary Space: O(1)How does this work We need to prove that every element is picked with 1/n probability where n is the number of items seen so far. For every new stream item x, we pick a random number from 0 to 'count -1', if the picked number is 'count-1', we replace the previous result with x.To simplify proof, let us first consider the last element, the last element replaces the previously-stored result with 1/n probability. So the probability of getting the last element as the result is 1/n.Let us now talk about the second last element. When the second last element processed the first time, the probability that it replaced the previous result is 1/(n-1). The probability that the previous result stays when the nth item is considered is (n-1)/n. So the probability that the second last element is picked in the last iteration is [1/(n-1)] * [(n-1)/n] which is 1/n. Similarly, we can prove for third last element and others.References: Reservoir Sampling Method 2: generate a random number from the stream with numpy random.choice() method. C++ #include <iostream> #include <cstdlib> #include <ctime> using namespace std; int main() { int stream[] = {1, 4, 5, 2, 7}; int length = sizeof(stream)/sizeof(stream[0]); // Set the seed for the random number generator srand(time(nullptr)); // Generate a random index between 0 and the length of the array int randomIndex = rand() % length; // Get the random element from the array using the generated index int randomNum = stream[randomIndex]; // Print the random number cout << "Random number is " << randomNum << endl; return 0; } // This code is contributed by shivhack999 Java import java.util.Random; public class Main { public static void main(String[] args) { int[] stream = { 1, 4, 5, 2, 7 }; // Using Random.nextInt() to get a random index int randomIndex = new Random().nextInt(stream.length); // Getting a random number from the stream array int randomNum = stream[randomIndex]; // Logging the random number System.out.println("Random number is " + randomNum); } } // This code is contributed by user_dtewbxkn77n Python3 import numpy as np # initializing list stream = [1, 4, 5, 2, 7] # using random.choice() to # get a random number random_num = np.random.choice(stream) # printing random number print("random number is ",random_num) C# using System; class Program { static void Main() { int[] stream = { 1, 4, 5, 2, 7 }; // Create an instance of the Random class Random rand = new Random(); // Generate a random index between 0 and the length of the array int randomIndex = rand.Next(stream.Length); // Get the random element from the array using the generated index int randomNum = stream[randomIndex]; // Print the random number Console.WriteLine("Random number is " + randomNum); } } JavaScript // initializing array const stream = [1, 4, 5, 2, 7]; // using Math.floor() and Math.random() to // get a random index const randomIndex = Math.floor(Math.random() * stream.length); // getting a random number from the stream array const randomNum = stream[randomIndex]; // logging the random number console.log(`random number is ${randomNum}`); OutputRandom number is 2 Time Complexity: O(n)Auxiliary Space: O(1) Comment More infoAdvertise with us Next Article Random number generator in arbitrary probability distribution fashion kartik Follow Improve Article Tags : Mathematical Randomized DSA Random Algorithms array-stream +1 More Practice Tags : Mathematical Similar Reads Randomized Algorithms Randomized algorithms in data structures and algorithms (DSA) are algorithms that use randomness in their computations to achieve a desired outcome. These algorithms introduce randomness to improve efficiency or simplify the algorithm design. By incorporating random choices into their processes, ran 2 min read Random Variable Random variable is a fundamental concept in statistics that bridges the gap between theoretical probability and real-world data. A Random variable in statistics is a function that assigns a real value to an outcome in the sample space of a random experiment. For example: if you roll a die, you can a 10 min read Binomial Random Variables In this post, we'll discuss Binomial Random Variables.Prerequisite : Random Variables A specific type of discrete random variable that counts how often a particular event occurs in a fixed number of tries or trials. For a variable to be a binomial random variable, ALL of the following conditions mus 8 min read Randomized Algorithms | Set 0 (Mathematical Background) Conditional Probability Conditional probability P(A | B) indicates the probability of even 'A' happening given that the even B happened.P(A|B) = \frac{P(A\cap B)}{P(B)} We can easily understand above formula using below diagram. Since B has already happened, the sample space reduces to B. So the pro 3 min read Randomized Algorithms | Set 1 (Introduction and Analysis) What is a Randomized Algorithm? An algorithm that uses random numbers to decide what to do next anywhere in its logic is called a Randomized Algorithm. For example, in Randomized Quick Sort, we use a random number to pick the next pivot (or we randomly shuffle the array). And in Karger's algorithm, 5 min read Randomized Algorithms | Set 2 (Classification and Applications) We strongly recommend to refer below post as a prerequisite of this. Randomized Algorithms | Set 1 (Introduction and Analysis) Classification Randomized algorithms are classified in two categories. Las Vegas: A Las Vegas algorithm were introduced by Laszlo Babai in 1979. A Las Vegas algorithm is an 13 min read Randomized Algorithms | Set 3 (1/2 Approximate Median) Time Complexity: We use a set provided by the STL in C++. In STL Set, insertion for each element takes O(log k). So for k insertions, time taken is O (k log k). Now replacing k with c log n =>O(c log n (log (clog n))) =>O (log n (log log n)) How is probability of error less than 2/n2? Algorithm make 2 min read Easy problems on randomized algorithmsWrite a function that generates one of 3 numbers according to given probabilitiesYou are given a function rand(a, b) which generates equiprobable random numbers between [a, b] inclusive. Generate 3 numbers x, y, z with probability P(x), P(y), P(z) such that P(x) + P(y) + P(z) = 1 using the given rand(a,b) function.The idea is to utilize the equiprobable feature of the rand(a,b) 5 min read Generate 0 and 1 with 25% and 75% probabilityGiven a function rand50() that returns 0 or 1 with equal probability, write a function that returns 1 with 75% probability and 0 with 25% probability using rand50() only. Minimize the number of calls to the rand50() method. Also, the use of any other library function and floating-point arithmetic ar 13 min read Implement rand3() using rand2()Given a function rand2() that returns 0 or 1 with equal probability, implement rand3() using rand2() that returns 0, 1 or 2 with equal probability. Minimize the number of calls to rand2() method. Also, use of any other library function and floating point arithmetic are not allowed. The idea is to us 6 min read Birthday ParadoxHow many people must be there in a room to make the probability 100% that at-least two people in the room have same birthday? Answer: 367 (since there are 366 possible birthdays, including February 29). The above question was simple. Try the below question yourself. How many people must be there in 7 min read Expectation or expected value of an arrayExpectation or expected value of any group of numbers in probability is the long-run average value of repetitions of the experiment it represents. For example, the expected value in rolling a six-sided die is 3.5, because the average of all the numbers that come up in an extremely large number of ro 5 min read Shuffle a deck of cardsGiven a deck of cards, the task is to shuffle them. Asked in Amazon Interview Prerequisite : Shuffle a given array Algorithm: 1. First, fill the array with the values in order. 2. Go through the array and exchange each element with the randomly chosen element in the range from itself to the end. // 5 min read Program to generate CAPTCHA and verify userA CAPTCHA (Completely Automated Public Turing test to tell Computers and Humans Apart) is a test to determine whether the user is human or not.So, the task is to generate unique CAPTCHA every time and to tell whether the user is human or not by asking user to enter the same CAPTCHA as generated auto 6 min read Find an index of maximum occurring element with equal probabilityGiven an array of integers, find the most occurring element of the array and return any one of its indexes randomly with equal probability.Examples: Input: arr[] = [-1, 4, 9, 7, 7, 2, 7, 3, 0, 9, 6, 5, 7, 8, 9] Output: Element with maximum frequency present at index 6 OR Element with maximum frequen 8 min read Randomized Binary Search AlgorithmWe are given a sorted array A[] of n elements. We need to find if x is present in A or not.In binary search we always used middle element, here we will randomly pick one element in given range.In Binary Search we had middle = (start + end)/2 In Randomized binary search we do following Generate a ran 13 min read Medium problems on randomized algorithmsMake a fair coin from a biased coinYou are given a function foo() that represents a biased coin. When foo() is called, it returns 0 with 60% probability, and 1 with 40% probability. Write a new function that returns 0 and 1 with a 50% probability each. Your function should use only foo(), no other library method. Solution: We know fo 6 min read Shuffle a given array using FisherâYates shuffle AlgorithmGiven an array, write a program to generate a random permutation of array elements. This question is also asked as "shuffle a deck of cards" or "randomize a given array". Here shuffle means that every permutation of array element should be equally likely. Let the given array be arr[]. A simple solut 10 min read Expected Number of Trials until SuccessConsider the following famous puzzle. In a country, all families want a boy. They keep having babies till a boy is born. What is the expected ratio of boys and girls in the country? This puzzle can be easily solved if we know following interesting result in probability and expectation. If probabilit 6 min read Strong Password Suggester ProgramGiven a password entered by the user, check its strength and suggest some password if it is not strong. Criteria for strong password is as follows : A password is strong if it has : At least 8 characters At least one special char At least one number At least one upper and one lower case char. Exampl 15 min read QuickSort using Random PivotingIn this article, we will discuss how to implement QuickSort using random pivoting. In QuickSort we first partition the array in place such that all elements to the left of the pivot element are smaller, while all elements to the right of the pivot are greater than the pivot. Then we recursively call 15+ min read Operations on Sparse MatricesGiven two sparse matrices (Sparse Matrix and its representations | Set 1 (Using Arrays and Linked Lists)), perform operations such as add, multiply or transpose of the matrices in their sparse form itself. The result should consist of three sparse matrices, one obtained by adding the two input matri 15+ min read Estimating the value of Pi using Monte CarloMonte Carlo estimation Monte Carlo methods are a broad class of computational algorithms that rely on repeated random sampling to obtain numerical results. One of the basic examples of getting started with the Monte Carlo algorithm is the estimation of Pi. Estimation of Pi The idea is to simulate ra 8 min read Implement rand12() using rand6() in one lineGiven a function, rand6() that returns random numbers from 1 to 6 with equal probability, implement the one-liner function rand12() using rand6() which returns random numbers from 1 to 12 with equal probability. The solution should minimize the number of calls to the rand6() method. Use of any other 7 min read Hard problems on randomized algorithmsGenerate integer from 1 to 7 with equal probabilityGiven a function foo() that returns integers from 1 to 5 with equal probability, write a function that returns integers from 1 to 7 with equal probability using foo() only. Minimize the number of calls to foo() method. Also, use of any other library function is not allowed and no floating point arit 6 min read Implement random-0-6-Generator using the given random-0-1-GeneratorGiven a function random01Generator() that gives you randomly either 0 or 1, implement a function that utilizes this function and generate numbers between 0 and 6(both inclusive). All numbers should have same probabilities of occurrence. Examples: on multiple runs, it gives 3 2 3 6 0 Approach : The i 5 min read Select a random number from stream, with O(1) spaceGiven a stream of numbers, generate a random number from the stream. You are allowed to use only O(1) space and the input is in the form of a stream, so can't store the previously seen numbers. So how do we generate a random number from the whole stream such that the probability of picking any numbe 10 min read Random number generator in arbitrary probability distribution fashionGiven n numbers, each with some frequency of occurrence. Return a random number with probability proportional to its frequency of occurrence. Example: Let following be the given numbers. arr[] = {10, 30, 20, 40} Let following be the frequencies of given numbers. freq[] = {1, 6, 2, 1} The output shou 11 min read Reservoir SamplingReservoir sampling is a family of randomized algorithms for randomly choosing k samples from a list of n items, where n is either a very large or unknown number. Typically n is large enough that the list doesn't fit into main memory. For example, a list of search queries in Google and Facebook.So we 11 min read Linearity of ExpectationPrerequisite: Random Variable This post is about mathematical concepts like expectation, linearity of expectation. It covers one of the required topics to understand Randomized Algorithms. Let us consider the following simple problem. Problem: Given a fair dice with 6 faces, the dice is thrown n tim 4 min read Introduction and implementation of Karger's algorithm for Minimum CutGiven an undirected and unweighted graph, find the smallest cut (smallest number of edges that disconnects the graph into two components). The input graph may have parallel edges. For example consider the following example, the smallest cut has 2 edges. A Simple Solution use Max-Flow based s-t cut a 15+ min read Select a Random Node from a Singly Linked ListGiven a singly linked list, select a random node from the linked list (the probability of picking a node should be 1/N if there are N nodes in the list). You are given a random number generator.Below is a Simple Solution Count the number of nodes by traversing the list. Traverse the list again and s 14 min read Select a Random Node from a tree with equal probabilityGiven a Binary Tree with children Nodes, Return a random Node with an equal Probability of selecting any Node in tree.Consider the given tree with root as 1. 10 / \ 20 30 / \ / \ 40 50 60 70 Examples: Input : getRandom(root); Output : A Random Node From Tree : 3 Input : getRandom(root); Output : A R 8 min read Freivaldâs Algorithm to check if a matrix is product of twoGiven three matrices A, B and C, find if C is a product of A and B. Examples: Input : A = 1 1 1 1 B = 1 1 1 1 C = 2 2 2 2 Output : Yes C = A x B Input : A = 1 1 1 1 1 1 1 1 1 B = 1 1 1 1 1 1 1 1 1 C = 3 3 3 3 1 2 3 3 3 Output : No A simple solution is to find product of A and B and then check if pro 12 min read Random Acyclic Maze Generator with given Entry and Exit pointGiven two integers N and M, the task is to generate any N * M sized maze containing only 0 (representing a wall) and 1 (representing an empty space where one can move) with the entry point as P0 and exit point P1 and there is only one path between any two movable positions. Note: P0 and P1 will be m 15+ min read Like