Lesson 2 - MaxCounters (from scratch)

Hmm, I've checked old solution after writing new one and now I don't
like any of them ;) Maybe next time I will write some hybrid ;).

Author: mrhead

max_counters.rb

@@ -1,20 +1,24 @@
 def max_counters(n, a)
   counters = Array.new(n, 0)
-
   max = 0
-  last_set_max = 0
+  min = 0
 
-  a.each do |counter|
-    if counter <= n
-      current_counter = counters[counter-1] += 1
-      max = max < current_counter ? current_counter : max
-    else
-      if last_set_max < max
-        counters = Array.new(n, max)
-        last_set_max = max
+  a.each do |v|
+    if v <= n
+      if counters[v - 1] < min + 1
+        counters[v - 1] = min + 1
+      else
+        counters[v - 1] += 1
       end
+      max = [max, counters[v - 1]].max
+    else
+      min = max
     end
   end
+
+  counters.each_index do |i|
+    counters[i] = min if counters[i] < min
+  end
   counters
 end
 
@@ -26,6 +30,10 @@ def test_example_input
   end
 
   def test_only_max_counter
-    assert_equal [0, 0, 0, 0, 0], max_counters(5, [6, 6, 6])
+    assert_equal [0, 0, 0], max_counters(3, [4, 4, 4, 4, 4])
+  end
+
+  def test_max_at_the_end
+    assert_equal [4, 4, 4], max_counters(3, [1, 1, 1, 1, 4])
   end
 end