aparoksha/levenshtein-transformations
1
0
Fork 1
Code Issues Pull Requests Actions Releases 2 Activity

Fix crashing when source or target array empty

Browse Source
This commit is contained in:
david-swift 2024-01-01 12:29:34 +01:00
parent cfaa94f08e
commit a88dee4c4b
1 changed files with 20 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
Sources/LevenshteinTransformations/LevenshteinTransformations.swift
View File

@ -80,24 +80,27 @@ enum LevenshteinTransformations {
operations[0][targetIndex] = .insert
}
for sourceIndex in 1...sourceCount {
for targetIndex in 1...targetCount {
if source[sourceIndex - 1] == target[targetIndex - 1] {
editDistances[sourceIndex][targetIndex] = editDistances[sourceIndex - 1][targetIndex - 1]
operations[sourceIndex][targetIndex] = .noChange
} else {
editDistances[sourceIndex][targetIndex] = 1 + min(
editDistances[sourceIndex - 1][targetIndex - 1],
editDistances[sourceIndex - 1][targetIndex],
editDistances[sourceIndex][targetIndex - 1]
)
if editDistances[sourceIndex][targetIndex] == editDistances[sourceIndex - 1][targetIndex - 1] + 1 {
operations[sourceIndex][targetIndex] = .replace
} else if editDistances[sourceIndex][targetIndex]
== editDistances[sourceIndex - 1][targetIndex] + 1 {
operations[sourceIndex][targetIndex] = .delete
if sourceCount > 0 && targetCount > 0 {
for sourceIndex in 1...sourceCount {
for targetIndex in 1...targetCount {
if source[sourceIndex - 1] == target[targetIndex - 1] {
editDistances[sourceIndex][targetIndex] = editDistances[sourceIndex - 1][targetIndex - 1]
operations[sourceIndex][targetIndex] = .noChange
} else {
operations[sourceIndex][targetIndex] = .insert
editDistances[sourceIndex][targetIndex] = 1 + min(
editDistances[sourceIndex - 1][targetIndex - 1],
editDistances[sourceIndex - 1][targetIndex],
editDistances[sourceIndex][targetIndex - 1]
)
if editDistances[sourceIndex][targetIndex]
== editDistances[sourceIndex - 1][targetIndex - 1] + 1 {
operations[sourceIndex][targetIndex] = .replace
} else if editDistances[sourceIndex][targetIndex]
== editDistances[sourceIndex - 1][targetIndex] + 1 {
operations[sourceIndex][targetIndex] = .delete
} else {
operations[sourceIndex][targetIndex] = .insert
}
}
}
}