1717. Maximum Score From Removing Substrings

Input: s = "cdbcbbaaabab", x = 4, y = 5
Output: 19
Explanation:
- Remove the "ba" underlined in "cdbcbbaaabab". Now, s = "cdbcbbaaab" and 5 points are added to the score.
- Remove the "ab" underlined in "cdbcbbaaab". Now, s = "cdbcbbaa" and 4 points are added to the score.
- Remove the "ba" underlined in "cdbcbbaa". Now, s = "cdbcba" and 5 points are added to the score.
- Remove the "ba" underlined in "cdbcba". Now, s = "cdbc" and 5 points are added to the score.
Total score = 5 + 4 + 5 + 5 = 19.

Input: s = "aabbaaxybbaabb", x = 5, y = 4
Output: 20

Rust Solution

struct Solution;

impl Solution {
    fn maximum_gain(s: String, x: i32, y: i32) -> i32 {
        let (a, b, x, y) = if x >= y {
            ('a', 'b', x, y)
        } else {
            ('b', 'a', y, x)
        };
        let mut res = 0;
        let mut stack1 = vec![];
        for c in s.chars() {
            if c == b {
                if let Some(&d) = stack1.last() {
                    if d == a {
                        res += x;
                        stack1.pop().unwrap();
                        continue;
                    }
                }
            }
            stack1.push(c);
        }
        let mut stack2 = vec![];
        for c in stack1 {
            if c == a {
                if let Some(&d) = stack2.last() {
                    if d == b {
                        res += y;
                        stack2.pop().unwrap();
                        continue;
                    }
                }
            }
            stack2.push(c);
        }
        res
    }
}

#[test]
fn test() {
    let s = "cdbcbbaaabab".to_string();
    let x = 4;
    let y = 5;
    let res = 19;
    assert_eq!(Solution::maximum_gain(s, x, y), res);
    let s = "aabbaaxybbaabb".to_string();
    let x = 5;
    let y = 4;
    let res = 20;
    assert_eq!(Solution::maximum_gain(s, x, y), res);
}