-
Notifications
You must be signed in to change notification settings - Fork 0
/
Find Maximum Equal sum of Three Stacks
103 lines (78 loc) · 2.6 KB
/
Find Maximum Equal sum of Three Stacks
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
//{ Driver Code Starts
import java.io.*;
import java.util.*;
class IntArray
{
public static int[] input(BufferedReader br, int n) throws IOException
{
String[] s = br.readLine().trim().split(" ");
int[] a = new int[n];
for(int i = 0; i < n; i++)
a[i] = Integer.parseInt(s[i]);
return a;
}
public static void print(int[] a)
{
for(int e : a)
System.out.print(e + " ");
System.out.println();
}
public static void print(ArrayList<Integer> a)
{
for(int e : a)
System.out.print(e + " ");
System.out.println();
}
}
class GFG {
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
int t;
t = Integer.parseInt(br.readLine());
while(t-- > 0){
int[] a = IntArray.input(br, 3);
int[] S1 = IntArray.input(br, a[0]);
int[] S2 = IntArray.input(br, a[1]);
int[] S3 = IntArray.input(br, a[2]);
Solution obj = new Solution();
int res = obj.maxEqualSum(a[0],a[1],a[2], S1, S2, S3);
System.out.println(res);
}
}
}
// } Driver Code Ends
class Solution {
public static int maxEqualSum(int N1,int N2,int N3, int[] S1, int[] S2, int[] S3) {
// code here
int sum1 = calculateSum(S1);
int sum2 = calculateSum(S2);
int sum3 = calculateSum(S3);
int top1 = 0, top2 = 0, top3 = 0;
while (true) {
// If any stack is empty, return 0
if (top1 == N1 || top2 == N2 || top3 == N3) {
return 0;
}
// If the sums are equal, return the sum
if (sum1 == sum2 && sum2 == sum3) {
return sum1;
}
// Find the stack with the largest sum and remove the top element from it
if (sum1 >= sum2 && sum1 >= sum3) {
sum1 -= S1[top1++];
} else if (sum2 >= sum1 && sum2 >= sum3) {
sum2 -= S2[top2++];
} else if (sum3 >= sum1 && sum3 >= sum2) {
sum3 -= S3[top3++];
}
}
}
// Helper method to calculate the sum of an array
private static int calculateSum(int[] arr) {
int sum = 0;
for (int i = 0; i < arr.length; i++) {
sum += arr[i];
}
return sum;
}
}