leetcode.942. 增减字符串匹配

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：

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

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

输入："DDI"
输出：[3,2,0,1]
 

提示：

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

题解：

首先遍历字符串，对于遍历到的第一个，若其是I，则我们发现直接让其为0-N中最小的0的话必定会满足题意。若是D，我们发现直接让其为0-N中最大的N的话也必定会满足题意。因此第一个即可解决；
同理对于第二个我们发现和第一个是同样的处理方法，只不过是0-N的范围变为了别的范围而已，因此我们还是可以按照同样的方法解决。

代码：

class Solution {
    public int[] diStringMatch(String s) {
        int N = s.length();
        int nmax = N;
        int nmin = 0;
        int[] res = new int[s.length()+1];
        for(int i = 0;i<N;i++){
            if(s.charAt(i)=='I'){
                res[i] = nmin;
                nmin++;
            }
            else{
                res[i] = nmax;
                nmax--;
            }
        }
        res[N] = nmax;
        return res;
    }
}