// A must be comparable sclass ArrayTreeMap<A, B> extends AbstractMap<A, B> { O[] array; // key, value, key, value... int size; *() {} *(Map<A, B> map) { array = new O[roundUpToPowerOfTwo(l(map))*2]; fill(sorted(keys(map)), map, 0, l(array)/2, 0); size = l(map); } public int size() { ret size; } public Set<Map.Entry<A, B>> entrySet() { ret new EntrySet; } bool inRange(int i) { ret i >= 0 && i < l(array); } int check(int i) { ret inRange(i) ? i : -1; } int parent(int i) { ret check(i == 0 ? -1 : ((i-2)/2) & ~1); } int left(int i) { ret check(i*2+2); } int right(int i) { ret check(i*2+4); } int first() { if (l(array) == 0) ret -1; int p = 0; while (left(p) >= 0) p = left(p); ret p; } int successor(int t) { if (!inRange(t)) ret -1; if (right(t) >= 0) { int p = right(t); while (left(p) >= 0) p = left(p); ret p; } int p = parent(t); int ch = t; while (p >= 0 && ch == right(p)) { ch = p; p = parent(p); } ret p; } void fill(L<A> l, Map<A, B> map, int lIdx1, int lIdx2, int aIdx) { if (lIdx2 <= lIdx1) ret; int middle = (lIdx1+lIdx2)/2; A key = _get(l, middle); array[aIdx*2] = key; array[aIdx*2+1] = map.get(key); fill(l, map, lIdx1, middle, aIdx*2+1); fill(l, map, middle+1, lIdx2, aIdx*2+2); } int find(O a) { int i = 0; while (i < l(array)) { int x = array[i] == null ? -1 : cmp(a, array[i]); if (x == 0) ret i; i = i*2+(x > 0 ? 4 : 2); } ret -1; } public bool containsKey(O a) { ret find(a) >= 0; } public B get(O a) { int i = find(a); ret i >= 0 ? (B) array[i+1] : null; } class EntrySet extends AbstractSet<Map.Entry<A, B>> { public Iterator<Map.Entry<A, B>> iterator() { ret new EntryIterator(first()); } public int size() { ret ArrayTreeMap.this.size(); } } class EntryIterator implements Iterator<Map.Entry<A, B>> { int next = -1, lastReturned = -1; *(int first) { next = first; } public final bool hasNext() { ret next >= 0; } public Map.Entry<A, B> next() { int e = next; if (e < 0) throw new NoSuchElementException; do { next = successor(next); } while (next >= 0 && array[next] == null); lastReturned = e; ret new SimpleImmutableEntry((A) array[e], (B) array[e+1]); } } }
Began life as a copy of #1011276
download show line numbers debug dex old transpilations
Travelled to 13 computer(s): aoiabmzegqzx, bhatertpkbcr, cbybwowwnfue, cfunsshuasjs, gwrvuhgaqvyk, ishqpsrjomds, lpdgvwnxivlt, mqqgnosmbjvj, pyentgdyhuwx, pzhvpgtvlbxg, tslmcundralx, tvejysmllsmz, vouqrxazstgt
No comments. add comment
Snippet ID: | #1011277 |
Snippet name: | ArrayTreeMap - quite compact map, but SortedImmutableList is more compact |
Eternal ID of this version: | #1011277/22 |
Text MD5: | 0adcd67678ce58199577c84e468d5799 |
Author: | stefan |
Category: | javax |
Type: | JavaX fragment (include) |
Public (visible to everyone): | Yes |
Archived (hidden from active list): | No |
Created/modified: | 2017-10-29 23:33:47 |
Source code size: | 2644 bytes / 117 lines |
Pitched / IR pitched: | No / No |
Views / Downloads: | 343 / 959 |
Version history: | 21 change(s) |
Referenced in: | [show references] |