需要帮助加快计算

Question

I wrote a program that generates a md5 hash onto a printed out bill. I want to be able to check the hash against a generated list of hashes. I then use a Levenshtein distance function to figure out which hash has the lowest edit distance from the printed out bill.

Here is my code:

func checkIfBillIsLegit(stringToCheck:String) -> Bool {
  for i in 0...((secretWords.count)) {                                 // for loop runs about 5 times
     let hashs = String().generateAll(secretWords[i])                  // create the md5 hashs to check against, returns an array with 50 elements
     for j in 0...(hashs.count) {
        if (stringToCheck.minimumEditDistance(hashs[j]) < 5) {        // Levenshtein distance function
           print("legit")
           print(secretWords[i])
           return true
        }
     }
  }

  print("not legit")
  return false
}

I want to be able to run this method multiple times per second. It works now, but it's slightly too slow for what I want to do. Problem is, the generateAll() method is too slow to generate 50 hashes per second. I was thinking of calling generateAll outside of this method, but I can't figure out how I would be able to keep track of the list?

Any help would be appreciated.

generateAll() method:

Answer 1

You can use an NSCache to ensure that you only calculate the hashes for a particular word when required. Typically this will be the first time your function is called, but it could also be if the array of secret words is expanded:

var hashCache = NSCache()

func checkIfBillIsLegit(stringToCheck:String) -> Bool {

    for secretWord in secretWords {                                 
        var hashes = hashCache.objectForKey(secretWord) as? [Hash]
        if hashes == nil {
            hashes= String().generateAll(secretWord)
            hashCache.setObject(hashes, forKey: secretWord)
        }

        for hash in hashes! {
            if stringToCheck.minimumEditDistance(hash) < 5 {
                print("legit")
                print(secretWord)
                return true
            }
        }
    }
    print("not legit")
    return false
}

If you want to know which "secret word" was the match, then I would change the function to return String? :

var hashCache = NSCache()

func checkIfBillIsLegit(stringToCheck:String) -> String? {

    for secretWord in secretWords {                                 
        var hashes = hashCache.objectForKey(secretWord) as? [Hash]
        if hashes == nil {
            hashes= String().generateAll(secretWord)
            hashCache.setObject(hashes, forKey: secretWord)
        }

        for hash in hashes! {
            if stringToCheck.minimumEditDistance(hash) < 5 {
                print("legit")
                print(secretWord)
                return secretWord
            }
        }
    }
    print("not legit")
    return nil
}