leetcode#68 文本左右对齐

leetcode#68 文本左右对齐

题目：

给定一个单词数组和一个长度 maxWidth，重新排版单词，使其成为每行恰好有 maxWidth 个字符，且左右两端对齐的文本。

你应该使用“贪心算法”来放置给定的单词；也就是说，尽可能多地往每行中放置单词。必要时可用空格 ’ ’ 填充，使得每行恰好有 maxWidth 个字符。

要求尽可能均匀分配单词间的空格数量。如果某一行单词间的空格不能均匀分配，则左侧放置的空格数要多于右侧的空格数。

文本的最后一行应为左对齐，且单词之间不插入额外的空格。

说明:

单词是指由非空格字符组成的字符序列。
每个单词的长度大于 0，小于等于 maxWidth。
输入单词数组 words 至少包含一个单词。

示例：

输入:
words = ["This", "is", "an", "example", "of", "text", "justification."]
maxWidth = 16
输出:
[
   "This    is    an",
   "example  of text",
   "justification.  "
]

思路：

1、模拟

题意要求每个单词后面要有一个空格，最后一个单词没有，那就用maxWidth-这些需要的位置数，剩下的就是要填的空格数，因为最后一个单词不需要空格，所以wordscnt–，然后用空格数除以wordscnt得出的就是每个单词后面的空格，余数mod，在前mod个单词后面添加1个空格，总的空格数就是1+div+(i-bg+1<=mod)，有两种特殊情况，1、只有一个单词，那就直接把单词后面全填写空格就可以了。2、最后一句左对齐，每个单词后面只有一个空格，最后填满空格就可以了。

代码:

class Solution
{
public:
    string solve(vector<string> &words, int bg, int ed, int maxWidth, bool last)
    {
        int have = 0, div = 0, mod = 0, cnt = ed - bg;
        string res;
        for (int i = bg; i <= ed; ++i)
            have += words[i].length() + 1;
        have -= 1;
        div = cnt ? (maxWidth - have) / cnt : 0;
        mod = cnt ? (maxWidth - have) % cnt : 0;
        for (int i = bg; i < ed; ++i)
        {
            res += words[i];
            if (last)
            {
                res += " ";
                continue;
            }
            for (int j = 0; j < div + 1 + ((i - bg + 1) <= mod); ++j)
                res += " ";
        }
        res += words[ed];
        int num=maxWidth - res.size();
        for (int i = 0; i < num; ++i)
            res += " ";
        return res;
    }
    vector<string> fullJustify(vector<string> &words, int maxWidth)
    {
        vector<string> ans;
        int size = words.size();
        int len = 0, bg = 0;
        for (int i = 0; i < size; ++i)
        {
            len += words[i].length() + 1;
            if (i == size - 1 || len + words[i + 1].length() > maxWidth)
            {
                ans.push_back(solve(words, bg, i, maxWidth, i == size - 1));
                len = 0;
                bg = i + 1;
            }
        }
        return ans;
    }
};