942. 增减字符串匹配,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：

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

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

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

提示：

1 <= S.length <= 1000
S 只包含字符 “I” 或 “D”。
思路：
D时肯定比后面的大，故从dN=S.length开始。每次遇到一个D就dN–。I时肯定比后面的小，故从iN=0开始。每次遇到一个I就iN++。最后多余的一个由S[S.length]决定。

class Solution {
public:
    vector<int> diStringMatch(string S) {
        int n=S.length();
        vector<int> a;
        int iN=0;
        int dN=n;
        for(int i=0;i<n+1;i++){
            if(i==n && S[i-1]=='I'){
                a.push_back(a[i-1]+1);
            }else if(i==n && S[i-1]=='D'){
                a.push_back(a[i-1]-1);
            }else if(S[i]=='I'){                    
                a.push_back(iN++);               
            }else{
                a.push_back(dN--);
            }
        }
        return a;
    }
};