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.  "

]

 

import java.util.*;

public class Solution {
    public ArrayList<String> fullJustify(String[] words, int maxWidth) {
        ArrayList<String> res = new ArrayList<String>();
        int n = words.length;
        int i = 0;
        while (i < n) {
            StringBuilder sb = new StringBuilder();
            int last = i + 1;
            int len = words[i].length();
            while (last < n && len + 1 + words[last].length() <= maxWidth) {
                len += 1 + words[last].length();
                last++;
            }
            // 最后一行
            if (last == n) {
                for (int j = i; j < n; j++) {
                    sb.append(words[j] + " ");
                }
                sb.deleteCharAt(sb.length() - 1);
                for (int j = sb.length(); j < maxWidth; j++) {
                    sb.append(" ");
                }
            } else {
                // 只有一个word
                if (last - i == 1) {
                    sb.append(words[i]);
                    for (int j = words[i].length(); j < maxWidth; j++)
                        sb.append(" ");
                } else {// 有多个单词
                    int wordNum = last - i;
                    int wordTotal = 0;
                    for (int j = i; j < last; j++) {
                        wordTotal += words[j].length();
                    }
                    // eachSpace为每个单词间的空格数；r是余数，表示前r个空格数为eachSpace+1
                    int eachSpace = (maxWidth - wordTotal) / (wordNum - 1);
                    int r = (maxWidth - wordTotal) % (wordNum - 1);

                    for (int j = i; j < last; j++) {
                        sb.append(words[j]);
                        if (j < last - 1) {
                            for (int k = 0; k < eachSpace + ((j - i) < r ? 1 : 0); k++) {
                                sb.append(" ");
                            }
                        }
                    }
                }
            }
            res.add(sb.toString());
            i = last;
        }
        return res;
    }
}