diff --git a/src/data-structures/heap/heap.ts b/src/data-structures/heap/heap.ts index f41ed7a..e11d95f 100644 --- a/src/data-structures/heap/heap.ts +++ b/src/data-structures/heap/heap.ts @@ -207,12 +207,7 @@ export class Heap extends IterableElementBase if (comparator) this._comparator = comparator; } - if (elements) { - for (const el of elements) { - if (this.toElementFn) this.add(this.toElementFn(el as R)); - else this.add(el as E); - } - } + this.addMany(elements); } protected _elements: E[] = []; @@ -254,14 +249,24 @@ export class Heap extends IterableElementBase * Time Complexity: O(log n) * Space Complexity: O(1) * - * Insert an element into the heap and maintain the heap properties. - * @param element - The element to be inserted. */ add(element: E): boolean { - this._elements.push(element); + this._elements.push(element as E); return this._bubbleUp(this.elements.length - 1); } + addMany(elements: Iterable | Iterable): boolean[] { + const ans: boolean[] = []; + for (const el of elements) { + if (this._toElementFn) { + ans.push(this.add(this._toElementFn(el as R))); + continue; + } + ans.push(this.add(el as E)); + } + return ans; + } + /** * Time Complexity: O(log n) * Space Complexity: O(1) diff --git a/src/data-structures/linked-list/doubly-linked-list.ts b/src/data-structures/linked-list/doubly-linked-list.ts index 93b2387..0110125 100644 --- a/src/data-structures/linked-list/doubly-linked-list.ts +++ b/src/data-structures/linked-list/doubly-linked-list.ts @@ -524,18 +524,15 @@ export class DoublyLinkedList extends IterableElementBase`. It is an optional parameter that allows you to pass additional * configuration options to customize the behavior of the DoublyLinkedList. */ - constructor(elements: Iterable | Iterable = [], options?: DoublyLinkedListOptions) { + constructor( + elements: Iterable | Iterable | Iterable> = [], + options?: DoublyLinkedListOptions + ) { super(options); this._head = undefined; this._tail = undefined; this._size = 0; - if (elements) { - for (const el of elements) { - if (this.toElementFn) { - this.push(this.toElementFn(el as R)); - } else this.push(el as E); - } - } + this.pushMany(elements); } protected _head: DoublyLinkedListNode | undefined; @@ -700,6 +697,30 @@ export class DoublyLinkedList extends IterableElementBase | Iterable | Iterable>) { + const ans: boolean[] = []; + for (const el of elements) { + if (this.toElementFn) { + ans.push(this.push(this.toElementFn(el as R))); + continue; + } + ans.push(this.push(el as E | DoublyLinkedListNode)); + } + return ans; + } + + unshiftMany(elements: Iterable | Iterable | Iterable>) { + const ans: boolean[] = []; + for (const el of elements) { + if (this.toElementFn) { + ans.push(this.unshift(this.toElementFn(el as R))); + continue; + } + ans.push(this.unshift(el as E | DoublyLinkedListNode)); + } + return ans; + } + /** * Time Complexity: O(n) * Space Complexity: O(1) diff --git a/src/data-structures/linked-list/singly-linked-list.ts b/src/data-structures/linked-list/singly-linked-list.ts index 53954dc..a6d4b7e 100644 --- a/src/data-structures/linked-list/singly-linked-list.ts +++ b/src/data-structures/linked-list/singly-linked-list.ts @@ -60,17 +60,12 @@ export class SinglyLinkedListNode { } export class SinglyLinkedList extends IterableElementBase> { - constructor(elements: Iterable | Iterable = [], options?: SinglyLinkedListOptions) { + constructor( + elements: Iterable | Iterable | Iterable> = [], + options?: SinglyLinkedListOptions + ) { super(options); - if (elements) { - for (const el of elements) { - if (this.toElementFn) { - this.push(this.toElementFn(el as R)); - } else { - this.push(el as E); - } - } - } + this.pushMany(elements); } protected _head: SinglyLinkedListNode | undefined; @@ -211,6 +206,30 @@ export class SinglyLinkedList extends IterableElementBase | Iterable | Iterable>) { + const ans: boolean[] = []; + for (const el of elements) { + if (this.toElementFn) { + ans.push(this.push(this.toElementFn(el as R))); + continue; + } + ans.push(this.push(el as E | SinglyLinkedListNode)); + } + return ans; + } + + unshiftMany(elements: Iterable | Iterable | Iterable>) { + const ans: boolean[] = []; + for (const el of elements) { + if (this.toElementFn) { + ans.push(this.unshift(this.toElementFn(el as R))); + continue; + } + ans.push(this.unshift(el as E | SinglyLinkedListNode)); + } + return ans; + } + /** * Time Complexity: O(n) * Space Complexity: O(1) diff --git a/src/data-structures/queue/deque.ts b/src/data-structures/queue/deque.ts index f922959..037ef6b 100644 --- a/src/data-structures/queue/deque.ts +++ b/src/data-structures/queue/deque.ts @@ -53,14 +53,7 @@ export class Deque extends IterableElementBase> 1) - (needBucketNum >> 1); this._firstInBucket = this._lastInBucket = (this._bucketSize - (_size % this._bucketSize)) >> 1; - - for (const el of elements) { - if (this.toElementFn) { - this.push(this.toElementFn(el as R)); - } else { - this.push(el as E); - } - } + this.pushMany(elements); } protected _bucketSize: number = 1 << 12; @@ -230,6 +223,33 @@ export class Deque extends IterableElementBase extends IterableElementBase | IterableWithSizeOrLength) { + const ans: boolean[] = []; + for (const el of elements) { + if (this.toElementFn) { + ans.push(this.push(this.toElementFn(el as R))); } else { - this._bucketFirst = 0; - this._firstInBucket = 0; + ans.push(this.push(el as E)); } } - this._size -= 1; - return element; + return ans; + } + + unshiftMany(elements: IterableWithSizeOrLength | IterableWithSizeOrLength = []) { + const ans: boolean[] = []; + for (const el of elements) { + if (this.toElementFn) { + ans.push(this.unshift(this.toElementFn(el as R))); + } else { + ans.push(this.unshift(el as E)); + } + } + return ans; } /** diff --git a/src/data-structures/queue/queue.ts b/src/data-structures/queue/queue.ts index ecdf06e..49aa52c 100644 --- a/src/data-structures/queue/queue.ts +++ b/src/data-structures/queue/queue.ts @@ -25,12 +25,7 @@ export class Queue extends IterableElementBase extends IterableElementBase | Iterable) { + const ans: boolean[] = []; + for (const el of elements) { + if (this.toElementFn) ans.push(this.push(this.toElementFn(el as R))); + else ans.push(this.push(el as E)); + } + return ans; + } + /** * Time Complexity: O(1) * Space Complexity: O(1) diff --git a/src/data-structures/stack/stack.ts b/src/data-structures/stack/stack.ts index 344bd4a..b508fc3 100644 --- a/src/data-structures/stack/stack.ts +++ b/src/data-structures/stack/stack.ts @@ -19,15 +19,7 @@ import { IterableElementBase } from '../base'; export class Stack extends IterableElementBase> { constructor(elements: Iterable | Iterable = [], options?: StackOptions) { super(options); - if (elements) { - for (const el of elements) { - if (this.toElementFn) { - this.push(this.toElementFn(el as R)); - } else { - this.push(el as E); - } - } - } + this.pushMany(elements); } protected _elements: E[] = []; @@ -48,11 +40,6 @@ export class Stack extends IterableElementBase extends IterableElementBase extends IterableElementBase | Iterable) { + const ans: boolean[] = []; + for (const el of elements) { + if (this.toElementFn) { + ans.push(this.push(this.toElementFn(el as R))); + } else { + ans.push(this.push(el as E)); + } + } + return ans; + } + + /** + * Time Complexity: O(n) + * Space Complexity: O(1) + * + * The toArray function returns a copy of the elements in an array. + * @returns An array of type E. */ delete(element: E): boolean { const index = this.elements.indexOf(element); @@ -125,9 +135,11 @@ export class Stack extends IterableElementBase extends IterableElementBase> { * @returns The `addMany` method returns an array of boolean values indicating whether each word in * the input iterable was successfully added to the data structure. */ - addMany(words: Iterable | Iterable = []): boolean[] { + addMany(words: Iterable | Iterable): boolean[] { const ans: boolean[] = []; for (const word of words) { if (this.toElementFn) {