1717. Maximum Score From Removing Substrings
JAVA
class Solution { int points = 0; public int maximumGain(String s, int x, int y) { if (x >= y) { s = remove1(s, x); s = remove2(s, y); } else { s = remove2(s, y); s = remove1(s, x); } return points; } public String remove1(String s, int x) { StringBuffer sb = new StringBuffer(); int length = s.length(); int index = 0; for (int i = 0; i < length; i++) { char c = s.charAt(i); if (index > 0 && c == 'b' && sb.charAt(index - 1) == 'a') { points += x; sb.deleteCharAt(index - 1); index--; } else { sb.append(c); index++; } } return sb.toString(); } public String remove2(String s, int y) { StringBuffer sb = new StringBuffer(); int length = s.length(); int index = 0; for (int i = 0; i < length; i++) { char c = s.charAt(i); if (index > 0 && c == 'a' && sb.charAt(index - 1) == 'b') { points += y; sb.deleteCharAt(index - 1); index--; } else { sb.append(c); index++; } } return sb.toString(); } } ############ class Solution { public int maximumGain(String s, int x, int y) { if (x < y) { return maximumGain(new StringBuilder(s).reverse().toString(), y, x); } int ans = 0; Deque<Character> stk1 = new ArrayDeque<>(); Deque<Character> stk2 = new ArrayDeque<>(); for (char c : s.toCharArray()) { if (c != 'b') { stk1.push(c); } else { if (!stk1.isEmpty() && stk1.peek() == 'a') { stk1.pop(); ans += x; } else { stk1.push(c); } } } while (!stk1.isEmpty()) { char c = stk1.pop(); if (c != 'b') { stk2.push(c); } else { if (!stk2.isEmpty() && stk2.peek() == 'a') { stk2.pop(); ans += y; } else { stk2.push(c); } } } return ans; } }
C++
class Solution { public: int maximumGain(string s, int x, int y) { if (x < y) { reverse(s.begin(), s.end()); return maximumGain(s, y, x); } int ans = 0; stack<char> stk1; stack<char> stk2; for (char c : s) { if (c != 'b') stk1.push(c); else { if (!stk1.empty() && stk1.top() == 'a') { stk1.pop(); ans += x; } else stk1.push(c); } } while (!stk1.empty()) { char c = stk1.top(); stk1.pop(); if (c != 'b') stk2.push(c); else { if (!stk2.empty() && stk2.top() == 'a') { stk2.pop(); ans += y; } else stk2.push(c); } } return ans; } };
Comments
Post a Comment