-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathRandomizedQueue.java
More file actions
139 lines (119 loc) · 3.66 KB
/
RandomizedQueue.java
File metadata and controls
139 lines (119 loc) · 3.66 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
/* *****************************************************************************
* Name: Jaya Mukherjee
* Coursera User ID: 123456
* Last modified: June 13, 2021
**************************************************************************** */
import edu.princeton.cs.algs4.StdOut;
import edu.princeton.cs.algs4.StdRandom;
import java.util.Iterator;
import java.util.NoSuchElementException;
public class RandomizedQueue<Item> implements Iterable<Item> {
private Item[] a;
private int size;
// construct an empty randomized queue
public RandomizedQueue() {
a = (Item[]) new Object[1];
size = 0;
}
private void resize(int max) {
if (max > size) {
Item[] temp = (Item[]) new Object[max];
for (int i = 0; i < size; i++) {
temp[i] = a[i];
}
a = temp;
}
}
// is the randomized queue empty?
public boolean isEmpty() {
return size == 0;
}
// return the number of items on the randomized queue
public int size() {
return size;
}
// add the item
public void enqueue(Item item) {
if (item == null) {
throw new IllegalArgumentException();
}
if (size == a.length) {
resize(2 * a.length);
}
a[size++] = item;
}
// remove and return a random item
public Item dequeue() {
if (size == 0) {
throw new NoSuchElementException("there are no more items!");
}
int r = StdRandom.uniform(0, size);
Item item = a[r];
a[r] = a[size - 1];
a[size - 1] = null;
size--;
if (size > 0 && size == a.length / 4) resize(a.length / 2);
return item;
}
// return a random item (but do not remove it)
public Item sample() {
if (size == 0) {
throw new NoSuchElementException("there are no more items!");
}
int r = StdRandom.uniform(0, size);
Item item = a[r];
return item;
}
// return an independent iterator over items in random order
public Iterator<Item> iterator() {
return new ReverseArrayIteratorItem();
}
private class ReverseArrayIteratorItem implements Iterator<Item> {
private int i;
private Item[] iter;
ReverseArrayIteratorItem() {
i = size;
iter = (Item[]) new Object[size];
for (int j = 0; j < size; j++) {
iter[j] = a[j];
}
}
public boolean hasNext() {
return i > 0;
}
public void remove() {
throw new UnsupportedOperationException();
}
public Item next() {
if (i <= 0) {
throw new NoSuchElementException("there are no more items!");
}
int r = StdRandom.uniform(0, i);
Item item = iter[r];
iter[r] = iter[i - 1];
iter[i - 1] = item;
i--;
return item;
}
}
public static void main(String[] args) {
RandomizedQueue<String> queue = new RandomizedQueue<String>();
queue.enqueue("a");
queue.enqueue("b");
queue.enqueue("c");
queue.enqueue("d");
queue.enqueue("e");
queue.dequeue();
StdOut.println(queue.size);
Iterator<String> iter1 = queue.iterator();
Iterator<String> iter2 = queue.iterator();
while (iter1.hasNext()) {
StdOut.println(iter1.next() + ",");
}
StdOut.println();
while (iter2.hasNext()) {
StdOut.println(iter2.next() + ",");
}
StdOut.println();
}
}