Leetcode68-文本左右对齐

wobushihaorenF

于 2022-12-06 13:36:18 发布

阅读量142

点赞数

分类专栏： Leetcode 文章标签： leetcode 算法 c++

本文链接：https://blog.csdn.net/aaa7888210/article/details/128201491

版权

Leetcode 专栏收录该内容

130 篇文章 1 订阅

订阅专栏

超级大模拟

先将能组成一句且宽度不超过maxWidth的单词分组，然后分别处理每一组单词。剩下的都是细节了。

class Solution {
public:
    vector<string> fullJustify(vector<string>& words, int maxWidth) {
        vector<vector<string>> s;
        vector<string> t,ans;
        vector<int> h;
        int len=0;
        for(auto c:words){
            if(t.size()==0||len+t.size()-1+c.size()<maxWidth){
                t.push_back(c);
                len+=c.size();
            }else{
                s.push_back(t);
                h.push_back(len);
                t.resize(0);
                len=c.size();
                t.push_back(c);
            }
        }
        s.push_back(t);
        h.push_back(len);
        for(int i=0;i<s.size();i++){
            string res;
            if(i==s.size()-1){
                for(auto c:s[i]){
                    if(res.size()) res+=' ';
                    res+=c;
                }
                while(res.size()!=maxWidth) res+=' ';
                ans.push_back(res);
            }else{
                if(s[i].size()==1){
                    res+=s[i][0];
                    while(res.size()!=maxWidth) res+=' ';
                    ans.push_back(res);
                }else{
                    int w=maxWidth-h[i];
                    int x=w/(s[i].size()-1),y=w%(s[i].size()-1);
                    for(auto c:s[i]){
                        if(res.size()){
                            for(int j=0;j<x;j++) res+=' ';
                        }
                        if(res.size()&&y){
                            res+=' ';
                            y--;
                        }
                        res+=c;
                    }
                    ans.push_back(res);
                }
            }
        }
        return ans;
    }
};

时间复杂度：O（n），n为words中所有单词的长度之和

空间复杂度：O（n）