-

cool-RR · cool-RR · commit 3266009fa846 · 2014-09-26T21:08:34.000+03:00
diff --git a/source_py3/python_toolbox/math_tools/sequences.py b/source_py3/python_toolbox/math_tools/sequences.py
@@ -60,22 +60,6 @@ def abs_stirling(n, k):
 
 _shitfuck_cache = {}
 
-def get_sub_counters_counter(counter):
-    from python_toolbox import nifty_collections
-    assert isinstance(counter, nifty_collections.FrozenCounter)
-    # d = {}
-    # for key_to_reduce, value_of_key_to_reduce in counter.items():
-        
-        # pass
-    return nifty_collections.FrozenCounter({
-        nifty_collections.FrozenCounter(
-            {(key - (key == key_to_reduce)): value for
-                                                 key, value in counter.items()}
-        ): value_of_key_to_reduce
-                   
-    })
-    
-
 def shitfuck(k, recurrence_counter):
     from python_toolbox import nifty_collections
     from python_toolbox import cute_iter_tools
diff --git a/source_py3/python_toolbox/nifty_collections/__init__.py b/source_py3/python_toolbox/nifty_collections/__init__.py
@@ -10,6 +10,7 @@
 from .lazy_tuple import LazyTuple
 from .frozen_dict import FrozenDict
 from .frozen_counter import FrozenCounter
+from .frozen_chunk_counter import FrozenChunkCounter
 from .default_sorted_dict import DefaultSortedDict
 
 from .emitting_ordered_set import EmittingOrderedSet
diff --git a/source_py3/python_toolbox/nifty_collections/frozen_chunk_counter.py b/source_py3/python_toolbox/nifty_collections/frozen_chunk_counter.py
@@ -1,6 +1,8 @@
 # Copyright 2009-2014 Ram Rachum.,
 # This program is distributed under the MIT license.
 
+import collections
+
 from python_toolbox import math_tools
 
 from .frozen_counter import FrozenCounter
@@ -18,3 +20,14 @@ def __init__(self, iterable):
             else:
                 raise TypeError('Keys to `FrozenChunkCounter` must be '
                                 'non-negative integers.')
+            
+    def get_sub_counters_counter(self):
+        sub_counters_counter = collections.Counter()
+        for key_to_reduce, value_of_key_to_reduce in sub_counter.items():
+            sub_counter = collections.Counter(sub_counter)
+            sub_counter[key_to_reduce] -= 1
+            sub_counter[key_to_reduce - 1] += 1
+            sub_counters_counter[FrozenChunkCounter(sub_counter)] = \
+                                                         value_of_key_to_reduce
+        return FrozenCounter(sub_counters_counter)
+            
