update 1717

fishercoder1534 · fishercoder1534 · commit ec4af0cfb47c · 2024-07-12T09:10:12.000-07:00
diff --git a/src/main/java/com/fishercoder/solutions/secondthousand/_1717.java b/src/main/java/com/fishercoder/solutions/secondthousand/_1717.java
@@ -1,35 +1,36 @@
 package com.fishercoder.solutions.secondthousand;
 
-import java.util.Stack;
+import java.util.Deque;
+import java.util.LinkedList;
 
 public class _1717 {
     public static class Solution1 {
         public int maximumGain(String s, int x, int y) {
-            Stack<Character> stack1 = new Stack<>();
-            int big = x > y ? x : y;
+            int big = Math.max(x, y);
             int small = big == x ? y : x;
-            char first = x == big ? 'a' : 'b';
+            char first = big == x ? 'a' : 'b';
             char second = first == 'a' ? 'b' : 'a';
-            int maximumGain = 0;
+            Deque<Character> stack1 = new LinkedList<>();
+            int max = 0;
             for (char c : s.toCharArray()) {
-                if (c == second && !stack1.isEmpty() && stack1.peek() == first) {
-                    stack1.pop();
-                    maximumGain += big;
+                if (c == second && !stack1.isEmpty() && stack1.peekLast() == first) {
+                    stack1.pollLast();
+                    max += big;
                 } else {
-                    stack1.push(c);
+                    stack1.addLast(c);
                 }
             }
-            Stack<Character> stack2 = new Stack<>();
+            Deque<Character> stack2 = new LinkedList<>();
             while (!stack1.isEmpty()) {
-                char c = stack1.pop();
-                if (c == second && !stack2.isEmpty() && stack2.peek() == first) {
-                    stack2.pop();
-                    maximumGain += small;
+                char c = stack1.pollLast();
+                if (!stack2.isEmpty() && c == second && stack2.peekLast() == first) {
+                    max += small;
+                    stack2.pollLast();
                 } else {
-                    stack2.push(c);
+                    stack2.addLast(c);
                 }
             }
-            return maximumGain;
+            return max;
         }
     }
 
diff --git a/src/test/java/com/fishercoder/secondthousand/_1717Test.java b/src/test/java/com/fishercoder/secondthousand/_1717Test.java
@@ -1,16 +1,16 @@
 package com.fishercoder.secondthousand;
 
 import com.fishercoder.solutions.secondthousand._1717;
-import org.junit.BeforeClass;
-import org.junit.Test;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
 
-import static org.junit.Assert.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertEquals;
 
 public class _1717Test {
     private static _1717.Solution1 solution1;
 
-    @BeforeClass
-    public static void setup() {
+    @BeforeEach
+    public void setup() {
         solution1 = new _1717.Solution1();
     }
 
