LeetCode——942. 增减字符串匹配

题目描述：

给定只含 “I”（增大）或 “D”（减小）的字符串 S ，令 N = S.length。返回 [0, 1, …, N] 的任意排列 A 使得对于所有 i = 0, …, N-1，都有：

• 如果 S[i] == “I”，那么 A[i] < A[i+1]
• 如果 S[i] == “D”，那么 A[i] > A[i+1]

提示：

• 1 <= S.length <= 10000
• S 只包含字符 “I” 或 “D”。

示例 1：
输入：“IDID”
输出：[0,4,1,3,2]

示例 2：
输入：“III”
输出：[0,1,2,3]

示例 3：
输入：“DDI”
输出：[3,2,0,1]

代码如下：

class Solution {
    public int[] diStringMatch(String S) {
        int n = S.length();
        char[] ch = S.toCharArray();
        int[] arr = new int[n + 1];
        int a = 0, b = n;
        for (int i = 0; i < n; i++) {
            if (ch[i] == 'I') {
                arr[i] = a;
                a++;
            }
            if (ch[i] == 'D') {
                arr[i] = b;
                b--;
            }
        }
        int ans = 0;
        for (int i = 0; i < n; i++) {
            ans ^= arr[i];
            ans ^= i;
        }
        ans ^= n;
        arr[n] = ans;
        return arr;
    }
}

执行结果：