文章目录
- 题目描述
- 解题方法
- 贪心
- java代码
- 复杂度分析
题目描述
给定一个单词数组 words
和一个长度 maxWidth
,重新排版单词,使其成为每行恰好有 maxWidth
个字符,且左右两端对齐的文本。
你应该使用 “贪心算法” 来放置给定的单词;也就是说,尽可能多地往每行中放置单词。必要时可用空格 ' '
填充,使得每行恰好有 maxWidth
个字符。
要求尽可能均匀分配单词间的空格数量。如果某一行单词间的空格不能均匀分配,则左侧放置的空格数要多于右侧的空格数。
文本的最后一行应为左对齐,且单词之间不插入额外的空格。
注意:
- 单词是指由非空格字符组成的字符序列。
- 每个单词的长度大于
0
,小于等于maxWidth
。 - 输入单词数组
words
至少包含一个单词。
示例 1:
输入: words = ["This", "is", "an", "example", "of", "text", "justification."], maxWidth = 16
输出:
["This is an","example of text","justification. "
]
示例 2:
输入:words = ["What","must","be","acknowledgment","shall","be"], maxWidth = 16
输出:
["What must be","acknowledgment ","shall be "
]
解释: 注意最后一行的格式应为 "shall be " 而不是 "shall be",因为最后一行应为左对齐,而不是左右两端对齐。 第二行同样为左对齐,这是因为这行只包含一个单词。
示例 3:
输入:words = ["Science","is","what","we","understand","well","enough","to","explain","to","a","computer.","Art","is","everything","else","we","do"],maxWidth = 20
输出:
["Science is what we","understand well","enough to explain to","a computer. Art is","everything else we","do "
]
提示:
1 <= words.length <= 300
1 <= words[i].length <= 20
words[i]
由小写英文字母和符号组成1 <= maxWidth <= 100
words[i].length <= maxWidth
解题方法
贪心
思路就是按照上面的规则来个每一行放置单词。
我们可以总结出以下几种情况。
- 除最后一行外,每一行的单词有两种情况。
- 当该行单词数 > 1时,需要给单词之间加空格,最后一个单词刚好放到末尾,后面无空格。
- 当该行单词数 = 1时,只需要给该单词后面加空格。
- 如果是最后一行,除最后一个单词外,需要给每个单词后面加一个空格。最后一个单词后面补充剩余的空格。
总结出以上情况后,那我们就可以构建思路了。
我们可以记录几个变量curWidth
代表当前行的单词长度,curNums
代表当前行的单词个数,words[i]
代表下一个要添加的单词。当curWidth + words[i].length() + curNums <= maxWidth
时,当前行就可以加入新单词;否则,将当前行组成的字符串加入结果集。curWidth
和curNums
重置为0
。
接下来就是怎么添加单词后面的空格了。我们可以计算当前行可以添加的空格数量,然后根据上面总结的情况将空格合理分配到单词后面。这块可以看代码实现,我已将部分重要的注释加入到代码中。
java代码
public List<String> fullJustify(String[] words, int maxWidth) {List<String> list = new ArrayList<>();int i = 0;// 记录当前行的单词长度int curWidth = 0;// 记录当前行的单词个数int curNums = 0;while (i < words.length) {// 当前行单词长度 + 下一个单词长度 + 当前行的单词数量(最少空格数量) < 最大宽度,说明可以把下一个单词加入当前行if (curWidth + words[i].length() + curNums <= maxWidth) {curWidth += words[i].length();curNums++;i++;} else {// 空格个数int spaces = maxWidth - curWidth;int a = 0;int b = 0;// 当前行单词数 > 1 时if (curNums - 1 != 0) {// 除当前行的最后一个单词外,每个单词后面至少加a个空格,前b个单词后面还需要再加1个空格a = spaces / (curNums - 1);b = spaces % (curNums - 1);}StringBuilder lineBuilder = new StringBuilder();for (int j = 0; j < curNums - 1; j++) {// 加单词lineBuilder.append(words[i - curNums + j]);// 加空格for (int k = 0; k < a; k++) {lineBuilder.append(' ');}if (b-- > 0) {lineBuilder.append(' ');}}// 加当前行的最后一个单词lineBuilder.append(words[i - 1]);// 如果该行只有一个单词,需要在单词后面的剩余空间加上空格if (curNums == 1) {while (spaces-- > 0) {lineBuilder.append(' ');}}// 当前行加入结果集list.add(lineBuilder.toString());// 当前行加入结果集后,当前行宽度和当前行单词个数重置为0curWidth = 0;curNums = 0;}}StringBuilder lineBuilder = new StringBuilder();// 最后一行结果集先加单词,再加1个空格for (int j = 0; j < curNums - 1; j++) {lineBuilder.append(words[i - curNums + j]);lineBuilder.append(' ');}// 剩余空格个数int spaces = maxWidth - curWidth - (curNums - 1);lineBuilder.append(words[i - 1]);// 最后一个单词后面把剩余空格都加上while (spaces-- > 0) {lineBuilder.append(' ');}// 最后一行加入结果集list.add(lineBuilder.toString());return list;
}
复杂度分析
时间复杂度: O ( m ) O(m) O(m), m m m是单词长度总和 + 空格总和。
空间复杂度: O ( m ) O(m) O(m),结果需要提供 O ( m ) O(m) O(m)的存储空间。
- 个人公众号
- 个人小游戏