forked from adityabisoi/ds-algo-solutions
-
Notifications
You must be signed in to change notification settings - Fork 0
/
solution.java
97 lines (76 loc) · 3.28 KB
/
solution.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
import java.io.BufferedWriter;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Scanner;
import java.util.Stack;
public class EqualStacks {
static int equalStacks(int[] ha1, int[] ha2, int[] ha3) {
Stack<Integer> s1 = new Stack<Integer>(); //First stack of cylinders
Stack<Integer> s2 = new Stack<Integer>(); //Second stack of cylinders
Stack<Integer> s3 = new Stack<Integer>(); //Third stack of cylinders
int h1 = 0; //Height of the first stack
int h2 = 0; //Height of the second stack
int h3 = 0; //Height of the third stack
int minStack; //The stack with the smallest height
for(int i = ha1.length-1; i >= 0 ; i--){
s1.push(ha1[i]);
h1 += ha1[i];
}
for(int i = ha2.length-1; i >= 0 ; i--){
s2.push(ha2[i]);
h2 += ha2[i];
}
for(int i = ha3.length-1; i >= 0 ; i--){
s3.push(ha3[i]);
h3 += ha3[i];
}
minStack = Math.min(h1,Math.min(h2,h3));
//Heights are not all equal enter the while loop
while(h1 != h2 || h1 != h3) {
//Remove cylinders from stack 1 until your height is <= the smallest
while(h1 > minStack){
h1 -= s1.pop();
}
minStack = Math.min(h1,Math.min(h2,h3)); //Recalculate min
//Remove cylinders from stack 2 until your height is <= the smallest
while(h2 > minStack){
h2 -= s2.pop();
}
minStack = Math.min(h1,Math.min(h2,h3)); //Recalculate min
//Remove cylinders from stack 3 until your height is <= the smallest
while(h3 > minStack){
h3 -= s3.pop();
}
minStack = Math.min(h1,Math.min(h2,h3)); //Recalculate min
}
return minStack;
}
private static final Scanner scanner = new Scanner(System.in);
// main function
public static void main(String[] args) throws IOException {
String[] n1N2N3 = scanner.nextLine().split(" ");
int n1 = Integer.parseInt(n1N2N3[0].trim()); // number of cylinders in first stack
int n2 = Integer.parseInt(n1N2N3[1].trim()); // number of cylinders in Second stack
int n3 = Integer.parseInt(n1N2N3[2].trim()); // number of cylinders in Third stack
int[] h1 = new int[n1]; // height of cylinder in first stack
String[] h1Items = scanner.nextLine().split(" ");
for (int h1Itr = 0; h1Itr < n1; h1Itr++) {
int h1Item = Integer.parseInt(h1Items[h1Itr].trim());
h1[h1Itr] = h1Item;
}
int[] h2 = new int[n2]; // height of cylinder in Second stack
String[] h2Items = scanner.nextLine().split(" ");
for (int h2Itr = 0; h2Itr < n2; h2Itr++) {
int h2Item = Integer.parseInt(h2Items[h2Itr].trim());
h2[h2Itr] = h2Item;
}
int[] h3 = new int[n3]; // height of cylinder in Third stack
String[] h3Items = scanner.nextLine().split(" ");
for (int h3Itr = 0; h3Itr < n3; h3Itr++) {
int h3Item = Integer.parseInt(h3Items[h3Itr].trim());
h3[h3Itr] = h3Item;
}
int result = equalStacks(h1, h2, h3);
System.out.println(result);
}
}