maxEqualFreq

Author: masx200

maximum-equal-frequency/export.go

@@ -0,0 +1,9 @@
+package index
+
+func Max(a, b int) int {
+	return max(a, b)
+}
+func MaxEqualFreq(nums []int) int {
+
+	return maxEqualFreq(nums)
+}

maximum-equal-frequency/go.mod

@@ -0,0 +1,3 @@
+module github.com/masx200/leetcode-test/maximum-equal-frequency
+
+go 1.19

maximum-equal-frequency/index.go

@@ -0,0 +1,28 @@
+package index
+
+func maxEqualFreq(nums []int) (ans int) {
+	freq := map[int]int{}
+	count := map[int]int{}
+	maxFreq := 0
+	for i, num := range nums {
+		if count[num] > 0 {
+			freq[count[num]]--
+		}
+		count[num]++
+		maxFreq = max(maxFreq, count[num])
+		freq[count[num]]++
+		if maxFreq == 1 ||
+			freq[maxFreq]*maxFreq+freq[maxFreq-1]*(maxFreq-1) == i+1 && freq[maxFreq] == 1 ||
+			freq[maxFreq]*maxFreq+1 == i+1 && freq[1] == 1 {
+			ans = max(ans, i+1)
+		}
+	}
+	return
+}
+
+func max(a, b int) int {
+	if b > a {
+		return b
+	}
+	return a
+}