단언문을 안전 호출 연산자로 수정

DaleStudy · Aug 29, 2024 · 42cf39c · 42cf39c
1 parent c31a82f
commit 42cf39c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/two-sum/jdalma.kt b/two-sum/jdalma.kt
@@ -19,7 +19,9 @@ class `two-sum` {
         nums.forEachIndexed { i, e ->
             val diff: Int = target - e
             if (map.containsKey(diff) && map[diff] != i) {
-                return intArrayOf(i , map[diff]!!)
+                return map[diff]?.let {
+                    intArrayOf(it, i)
+                } ?: intArrayOf()
             }
         }
         return intArrayOf()
@@ -35,7 +37,9 @@ class `two-sum` {
         for (index in nums.indices) {
             val diff = target - nums[index]
             if (map.containsKey(diff)) {
-                return intArrayOf(map[diff]!!, index)
+                return map[diff]?.let {
+                    intArrayOf(it, index)
+                } ?: intArrayOf()
             }
             map[nums[index]] = index
         }