leetcode 1324. 竖直打印单词（C++）

给你一个字符串 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"]

 

提示：

• 1 <= s.length <= 200
• s 仅含大写英文字母。
• 题目数据保证两个单词之间只有一个空格。

C++

class Solution {
public:
    vector<string> printVertically(string s) 
    {
        vector<string> res;
        int n=s.size();
        vector<string> tmp;
        int start=0;
        int len=0;
        for(int i=0;i<n;i++)
        {
            if(' '==s[i])
            {
                len=max(len,i-start);
                tmp.push_back(s.substr(start,i-start));
                start=i+1;
            }
        }
        tmp.push_back(s.substr(start));
        len=max(len,n-start);
        for(int k=0;k<len;k++)
        {
            string str;
            for(int i=0;i<tmp.size();i++)
            {
                if(k<tmp[i].size())
                {
                    str+=tmp[i][k];
                }
                else
                {
                    str+=' ';
                }
            }
            int num=str.size();
            for(int i=str.size()-1;i>=0;i--)
            {
                if(str[i]!=' ')
                {
                    num=i+1;
                    break;
                }
            }
            res.push_back(str.substr(0,num));
        }
        return res;
    }
};