-
Notifications
You must be signed in to change notification settings - Fork 47
/
Copy pathmin_pq.cpp
134 lines (123 loc) · 3 KB
/
min_pq.cpp
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
#include <bits/stdc++.h>
using namespace std;
class PriorityQueue
{
vector<int> pq;
public:
PriorityQueue()
{
}
bool isEmpty()
{
return pq.size() == 0;
}
// Return the size of priorityQueue - no of elements present
int getSize()
{
return pq.size();
}
int getMin()
{
if (isEmpty())
return 0; // Priority Queue is empty
return pq[0];
}
void insert(int element)
{
pq.push_back(element);
int childIndex = pq.size() - 1;
//Compare data we just inserted to parents data. If its less than parents we swap
//Continue to do this until we reach the root node
while (childIndex > 0)
{
int parentIndex = (childIndex - 1) / 2;
if (pq[childIndex] < pq[parentIndex])
{
int temp = pq[childIndex];
pq[childIndex] = pq[parentIndex];
pq[parentIndex] = temp;
}
else
{
break;
}
childIndex = parentIndex;
}
}
int removeMin()
{
if (pq.size() == 0)
return 0;
int minindex;
int todel = pq[0];
pq[0] = pq[pq.size() - 1];
pq.pop_back();
int parentindex = 0;
int childindex1 = 1;
int childindex2 = 2;
while (childindex1 < pq.size())
{
if (pq[childindex1] < pq[parentindex] || pq[childindex2] < pq[parentindex])
{
if (pq[childindex1] < pq[childindex2])
minindex = childindex1;
else
minindex = childindex2;
int temp = pq[parentindex];
pq[parentindex] = pq[minindex];
pq[minindex] = temp;
parentindex = minindex;
childindex1 = 2 * parentindex + 1;
childindex2 = 2 * parentindex + 2;
}
else
break;
}
return todel;
}
void display()
{
for (int i = 0; i < pq.size(); i++)
cout << pq[i] << " ";
cout << endl;
}
};
int main()
{
PriorityQueue pq;
int choice;
cin >> choice;
while (choice != -1)
{
switch (choice)
{
case 1: // insert
int element;
cin >> element;
pq.insert(element);
break;
case 2: // getMin
cout << pq.getMin() << endl;
break;
case 3: // removeMin
cout << pq.removeMin() << endl;
break;
case 4: // size
cout << pq.getSize() << endl;
break;
case 5: // isEmpty
if (pq.isEmpty())
cout << "true" << endl;
else
cout << "false" << endl;
case 6: // show
pq.display();
break;
default:
return 0;
}
cin >> choice;
}
getchar();
return 0;
}