fixed mode count when all numbers are equal

kkloberdanz · Sep 25, 2021 · 01f1af4 · 01f1af4
1 parent ec0692f
commit 01f1af4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/stc.go b/stc.go
@@ -116,6 +116,7 @@ func main() {
 		sort.Slice(allNumbers, func(i, j int) bool {
 			return allNumbers[i].num < allNumbers[j].num
 		})
+		mode = allNumbers[0].num
 		allNumbersLen := len(allNumbers)
 		if allNumbersLen%2 == 0 {
 			a := allNumbers[(allNumbersLen/2)-1].num
@@ -160,6 +161,9 @@ func main() {
 				currentCount = 1
 			}
 		}
+		if modeCount == 0 {
+			modeCount = currentCount
+		}
 	}
 
 	if *gPtr {