1324. 竖直打印单词-力扣双百代码

1324. 竖直打印单词

给你一个字符串 s。请你按照单词在 s 中的出现顺序将它们全部竖直返回。
单词应该以字符串列表的形式返回，必要时用空格补位，但输出尾部的空格需要删除（不允许尾随空格）。
每个单词只能放在一列上，每一列中也只能有一个单词。

示例 1：

输入：s = “HOW ARE YOU”
输出：[“HAY”,“ORO”,“WEU”]
解释：每个单词都应该竖直打印。
“HAY”
“ORO”
“WEU”

示例 2：

输入：s = “TO BE OR NOT TO BE”
输出：[“TBONTB”,“OEROOE”," T"]
解释：题目允许使用空格补位，但不允许输出末尾出现空格。
“TBONTB”
“OEROOE”
" T"

示例 3：

输入：s = “CONTEST IS COMING”
输出：[“CIC”,“OSO”,“N M”,“T I”,“E N”,“S G”,“T”]

这题其实看着难，但是中规中矩去做就可以了，算法思想很简单，不过我们需要先估算一下空间复杂度，这样可以极大降低时间效率。
解题代码如下：

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
char ** printVertically(char * s, int* returnSize){
    int n=0;
    int i;
    i=0;
    int max=0;
    int num=0;
    while(s[i]!='\0'){
      
        if(s[i]==' '){
            n++;
            i++;
            num=0;
            continue;
            
        }
          num++;
         if(num>max){
            max=num;
        }
        i++;
    }
    char **ch=(char **)malloc(sizeof(char *)*max);
    for(i=0;i<max;i++){
        ch[i]=(char *)malloc(sizeof(char)*(n+2));
    }
    int j;
     for(i=0;i<max;i++){
         for(j=0;j<n+2;j++){
                ch[i][j]=' ';
         }
    }
    n=0;
    i=0;
    num=0;
     while(s[i]!='\0'){
        if(s[i]==' '){
            n++;
            num=0;
            i++;
            continue;  
        }
       ch[num][n]=s[i];
        num++;
        i++;
    }
    *returnSize=max;
    for(i=0;i<max;i++){
        j=n+1;
      
      while(ch[i][j]==' '){
            j--;
        }
        ch[i][j+1]='\0';
       
       
    }
    return ch;


}