-
Notifications
You must be signed in to change notification settings - Fork 35
/
Copy pathB.java
100 lines (84 loc) · 3.24 KB
/
B.java
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
mport javafx.util.Pair;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Scanner;
public class B {
public static void main(String[] args) throws IOException {
class MinHeap {
public ArrayList<Pair<Integer, Integer>> heap;
MinHeap(ArrayList<Pair<Integer, Integer>> a) {
heap = a;
}
public void siftDown(int i) {
while (2 * i + 1 < heap.size()) {
int left = 2 * i + 1;
int right = 2 * i + 2;
int j = left;
if ((right < heap.size()) && ((heap.get(right).getValue()) < (heap.get(left).getValue()))) {
j = right;
}
if ((heap.get(i).getValue()) <= (heap.get(j).getValue())) {
break;
}
Pair t = heap.get(i);
heap.set(i, heap.get(j));
heap.set(j, t);
i = j;
}
}
public void siftUp(int i) {
while ((heap.get(i).getValue()) < heap.get((i - 1) / 2).getValue()) {
Pair t = heap.get(i);
heap.set(i, heap.get((i - 1) / 2));
heap.set((i - 1) / 2, t);
i = (i - 1) / 2;
}
}
public Pair extractMin() {
Pair min = heap.get(0);
heap.set(0, heap.get(heap.size() - 1));
heap.remove((heap.size() - 1));
siftDown(0);
return min;
}
public void insert(Pair k) {
heap.add(k);
siftUp((heap.size() - 1));
}
}
MinHeap myheap = new MinHeap(new ArrayList<Pair<Integer, Integer>>());
Scanner scanner = new Scanner(new File("priorityqueue.in"));
FileWriter fileWriter = new FileWriter(new File("priorityqueue.out"));
int count = 0;
while (scanner.hasNextLine()) {
String split[] = scanner.nextLine().split(" ");
count++;
switch (split[0].charAt(0)) {
case 'p':
myheap.insert(new Pair<Integer, Integer>(count, Integer.parseInt(split[1])));
break;
case 'e':
if (myheap.heap.size() < 1) {
fileWriter.write("*" + "\r\n");
fileWriter.flush();
} else {
fileWriter.write(myheap.extractMin().getValue() + "\r\n");
fileWriter.flush();
}
break;
case 'd':
int num1 = Integer.parseInt(split[1]);
for (int i = 0; i < myheap.heap.size(); i++) {
if (myheap.heap.get(i).getKey() == num1) {
myheap.heap.set(i, new Pair<Integer, Integer>(num1, Integer.parseInt(split[2])));
myheap.siftUp(i);
break;
}
}
break;
}
}
}
}