【LeetCode】68. Text Justification 文本排版实现两端对齐

一、概述

输入一个字符串数组和一个数字m，其中元素是一个个单词，对单词进行重新组合，将一定数量的单词按原来的顺序组合在一起，在其中插入若干空格，以保证：

不是最后一行的，保证字母数量加空格数量为m，单词间的空格数量尽可能相同，不相同的话，保证左边的较多，保证本行最后一个字符不是空格；

是最后一行的，保证，单词间的空格数量都为1，字母数量加空格小于m时，剩余空格全加到行尾。

有点复杂，总而言之就是对一句话进行排版，分为两端对齐的多行。

二、分析

首先要分成两种情况：

到最后一行和不到最后一行。

不到最后一行是普遍情况，先讨论这个。主要思路就是维护两个指针i和left，一个变量len。left指向当前行的第一个单词，i指向当前正在处理的单词。

对于正在处理的单词，有三种情况：

第一种，i指向的单词长度加上len仍然小于m，那么就在这个单词后插入一个空格，len加一加上单词长度，再判断下一个单词——为什么要再加一呢？我们知道，在一行中，单词之间至少插入一个空格，这里的加一就是把这个空格算上了。

第二种，i指向的单词长度加上len大于m，那么就到上一个单词为止，开始准备整理成行。这里有一个问题：最后一个单词后面不能有空格，别急，一会判断这个。

第三种，i指向的单词长度加上len等于m，那么同样到上一个单词为止。

现在后两种情况都是到上一个单词为止，但是第二种情况下，我们需要把上一个单词最后的空格删去，第三种情况下，可以把当前单词加上——注意后面没有空格。这样就确定了当前行中容纳的所有单词。然后就要插入空格了，插多少呢？个数是m-len。从第一个单词往后加到倒数第二个单词，要是还有空格，那就从头再继续。

当我们到了最后一行，需要做的就是直接在行末补空格直到长度为m即可。

代码如下：

class Solution {
public:
    vector<string> fullJustify(vector<string>& words, int maxWidth) {
        int len=0,left=0,right=0,i=0;
        vector<string> res;
        while(i<=words.size())
        {
            if(i<words.size()&&len+words[i].size()<maxWidth)
            {
                len+=(words[i].size()+1);
                words[i]+=' ';
                ++i;
                continue;
            }
            if(i==words.size()||len+words[i].size()>=maxWidth)
            {
                if(i<words.size()&&len+words[i].size()==maxWidth)
                {
                    len+=words[i].size();
                    ++i;
                }
                else
                {
                    words[i-1].erase(words[i-1].size()-1);
                    --len;
                }
                int exs=maxWidth-len;
                right=i-1;
                string tmp;
                int j=0;
                if(i!=words.size())
                {
                    for(j=left;exs>0;++j)
                    {
                        words[j]+=' ';
                        --exs;
                        if(j>=right-1||j>words.size()-1)
                            j=left-1;
                    }
                }
                else
                {
                    while(exs>0)
                    {
                        words[words.size()-1]+=' ';
                        --exs;
                    }
                }
                for(int j=left;j<=right;++j)
                    tmp+=words[j];
                res.push_back(tmp);
                if(i==words.size())
                    break;
                len=0;
                left=i;
                len=0;
            }
        }
        return res;
    }
};

总而言之抓住两点：

其一是单词之间必须至少有一个空格；其二是维护len来继续插入空格。

三、总结

题目本身不难，主要是边界条件。还有后面加空格的时机：是在挑选的时候加，还是在确定之后一起加？都可以。