LeetCode 1324. 竖直打印单词

最新推荐文章于 2021-12-23 18:00:00 发布

qq789045

最新推荐文章于 2021-12-23 18:00:00 发布

阅读量1.1k

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/qq789045/article/details/105366003

版权

LeetCode 专栏收录该内容

99 篇文章 0 订阅

订阅专栏

https://leetcode-cn.com/problems/print-words-vertically/

这道题遍历字符串，需要记录当前是第几个单词，这个单词的第几个字符，然后把当前的字符添加到对应的答案行中，第一个字符添加到第一行中，第二个字符添加到第二行中，如果当前的答案行长度和当前的第几个单词对应不上，需要添加空格。

class Solution {
public:
    vector<string> printVertically(string s) {
        vector<string> ans;
        
        for(int i=0,row=0,count=0;i<s.size();i++){
            if(s[i]==' '){
                row=0;
                count++;
            }
            else{
                if(row+1>ans.size())
                    ans.push_back("");
                while(ans[row].size()<count)
                    ans[row]+=' ';

                ans[row]+=s[i];
                row++;
            }

        }
        return ans;
    }
};