-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathImplement two stacks in an array.cpp
108 lines (92 loc) · 1.88 KB
/
Implement two stacks in an array.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
//{ Driver Code Starts
#include <bits/stdc++.h>
using namespace std;
// } Driver Code Ends
/*
Array Implementation of two stacks:
t1 t2
1 2 3 6 5 4
- - - - - - - - - -
-1 0 1 2 3 4 5 6 7 8 9 size = 10
*/
class twoStacks
{
int *arr;
int size;
int top1, top2;
public:
twoStacks(int n = 100)
{
size = n;
arr = new int[n];
top1 = -1;
top2 = size;
}
// Function to push an integer into the stack1.
void push1(int x)
{
top1++;
arr[top1] = x;
}
// Function to push an integer into the stack2.
void push2(int x)
{
top2--;
arr[top2] = x;
}
// Function to remove an element from top of the stack1.
int pop1()
{
if (top1 == -1)
{
return -1;
}
return arr[top1--];
}
// Function to remove an element from top of the stack2.
int pop2()
{
if (top2 == size)
{
return -1;
}
return arr[top2++];
}
};
//{ Driver Code Starts.
int main()
{
int T;
cin >> T;
while (T--)
{
twoStacks *sq = new twoStacks();
int Q;
cin >> Q;
while (Q--)
{
int stack_no;
cin >> stack_no;
int QueryType = 0;
cin >> QueryType;
if (QueryType == 1)
{
int a;
cin >> a;
if (stack_no == 1)
sq->push1(a);
else if (stack_no == 2)
sq->push2(a);
}
else if (QueryType == 2)
{
if (stack_no == 1)
cout << sq->pop1() << " ";
else if (stack_no == 2)
cout << sq->pop2() << " ";
}
}
cout << endl;
}
}
// } Driver Code Ends