[Swift]LeetCode753. 破解保險箱 | Cracking the Safe

★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★
➤微信公衆號：山青詠芝（shanqingyongzhi）
➤博客園地址：山青詠芝（https://www.cnblogs.com/strengthen/）
➤GitHub地址：https://github.com/strengthen/LeetCode
➤原文地址： http://www.javashuo.com/article/p-boutfhwv-me.html 
➤若是連接不是山青詠芝的博客園地址，則多是爬取做者的文章。
➤原文已修改更新！強烈建議點擊原文地址閱讀！支持做者！支持原創！
★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★

There is a box protected by a password. The password is n digits, where each letter can be one of the first k digits 0, 1, ..., k-1.

You can keep inputting the password, the password will automatically be matched against the last n digits entered.

For example, assuming the password is "345", I can open it when I type "012345", but I enter a total of 6 digits.

Please return any string of minimum length that is guaranteed to open the box after the entire string is inputted.

Example 1:

Input: n = 1, k = 2
Output: "01"
Note: "10" will be accepted too. 

Example 2:

Input: n = 2, k = 2
Output: "00110"
Note: "01100", "10011", "11001" will be accepted too. 

Note:

1. n will be in the range [1, 4].
2. k will be in the range [1, 10].
3. k^n will be at most 4096.

---

有一個須要密碼才能打開的保險箱。密碼是 n 位數, 密碼的每一位是 k 位序列 0, 1, ..., k-1 中的一個 。

你能夠隨意輸入密碼，保險箱會自動記住最後 n 位輸入，若是匹配，則可以打開保險箱。

舉個例子，假設密碼是 "345"，你能夠輸入 "012345" 來打開它，只是你輸入了 6 個字符.

請返回一個能打開保險箱的最短字符串。 

示例1:

輸入: n = 1, k = 2
輸出: "01"
說明: "10"也能夠打開保險箱。 

示例2:

輸入: n = 2, k = 2
輸出: "00110"
說明: "01100", "10011", "11001" 也能打開保險箱。 

提示：

1. n 的範圍是 [1, 4]。
2. k 的範圍是 [1, 10]。
3. k^n 最大可能爲 4096。

---

12ms

 1 func deBruijnSequence<Alphabets: RandomAccessCollection>(of alphabets:Alphabets, length: Int) -> [Alphabets.Element] {
 2     typealias Alphabet = Alphabets.Element
 3     
 4     let alphabetCount = alphabets.count
 5     let cycleCount = repeatElement(alphabetCount, count: length - 1).reduce(1, *)
 6     let debruijnLength = cycleCount * alphabetCount
 7     
 8     var used = Array(repeating: false, count: debruijnLength)
 9     var result: [Alphabet] = []
10     result.reserveCapacity(debruijnLength)
11     
12     for index in 0..<debruijnLength {
13         var current = index
14         while !used[current] {
15             used[current] = true
16             
17             let elementIndex = current / cycleCount
18             current = (current % cycleCount) * alphabetCount + elementIndex
19             result.append(alphabets[alphabets.index(alphabets.startIndex, offsetBy: elementIndex)])
20         }
21         assert(current == index)
22     }
23     
24     assert(result.count == debruijnLength)
25     return result
26 }
27 
28 class Solution {
29     func crackSafe(_ n: Int, _ k: Int) -> String {
30         let result = deBruijnSequence(of: (0..<k).map(String.init), length: n)
31         return (result + repeatElement("0", count: n - 1)).joined()
32     }
33 }

---

36ms

 1 class Solution {
 2     func crackSafe(_ n: Int, _ k: Int) -> String {
 3         let total = Int(pow(Double(k), Double(n)))
 4         var current = [Int](repeating: 0, count: n)
 5         var used = Set<String>()
 6         used.insert(current.reduce("") { $0 + String($1) })
 7         dfs(n, k, total, &used, &current)
 8         return current.reduce("") { $0 + String($1) }
 9     }
10         
11     private func dfs(_ n: Int, _ k: Int, _ total: Int, _ used: inout Set<String>, _ current: inout [Int]) -> Bool {
12         guard used.count < total else {
13             return true
14         }
15         var prefix = Array(current[(current.count - n + 1)...]).reduce("") { $0 + String($1) }
16         for num in 0..<k {
17             let currentSegment = prefix + String(num)
18             guard !used.contains(currentSegment) else {
19                 continue
20             }
21             used.insert(currentSegment)
22             current.append(num)
23             if dfs(n, k, total, &used, &current) {
24                 return true
25             }
26             current.removeLast()
27             used.remove(currentSegment)
28         }
29         return false
30     }
31 }

---

152ms

 1 class Solution {
 2     func dfs(_ n: Int, _ k: Int, _ len: Int, _ data: inout Set<String>, _ res:inout String) -> Bool {
 3         if (res.count == len) { return true; }
 4         let suf = res.suffix(n-1)
 5         for c in 0..<k {
 6             if !data.contains(suf + "\(c)") {
 7                 data.insert(suf + "\(c)")
 8                 res += "\(c)"
 9                 if (dfs(n, k, len, &data, &res)) {
10                     return true
11                 }
12                 data.remove(suf + "\(c)")
13                 res.removeLast()
14             }
15         }
16         return false
17     }
18     func crackSafe(_ n: Int, _ k: Int) -> String {
19         let len = Int(pow(Double(k), Double(n))) + n - 1
20         var res = String((0..<n).map{_ in Character("0")})
21         var data = Set([res])
22         dfs(n, k, len, &data, &res)
23         return res
24     }
25 }

---

Runtime: 320 ms

Memory Usage: 22.2 MB

 1 class Solution {
 2     func crackSafe(_ n: Int, _ k: Int) -> String {
 3         var res:String = "0".repeatString(n - 1)
 4         var visited:Set<String> = [res]
 5         var num:Int = Int(pow(Double(k), Double(n)))
 6         helper(n, k,num, &visited, &res)
 7         return res
 8     }
 9     
10     func helper(_ n:Int,_ k:Int,_ total:Int,_ visited:inout Set<String>,_ res:inout String)
11     {
12         if visited.count == total
13         {
14             return
15         }
16         var pre:String = res.subString(res.count - n + 1, n - 1)
17         for i in stride(from:k - 1,through:0,by:-1)
18         {
19             var cur:String = pre + String(i)
20             if visited.contains(cur) {continue}
21             visited.insert(cur)
22             res += String(i)
23             helper(n, k, total, &visited, &res)
24         }        
25     }
26 }
27 
28 extension String {
29     //獲取重複指定次數的字符串
30     func repeatString(_ times: Int ) -> String
31     {
32         var result = String()
33         for i in 0...times {
34             result += self
35         }
36         return result 
37     }
38     
39     // 截取字符串：指定索引和字符數
40     // - begin: 開始截取處索引
41     // - count: 截取的字符數量
42     func subString(_ begin:Int,_ count:Int) -> String {
43         let start = self.index(self.startIndex, offsetBy: max(0, begin))
44         let end = self.index(self.startIndex, offsetBy:  min(self.count, begin + count))
45         return String(self[start..<end]) 
46     }    
47 }