1. 程式人生 > >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 import java.util.*;
 2 
 3 public class Solution {
 4     public ArrayList<String> fullJustify(String[] words, int maxWidth) {
 5         ArrayList<String> res = new ArrayList<String>();
 6         int n = words.length;
 7         int i = 0;
 8         while (i < n) {
9 StringBuilder sb = new StringBuilder(); 10 int last = i + 1; 11 int len = words[i].length(); 12 while (last < n && len + 1 + words[last].length() <= maxWidth) { 13 len += 1 + words[last].length(); 14 last++; 15 } 16 // 最後一行 17 if (last == n) { 18 for (int j = i; j < n; j++) { 19 sb.append(words[j] + " "); 20 } 21 sb.deleteCharAt(sb.length() - 1); 22 for (int j = sb.length(); j < maxWidth; j++) { 23 sb.append(" "); 24 } 25 } else { 26 // 只有一個word 27 if (last - i == 1) { 28 sb.append(words[i]); 29 for (int j = words[i].length(); j < maxWidth; j++) 30 sb.append(" "); 31 } else {// 有多個單詞 32 int wordNum = last - i; 33 int wordTotal = 0; 34 for (int j = i; j < last; j++) { 35 wordTotal += words[j].length(); 36 } 37 // eachSpace為每個單詞間的空格數;r是餘數,表示前r個空格數為eachSpace+1 38 int eachSpace = (maxWidth - wordTotal) / (wordNum - 1); 39 int r = (maxWidth - wordTotal) % (wordNum - 1); 40 41 for (int j = i; j < last; j++) { 42 sb.append(words[j]); 43 if (j < last - 1) { 44 for (int k = 0; k < eachSpace + ((j - i) < r ? 1 : 0); k++) { 45 sb.append(" "); 46 } 47 } 48 } 49 } 50 } 51 res.add(sb.toString()); 52 i = last; 53 } 54 return res; 55 } 56 }