큰 수 만들기

Author: co3oing

탐욕법(Greedy)/큰 수 만들기.swift

@@ -0,0 +1,26 @@
+import Foundation
+
+func solution(_ number:String, _ k:Int) -> String {
+    let numbers = Array(number).map { Int(String($0))! }
+    var removeCount = 0
+    var answer: [Int] = []
+    
+    for num in numbers {
+        if answer == [] || answer.last! >= num {
+            answer.append(num)
+        } else {
+            while answer != [] && removeCount < k && answer.last! < num {
+                answer.popLast()!
+                removeCount += 1
+            }
+            answer.append(num)
+        }
+    }
+    
+    while removeCount < k {
+        answer.popLast()!
+        removeCount += 1
+    }
+    
+    return answer.map{ String($0) }.joined()
+}