68 accepted

Author: aQua

Algorithms/0068.text-justification/text-justification.go

@@ -1,43 +1,54 @@
 package Problem0068
 
+import "strings"
+
 func fullJustify(words []string, maxWidth int) []string {
 	res := []string{}
 	temp := []string{}
 	width := 0
+	isLast := false
 
 	for len(words) > 0 {
-		words, temp, width = split(words, maxWidth)
-		res = append(res, combine(temp, width, maxWidth))
+		words, temp, width, isLast = split(words, maxWidth)
+		res = append(res, combine(temp, width, maxWidth, isLast))
 	}
 
 	return res
 }
 
-// 返回待组合的单词，需要组合的单词，和这些单词的长度之和
-func split(words []string, maxWidth int) ([]string, []string, int) {
+// 返回待组合的单词，需要组合的单词，和这些单词的长度之和, 是否是结尾
+func split(words []string, maxWidth int) ([]string, []string, int, bool) {
 	res := make([]string, 1)
 	res[0] = words[0]
 	width := len(words[0])
+
 	i := 1
 	for ; i < len(words); i++ {
-		if width+len(words[i]) > maxWidth {
+		if width+len(res)+len(words[i]) > maxWidth {
 			break
 		}
 		res = append(res, words[i])
 		width += len(words[i])
 	}
-	return words[i:], res, width
+
+	return words[i:], res, width, i == len(words)
 }
 
-func combine(words []string, width, maxWidth int) string {
+func combine(words []string, width, maxWidth int, isLast bool) string {
 	wordCount := len(words)
-	spaces := makeSpaces(wordCount-1, maxWidth-width)
+	if wordCount == 1 || isLast {
+		return combineSpecial(words, maxWidth)
+	}
+	spaceCount := wordCount - 1
+	spaces := makeSpaces(spaceCount, maxWidth-width)
 
 	res := ""
 	for i, v := range spaces {
 		res += words[i] + v
 	}
-	res += words[wordCount-1]
+	if wordCount > 1 {
+		res += words[wordCount-1]
+	}
 
 	return res
 }
@@ -50,3 +61,13 @@ func makeSpaces(Len, count int) []string {
 
 	return res
 }
+
+func combineSpecial(words []string, maxWidth int) string {
+	res := strings.Join(words, " ")
+
+	for len(res) < maxWidth {
+		res += " "
+	}
+
+	return res
+}

Algorithms/0068.text-justification/text-justification_test.go

@@ -38,6 +38,15 @@ func Test_Problem0068(t *testing.T) {
 			},
 		},
 
+		question{
+			para{
+				[]string{"What", "must", "be", "shall", "be."},
+				12,
+			},
+			ans{
+				[]string{"What must be", "shall be.   "},
+			},
+		},
 		// 如需多个测试，可以复制上方元素。
 	}