每日一道Leetcode - 942. 增减字符串匹配

class Solution {
    
    
    public int[] diStringMatch(String S) {
    
    
        // I 偶数 从小到大分配
        // D 奇数 从大到小分配
        // 排放
        int N = S.length();
        int[] ans = new int[N+1];
        int i = 0;
        int j = N;
        int k = 0;
        while(k<=N+1){
    
    
            if(k>=N){
    
    
                ans[k] = i;
                break;
            }else{
    
    
                if(S.charAt(k)=='I'){
    
    
                    ans[k] = i;
                    i+=1;
                }else if(S.charAt(k)=='D'){
    
    
                    ans[k] = j;
                    j-=1;
                }
            }
            k++;
        }
        return ans;
    }
}