#3: Fix doubling of map elements in result of solveMax func

arthurkushman · Jul 29, 2019 · 1000938 · 1000938
1 parent f9f997c
commit 1000938
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 44 deletions.
diff --git a/Gopkg.lock b/Gopkg.lock
diff --git a/Gopkg.toml b/Gopkg.toml
diff --git a/hungarian.go b/hungarian.go
@@ -229,6 +229,8 @@ func SolveMax(matrix [][]float64) map[int]map[int]float64 {
 
 	b.setValues()
 
+	b.removeExtra()
+
 	return b.ReducedExtremums
 }
 

diff --git a/hungarian_test.go b/hungarian_test.go
@@ -1,7 +1,7 @@
 package hungarian_test
 
 import (
-	"hungarian"
+	"go-hungarian"
 	"testing"
 )
-Original file line number
+Diff line change
@@ Expand Up / @@ -229,6 +229,8 @@ func SolveMax(matrix [][]float64) map[int]map[int]float64 { @@
     	b.setValues()
+    	b.removeExtra()
     	return b.ReducedExtremums
     }
@@ Expand Down @@