@gkucmierz/utils 1.18.0 → 1.20.0

package/main.mjs

@@ -23,12 +23,18 @@ import {
 import {
   gpn, gpnBI
 } from './src/gpn.mjs'
+import {
+  Heap
+} from './src/heap.mjs'
 import {
   heronsFormula, heronsFormulaBI
 } from './src/herons-formula.mjs'
 import {
   lcm, lcmBI
 } from './src/lcm.mjs'
+import {
+  memoize
+} from './src/memoize.mjs'
 import {
   mod, modBI
 } from './src/mod.mjs'
@@ -56,8 +62,10 @@ export * from './src/factors.mjs';
 export * from './src/gcd.mjs';
 export * from './src/get-type.mjs';
 export * from './src/gpn.mjs';
+export * from './src/heap.mjs';
 export * from './src/herons-formula.mjs';
 export * from './src/lcm.mjs';
+export * from './src/memoize.mjs';
 export * from './src/mod.mjs';
 export * from './src/phi.mjs';
 export * from './src/pow-mod.mjs';
@@ -66,5 +74,5 @@ export * from './src/square-root.mjs';
 export * from './src/tonelli-shanks.mjs';
 
 export default [
-  SetCnt, fromBase64, fromBase64Url, toBase64, toBase64Url, bijective2num, bijective2numBI, num2bijective, num2bijectiveBI, egcd, factors, factorsBI, gcd, gcdBI, getType, gpn, gpnBI, heronsFormula, heronsFormulaBI, lcm, lcmBI, mod, modBI, phi, phiBI, powMod, powModBI, array2range, range2array, squareRoot, squareRootBI, tonelliShanksBI
+  SetCnt, fromBase64, fromBase64Url, toBase64, toBase64Url, bijective2num, bijective2numBI, num2bijective, num2bijectiveBI, egcd, factors, factorsBI, gcd, gcdBI, getType, gpn, gpnBI, Heap, heronsFormula, heronsFormulaBI, lcm, lcmBI, memoize, mod, modBI, phi, phiBI, powMod, powModBI, array2range, range2array, squareRoot, squareRootBI, tonelliShanksBI
 ];

package/package-lock.json

@@ -1,12 +1,12 @@
 {
   "name": "@gkucmierz/utils",
-  "version": "1.18.0",
+  "version": "1.20.0",
   "lockfileVersion": 2,
   "requires": true,
   "packages": {
     "": {
       "name": "@gkucmierz/utils",
-      "version": "1.18.0",
+      "version": "1.20.0",
       "license": "MIT",
       "devDependencies": {
         "husky": "^8.0.1",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gkucmierz/utils",
-  "version": "1.18.0",
+  "version": "1.20.0",
   "description": "Usefull functions for solving programming tasks",
   "main": "main.mjs",
   "scripts": {

package/spec/heap.spec.mjs

@@ -0,0 +1,32 @@
+
+import { Heap } from '../src/heap.mjs';
+
+describe('Heap', () => {
+  it('size', () => {
+    const heap = Heap();
+    expect(heap.size()).toBe(0);
+    heap.add(1);
+    expect(heap.size()).toBe(1);
+    heap.add(1);
+    heap.add(1);
+    expect(heap.size()).toBe(3);
+  });
+
+  it('take', () => {
+    const heap = Heap();
+    heap.add(2);
+    heap.add(1);
+    heap.add(3);
+    expect(heap.take()).toBe(1);
+    expect(heap.take()).toBe(2);
+    expect(heap.take()).toBe(3);
+  });
+
+  it('valFn', () => {
+    const heap = Heap(obj => obj.value);
+    heap.add({ value: 42, other: 'second' });
+    heap.add({ value: 23, other: 'first' });
+    expect(heap.take().other).toBe('first');
+    expect(heap.take().other).toBe('second');
+  });
+});

package/spec/memoize.spec.mjs

@@ -0,0 +1,32 @@
+
+import {
+  memoize,
+} from '../src/memoize.mjs';
+
+describe('memoize', () => {
+  it('called once', () => {
+    let cnt = 0;
+    const add = memoize((a, b) => {
+      ++cnt;
+      return a + b;
+    });
+
+    expect(add(1, 2)).toBe(3);
+    expect(cnt).toBe(1);
+    expect(add(1, 2)).toBe(3);
+    expect(cnt).toBe(1);
+
+    expect(add(2, 1)).toBe(3);
+    expect(cnt).toBe(2);
+  });
+
+  it('variable args length', () => {
+    const fn = memoize((...args) => args.length);
+
+    expect(fn()).toBe(0);
+    expect(fn(1)).toBe(1);
+    expect(fn(1, 2)).toBe(2);
+    expect(fn(1, 2, 3)).toBe(3);
+  });
+});
+

package/src/heap.mjs

@@ -0,0 +1,46 @@
+
+export const Heap = (valFn = n => n) => {
+  const arr = [-1];
+
+  const up = idx => {
+    while (idx > 1) {
+      const ni = idx / 2 | 0;
+      if (valFn(arr[idx]) < valFn(arr[ni])) {
+        [arr[idx], arr[ni]] = [arr[ni], arr[idx]];
+      }
+      idx = ni;
+    }
+    return idx;
+  };
+  
+  return {
+    add: el => up(arr.push(el) - 1),
+    take: () => {
+      const len = arr.length;
+      if (len <= 1) return [][0];
+      let idx = 1;
+      const res = arr[idx];
+      while (idx < len) {
+        const ia = idx * 2;
+        const ib = idx * 2 + 1;
+        if (ia >= len) break;
+        if (ib >= len || valFn(arr[ia]) < valFn(arr[ib])) {
+          arr[idx] = arr[ia];
+          idx = ia;
+        } else {
+          arr[idx] = arr[ib];
+          idx = ib;
+        }
+      }
+      if (idx === arr.length - 1) {
+        arr.pop();
+      } else {
+        arr[idx] = arr.pop();
+        up(idx);
+      }
+      return res;
+    },
+    size: () => arr.length - 1,
+    data: () => arr.slice(1),
+  };
+};

package/src/memoize.mjs

@@ -0,0 +1,38 @@
+
+export const memoize = fn => {
+  const maps = [];
+
+  const getLeafMap = args => {
+    if (!maps[args.length]) maps[args.length] = new Map();
+    const map = maps[args.length];
+
+    const leaf = args.reduce((map, arg) => {
+      if (map.has(arg)) return map.get(arg);
+      const tmp = new Map();
+      map.set(arg, tmp);
+      return tmp;
+    }, map);
+
+    return leaf;
+  };
+
+  const zero = {
+    called: false,
+    ret: null,
+  };
+  
+  return function() {
+    const args = [...arguments];
+    if (args.length === 0) {
+      if (zero.called) return zero.ret;
+      zero.called = true;
+      return zero.ret = fn();
+    }
+    const [firstArg, ...restArgs] = args;
+    const map = getLeafMap(restArgs);
+    if (map.has(firstArg)) return map.get(firstArg);
+    const ret = fn(...args);
+    map.set(firstArg, ret);
+    return ret;
+  };
+};