2009-04-05 18:02:44 -04:00
|
|
|
module RailsGuides
|
|
|
|
module Levenshtein
|
2014-06-28 03:19:46 -04:00
|
|
|
# This code is based directly on the Text gem implementation
|
|
|
|
# Returns a value representing the "cost" of transforming str1 into str2
|
|
|
|
def self.distance str1, str2
|
|
|
|
s = str1
|
|
|
|
t = str2
|
|
|
|
n = s.length
|
|
|
|
m = t.length
|
|
|
|
max = n/2
|
2009-03-21 19:40:35 -04:00
|
|
|
|
2014-06-28 03:19:46 -04:00
|
|
|
return m if (0 == n)
|
|
|
|
return n if (0 == m)
|
|
|
|
return n if (n - m).abs > max
|
2009-03-21 19:40:35 -04:00
|
|
|
|
2014-06-28 03:19:46 -04:00
|
|
|
d = (0..m).to_a
|
|
|
|
x = nil
|
|
|
|
|
|
|
|
str1.each_char.each_with_index do |char1,i|
|
|
|
|
e = i+1
|
|
|
|
|
|
|
|
str2.each_char.each_with_index do |char2,j|
|
|
|
|
cost = (char1 == char2) ? 0 : 1
|
|
|
|
x = [
|
|
|
|
d[j+1] + 1, # insertion
|
|
|
|
e + 1, # deletion
|
|
|
|
d[j] + cost # substitution
|
|
|
|
].min
|
|
|
|
d[j] = e
|
|
|
|
e = x
|
2009-04-05 18:02:44 -04:00
|
|
|
end
|
2014-06-28 03:19:46 -04:00
|
|
|
|
|
|
|
d[m] = x
|
2009-03-21 19:40:35 -04:00
|
|
|
end
|
2009-03-15 15:03:37 -04:00
|
|
|
|
2014-06-28 03:19:46 -04:00
|
|
|
return x
|
2009-04-05 18:02:44 -04:00
|
|
|
end
|
2009-03-21 19:40:35 -04:00
|
|
|
end
|
2009-03-15 15:03:37 -04:00
|
|
|
end
|