测试

Author: masx200

split-message-based-on-limit/index.kt

@@ -1,12 +1,21 @@
-package com.github.masx200.split_message_based_on_limit
+package com.github.masx200.leetcode_test.split_message_based_on_limit
 
 class Solution {
     fun splitMessage(message: String, limit: Int): Array<String> {
         if (limit <= 5) return arrayOf()
+        val results = splitStrings(message, limit, 0, 10)
+        if (results.isNotEmpty()) return results
 
+        val results2 = splitStrings(message, limit, 10, message.length)
+        if (results2.isNotEmpty()) return results2
+
+        return arrayOf()
+    }
+
+    private fun splitStrings(message: String, limit: Int, lleft: Int, rright: Int): Array<String> {
+        var left = lleft
+        var right = rright
 
-        var left = 0
-        var right = message.length
 
         while (left < right) {
             val mid = (left + right) / 2
@@ -16,7 +25,8 @@ class Solution {
                 left = mid + 1
             }
         }
-        if (left > 0 && left < message.length + 1) {
+
+        if (left > 0 && left < rright) {
 
             var len = 0
             return Array(left) {
@@ -26,6 +36,7 @@ class Solution {
                 )
 
                 val s = message.substring(len, endIndex) + "<${it + 1}/${left}>";
+
                 if (len == endIndex) return arrayOf()
                 len = endIndex
                 s
@@ -46,4 +57,4 @@ private fun checkMessage(message: String, limit: Int, count: Int): Boolean {
         len += limit - (3 + i.toString().length + count.toString().length)
     }
     return len >= message.length
-}
+}

test/com/github/masx200/split_message_based_on_limit/SolutionTest.kt

@@ -1,4 +1,4 @@
-package com.github.masx200.split_message_based_on_limit
+package com.github.masx200.leetcode_test.split_message_based_on_limit
 
 import org.junit.jupiter.api.Test
 import kotlin.test.assertContentEquals
@@ -41,6 +41,12 @@ internal class SolutionTest {
             Solution().splitMessage(
                 "short message", 15
             )
+        );
+        assertContentEquals(
+            arrayOf("abc<1/7>", "dab<2/7>", "cda<3/7>", "bcd<4/7>", "abc<5/7>", "dab<6/7>", "cd<7/7>"),
+            Solution().splitMessage(
+                "abcdabcdabcdabcdabcd", 8
+            )
         )
     }
 }