leetcode 增减字符串匹配 C

给定只含 "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. 1 <= S.length <= 1000
2. S 只包含字符 "I" 或 "D"。

   /**
    * Return an array of size *returnSize.
    * Note: The returned array must be malloced, assume caller calls free().
    */
   int* diStringMatch(char* S, int* returnSize) {
   
       int N=strlen(S);
       int* res=(int*)malloc(sizeof(int)*(N+1));
       int a=N;
       int b=0;
       for(int i=0;i<N;i++)
       {
           if('I'==S[i])
           {
               res[i]=b;
               b++;
           }
           else
           {
               res[i]=a;
               a--;
           }
       }
       res[N]=b;
       *returnSize=N+1;
       return res;
   
   }