text 蛮力找到带前缀的单词

Posted 2021-04-30

import Foundation
import PlaygroundSupport

func wordsWith(prefix: String, in dictionary: [String]) -> [String] {
    var output: [String] = []
    let prefixCount = prefix.count
    let prefixCharArray = Array(prefix)
    wordLoop: for word in dictionary {
        let charArray = Array(word)
        var wordIndex = 0
        var prefixIndex = 0
        while prefixIndex < prefixCount {
            if charArray[wordIndex] != prefixCharArray[prefixIndex] {
                continue wordLoop
            }
            wordIndex += 1
            prefixIndex += 1
        }
        output.append(word)
    }
    return output
}
let dictionary = ["CAT", "DOG", "BUNNY", "CAN", "CUT", "DOLL"]
print(wordsWith(prefix: "CA", in: dictionary))