Dynamic Array
(ds/dynamic_array/dynamic_array.hpp)
- View this file on GitHub
- Last update: 2026-06-20 20:05:21+09:00
- Include:
#include "ds/dynamic_array/dynamic_array.hpp"
Overview
DynamicArray is an optimized implementation of an implicit treap, backed by a std::vector memory pool. It acts as a highly dynamic array, supporting index-based insertion, deletion, reversal, rotation, splitting, and concatenation.
Because it relies on an array-based memory pool instead of standard pointers, copying the data structure copies one contiguous pool and avoids per-node heap allocation.
Complexity Notation
In this document:
-
Nis the current number of elements in the array. -
Mis the number of elements inserted or appended from another container. -
Kis the number of elements returned or moved into a newly returned array. -
Vis the current size of the internal memory pool, including erased nodes that have not been reused.
Template Parameters
-
T: The underlying data type of the elements.
Constructors and Assignment Operators
-
DynamicArray()Constructs an empty dynamic array. ($O(1)$) -
DynamicArray(int n)Constructs a dynamic array withnvalue-initialized elements, likestd::vector<T>(n). ($O(N)$) -
DynamicArray(int n, const T& value)Constructs a dynamic array withncopies ofvalue, likestd::vector<T>(n, value). ($O(N)$) -
DynamicArray(const DynamicArray& other)Copy constructor. Deep copies the array structure and memory pool. ($O(V)$) -
DynamicArray(DynamicArray&& other)Move constructor. Takes ownership of the other array’s memory pool without reallocating. ($O(1)$) -
DynamicArray(const std::vector<T>& v)Constructs the dynamic array initialized with elements fromv. ($O(N)$) -
DynamicArray(std::vector<T>&& v)Constructs the dynamic array by moving elements from an rvalue reference of astd::vector. ($O(N)$) -
DynamicArray(std::initializer_list<T> init)Constructs the dynamic array initialized with an initializer list. ($O(N \log N)$) -
operator=Supports both copy and move assignment.
Methods
| Method | Description | Complexity |
|---|---|---|
int size() const |
Returns the current number of elements. | $O(1)$ |
bool empty() const |
Returns whether the array is empty. | $O(1)$ |
void clear() |
Removes all elements. | $O(1)$ |
void push_back(T val) |
Appends val to the end. |
$O(\log N)$ |
void push_front(T val) |
Inserts val at the beginning. |
$O(\log N)$ |
void append(const std::vector<T>& v) |
Appends every element in v. |
$O(M + \log N)$ |
void append(const DynamicArray& other) |
Appends a copy of another dynamic array. | $O(M + \log N)$ |
void insert(int pos, T val) |
Inserts val before index pos. |
$O(\log N)$ |
void insert(int pos, const std::vector<T>& v) |
Inserts all elements of v before index pos. |
$O(M + \log N)$ |
void insert(int pos, const DynamicArray& other) |
Inserts a copy of other before index pos. |
$O(M + \log N)$ |
void erase(int pos) |
Removes the element at index pos. |
$O(\log N)$ |
void erase(int l, int r) |
Removes the half-open range [l, r). |
$O(\log N)$ |
void pop_back(), void pop_front()
|
Removes one element from the back or front. | $O(\log N)$ |
T& at(int pos), const T& at(int pos) const
|
Returns a reference to the element at pos. |
$O(\log N)$ |
operator[] |
Alias for at(pos). |
$O(\log N)$ |
T& front(), T& back()
|
Returns a reference to the first or last element. | $O(\log N)$ |
T get(int pos) const |
Returns a copy of the element at pos. |
$O(\log N)$ |
void set(int pos, T val) |
Overwrites the element at index pos with val. |
$O(\log N)$ |
void reverse(int l, int r) |
Reverses the half-open range [l, r) lazily. |
$O(\log N)$ |
void reverse() |
Reverses the entire array lazily. | Amortized $O(1)$ |
void rotate(int l, int m, int r) |
Moves [m, r) before [l, m), like std::rotate. |
$O(\log N)$ |
std::vector<T> to_vector() const |
Dumps the entire array to std::vector. |
$O(N)$ |
std::vector<T> to_vector(int l, int r) const |
Dumps [l, r) to std::vector, where K = r - l. |
$O(K + \log N)$ |
DynamicArray split_off(int pos) |
Removes [pos, N) and returns it as a new array with its own pool, where K = N - pos. |
$O(K + \log N)$ |
Example
#include "ds/dynamic_array/dynamic_array.hpp"
#include <iostream>
#include <vector>
using namespace m1une::ds;
int main() {
DynamicArray<int> arr = {1, 2, 3, 4, 5};
arr.reverse(1, 4); // {1, 4, 3, 2, 5}
arr.rotate(1, 3, 5); // {1, 2, 5, 4, 3}
arr.insert(2, {10, 11}); // {1, 2, 10, 11, 5, 4, 3}
arr.erase(3, 5); // {1, 2, 10, 4, 3}
arr.push_front(0); // {0, 1, 2, 10, 4, 3}
DynamicArray<int> tail = arr.split_off(3);
arr.append(tail); // {0, 1, 2, 10, 4, 3}
for (int x : arr.to_vector()) {
std::cout << x << " ";
}
std::cout << "\n";
return 0;
}
Required by
Verified with
Code
#ifndef M1UNE_DYNAMIC_ARRAY_HPP
#define M1UNE_DYNAMIC_ARRAY_HPP 1
#include <cassert>
#include <chrono>
#include <cstdint>
#include <initializer_list>
#include <utility>
#include <vector>
namespace m1une {
namespace ds {
template <typename T>
struct DynamicArray {
private:
struct Node {
T val;
int priority;
int count;
int l, r;
bool rev;
Node() : val(T()), priority(0), count(0), l(0), r(0), rev(false) {}
Node(T value, int node_priority)
: val(std::move(value)), priority(node_priority), count(1), l(0), r(0), rev(false) {}
};
std::vector<Node> pool;
int root;
std::uint32_t rng_state;
int new_node(T val) {
pool.push_back(Node(std::move(val), next_priority()));
return pool.size() - 1;
}
int next_priority() {
rng_state ^= rng_state << 13;
rng_state ^= rng_state >> 17;
rng_state ^= rng_state << 5;
return int(rng_state);
}
void update(int t) {
if (t) {
pool[t].count = 1 + pool[pool[t].l].count + pool[pool[t].r].count;
}
}
void apply_reverse(int t) {
if (t) {
pool[t].rev = !pool[t].rev;
}
}
void push(int t) {
if (!t || !pool[t].rev) return;
std::swap(pool[t].l, pool[t].r);
apply_reverse(pool[t].l);
apply_reverse(pool[t].r);
pool[t].rev = false;
}
void split(int t, int pos, int& l, int& r) {
if (!t) {
l = r = 0;
return;
}
if (pos == 0) {
l = 0;
r = t;
return;
}
if (pos == pool[t].count) {
l = t;
r = 0;
return;
}
push(t);
int left_count = pool[pool[t].l].count;
if (pos == left_count) {
l = pool[t].l;
pool[t].l = 0;
update(t);
r = t;
return;
}
if (pos == left_count + 1) {
r = pool[t].r;
pool[t].r = 0;
update(t);
l = t;
return;
}
if (pos <= left_count) {
split(pool[t].l, pos, l, pool[t].l);
r = t;
} else {
split(pool[t].r, pos - left_count - 1, pool[t].r, r);
l = t;
}
update(t);
}
int merge(int l, int r) {
if (!l || !r) return l ? l : r;
if (pool[l].priority > pool[r].priority) {
push(l);
if (pool[l].r) {
pool[l].r = merge(pool[l].r, r);
} else {
pool[l].r = r;
}
update(l);
return l;
} else {
push(r);
if (pool[r].l) {
pool[r].l = merge(l, pool[r].l);
} else {
pool[r].l = l;
}
update(r);
return r;
}
}
int insert_node(int t, int pos, int node) {
if (!t) return node;
if (pool[node].priority > pool[t].priority) {
split(t, pos, pool[node].l, pool[node].r);
update(node);
return node;
}
push(t);
int left_count = pool[pool[t].l].count;
if (pos <= left_count) {
pool[t].l = insert_node(pool[t].l, pos, node);
} else {
pool[t].r = insert_node(pool[t].r, pos - left_count - 1, node);
}
update(t);
return t;
}
int erase_node(int t, int pos) {
push(t);
int left_count = pool[pool[t].l].count;
if (pos < left_count) {
pool[t].l = erase_node(pool[t].l, pos);
update(t);
return t;
}
if (pos == left_count) {
return merge(pool[t].l, pool[t].r);
}
pool[t].r = erase_node(pool[t].r, pos - left_count - 1);
update(t);
return t;
}
int find_node(int t, int pos) {
while (t) {
push(t);
int left_count = pool[pool[t].l].count;
if (pos < left_count) {
t = pool[t].l;
} else if (pos == left_count) {
return t;
} else {
pos -= left_count + 1;
t = pool[t].r;
}
}
return 0;
}
int find_node(int t, int pos, bool reversed) const {
while (t) {
bool cur_reversed = reversed ^ pool[t].rev;
int left = cur_reversed ? pool[t].r : pool[t].l;
int right = cur_reversed ? pool[t].l : pool[t].r;
int left_count = pool[left].count;
if (pos < left_count) {
t = left;
reversed = cur_reversed;
} else if (pos == left_count) {
return t;
} else {
pos -= left_count + 1;
t = right;
reversed = cur_reversed;
}
}
return 0;
}
void dump_dfs(int t, std::vector<T>& res, bool reversed = false) const {
if (!t) return;
bool cur_reversed = reversed ^ pool[t].rev;
int left = cur_reversed ? pool[t].r : pool[t].l;
int right = cur_reversed ? pool[t].l : pool[t].r;
dump_dfs(left, res, cur_reversed);
res.push_back(pool[t].val);
dump_dfs(right, res, cur_reversed);
}
void dump_range_dfs(int t, int ql, int qr, int offset, std::vector<T>& res, bool reversed = false) const {
if (!t || qr <= offset || offset + pool[t].count <= ql) return;
bool cur_reversed = reversed ^ pool[t].rev;
int left = cur_reversed ? pool[t].r : pool[t].l;
int right = cur_reversed ? pool[t].l : pool[t].r;
int left_count = pool[left].count;
int node_pos = offset + left_count;
dump_range_dfs(left, ql, qr, offset, res, cur_reversed);
if (ql <= node_pos && node_pos < qr) {
res.push_back(pool[t].val);
}
dump_range_dfs(right, ql, qr, node_pos + 1, res, cur_reversed);
}
int clone_subtree_from(const DynamicArray& other, int t) {
if (!t) return 0;
int res = static_cast<int>(pool.size());
pool.push_back(other.pool[t]);
pool[res].l = clone_subtree_from(other, other.pool[t].l);
pool[res].r = clone_subtree_from(other, other.pool[t].r);
return res;
}
void update_dfs(int t) {
if (!t) return;
update_dfs(pool[t].l);
update_dfs(pool[t].r);
update(t);
}
int build_cartesian(int first, int last) {
if (first == last) return 0;
std::vector<int> stack;
stack.reserve(last - first);
for (int i = first; i < last; i++) {
int left_child = 0;
while (!stack.empty() && pool[stack.back()].priority < pool[i].priority) {
left_child = stack.back();
stack.pop_back();
}
pool[i].l = left_child;
if (!stack.empty()) {
pool[stack.back()].r = i;
}
stack.push_back(i);
}
int res = stack.front();
update_dfs(res);
return res;
}
int build_from_vector(const std::vector<T>& v) {
int first = int(pool.size());
pool.reserve(pool.size() + v.size());
for (const T& x : v) {
new_node(x);
}
return build_cartesian(first, int(pool.size()));
}
int build_from_vector(std::vector<T>&& v) {
int first = int(pool.size());
pool.reserve(pool.size() + v.size());
for (T& x : v) {
new_node(std::move(x));
}
return build_cartesian(first, int(pool.size()));
}
void reset_to_empty() {
pool.clear();
pool.push_back(Node());
root = 0;
}
public:
DynamicArray() : root(0), rng_state(std::uint32_t(std::chrono::steady_clock::now().time_since_epoch().count())) {
pool.push_back(Node());
if (rng_state == 0) rng_state = 1;
}
DynamicArray(const DynamicArray& other) : pool(other.pool), root(other.root), rng_state(other.rng_state) {}
DynamicArray(DynamicArray&& other) noexcept
: pool(std::move(other.pool)), root(other.root), rng_state(other.rng_state) {
other.reset_to_empty();
}
DynamicArray& operator=(const DynamicArray& other) {
if (this != &other) {
pool = other.pool;
root = other.root;
rng_state = other.rng_state;
}
return *this;
}
DynamicArray& operator=(DynamicArray&& other) noexcept {
if (this != &other) {
pool = std::move(other.pool);
root = other.root;
rng_state = other.rng_state;
other.reset_to_empty();
}
return *this;
}
explicit DynamicArray(int n) : DynamicArray(n, T()) {}
DynamicArray(int n, const T& value) : DynamicArray() {
assert(0 <= n);
pool.reserve(n + 1);
int first = int(pool.size());
for (int i = 0; i < n; i++) {
new_node(value);
}
root = build_cartesian(first, int(pool.size()));
}
explicit DynamicArray(const std::vector<T>& v) : DynamicArray() {
pool.reserve(v.size() + 1);
root = build_from_vector(v);
}
explicit DynamicArray(std::vector<T>&& v) : DynamicArray() {
pool.reserve(v.size() + 1);
root = build_from_vector(std::move(v));
}
DynamicArray(std::initializer_list<T> init) : DynamicArray() {
pool.reserve(init.size() + 1);
for (const T& x : init) push_back(x);
}
int size() const {
return pool[root].count;
}
bool empty() const {
return size() == 0;
}
void clear() {
reset_to_empty();
}
void insert(int pos, T val) {
assert(0 <= pos && pos <= size());
root = insert_node(root, pos, new_node(std::move(val)));
}
void insert(int pos, const std::vector<T>& v) {
assert(0 <= pos && pos <= size());
pool.reserve(pool.size() + v.size());
int mid = build_from_vector(v);
int l, r;
split(root, pos, l, r);
root = merge(merge(l, mid), r);
}
void insert(int pos, std::vector<T>&& v) {
assert(0 <= pos && pos <= size());
pool.reserve(pool.size() + v.size());
int mid = build_from_vector(std::move(v));
int l, r;
split(root, pos, l, r);
root = merge(merge(l, mid), r);
}
void insert(int pos, std::initializer_list<T> init) {
insert(pos, std::vector<T>(init));
}
void insert(int pos, const DynamicArray& other) {
assert(0 <= pos && pos <= size());
if (other.empty()) return;
pool.reserve(pool.size() + other.size());
int mid = clone_subtree_from(other, other.root);
int l, r;
split(root, pos, l, r);
root = merge(merge(l, mid), r);
}
void push_back(T val) {
insert(size(), std::move(val));
}
void push_front(T val) {
insert(0, std::move(val));
}
void append(const std::vector<T>& v) {
insert(size(), v);
}
void append(std::vector<T>&& v) {
insert(size(), std::move(v));
}
void append(const DynamicArray& other) {
insert(size(), other);
}
void erase(int pos) {
assert(0 <= pos && pos < size());
root = erase_node(root, pos);
}
void erase(int l, int r) {
assert(0 <= l && l <= r && r <= size());
if (l == r) return;
int a, b, c;
split(root, l, a, b);
split(b, r - l, b, c);
root = merge(a, c);
}
void pop_back() {
assert(!empty());
erase(size() - 1);
}
void pop_front() {
assert(!empty());
erase(0);
}
T& at(int pos) {
assert(0 <= pos && pos < size());
return pool[find_node(root, pos)].val;
}
const T& at(int pos) const {
assert(0 <= pos && pos < size());
return pool[find_node(root, pos, false)].val;
}
T& operator[](int pos) {
return at(pos);
}
const T& operator[](int pos) const {
return at(pos);
}
T& front() {
assert(!empty());
return at(0);
}
const T& front() const {
assert(!empty());
return at(0);
}
T& back() {
assert(!empty());
return at(size() - 1);
}
const T& back() const {
assert(!empty());
return at(size() - 1);
}
void reverse(int l, int r) {
assert(0 <= l && l <= r && r <= size());
if (l == r) return;
int a, b, c;
split(root, l, a, b);
split(b, r - l, b, c);
apply_reverse(b);
root = merge(merge(a, b), c);
}
void reverse() {
apply_reverse(root);
}
void rotate(int l, int m, int r) {
assert(0 <= l && l <= m && m <= r && r <= size());
if (l == m || m == r) return;
int a, b, c, d;
split(root, l, a, b);
split(b, m - l, b, c);
split(c, r - m, c, d);
root = merge(merge(a, c), merge(b, d));
}
T get(int pos) const {
return at(pos);
}
void set(int pos, T val) {
at(pos) = std::move(val);
}
std::vector<T> to_vector() const {
std::vector<T> res;
res.reserve(size());
dump_dfs(root, res);
return res;
}
std::vector<T> to_vector(int l, int r) const {
assert(0 <= l && l <= r && r <= size());
std::vector<T> res;
res.reserve(r - l);
dump_range_dfs(root, l, r, 0, res);
return res;
}
DynamicArray split_off(int pos) {
assert(0 <= pos && pos <= size());
int l, r;
split(root, pos, l, r);
root = l;
DynamicArray res;
res.pool.reserve(pool[r].count + 1);
res.root = res.clone_subtree_from(*this, r);
return res;
}
};
} // namespace ds
} // namespace m1une
#endif // M1UNE_DYNAMIC_ARRAY_HPP#line 1 "ds/dynamic_array/dynamic_array.hpp"
#include <cassert>
#include <chrono>
#include <cstdint>
#include <initializer_list>
#include <utility>
#include <vector>
namespace m1une {
namespace ds {
template <typename T>
struct DynamicArray {
private:
struct Node {
T val;
int priority;
int count;
int l, r;
bool rev;
Node() : val(T()), priority(0), count(0), l(0), r(0), rev(false) {}
Node(T value, int node_priority)
: val(std::move(value)), priority(node_priority), count(1), l(0), r(0), rev(false) {}
};
std::vector<Node> pool;
int root;
std::uint32_t rng_state;
int new_node(T val) {
pool.push_back(Node(std::move(val), next_priority()));
return pool.size() - 1;
}
int next_priority() {
rng_state ^= rng_state << 13;
rng_state ^= rng_state >> 17;
rng_state ^= rng_state << 5;
return int(rng_state);
}
void update(int t) {
if (t) {
pool[t].count = 1 + pool[pool[t].l].count + pool[pool[t].r].count;
}
}
void apply_reverse(int t) {
if (t) {
pool[t].rev = !pool[t].rev;
}
}
void push(int t) {
if (!t || !pool[t].rev) return;
std::swap(pool[t].l, pool[t].r);
apply_reverse(pool[t].l);
apply_reverse(pool[t].r);
pool[t].rev = false;
}
void split(int t, int pos, int& l, int& r) {
if (!t) {
l = r = 0;
return;
}
if (pos == 0) {
l = 0;
r = t;
return;
}
if (pos == pool[t].count) {
l = t;
r = 0;
return;
}
push(t);
int left_count = pool[pool[t].l].count;
if (pos == left_count) {
l = pool[t].l;
pool[t].l = 0;
update(t);
r = t;
return;
}
if (pos == left_count + 1) {
r = pool[t].r;
pool[t].r = 0;
update(t);
l = t;
return;
}
if (pos <= left_count) {
split(pool[t].l, pos, l, pool[t].l);
r = t;
} else {
split(pool[t].r, pos - left_count - 1, pool[t].r, r);
l = t;
}
update(t);
}
int merge(int l, int r) {
if (!l || !r) return l ? l : r;
if (pool[l].priority > pool[r].priority) {
push(l);
if (pool[l].r) {
pool[l].r = merge(pool[l].r, r);
} else {
pool[l].r = r;
}
update(l);
return l;
} else {
push(r);
if (pool[r].l) {
pool[r].l = merge(l, pool[r].l);
} else {
pool[r].l = l;
}
update(r);
return r;
}
}
int insert_node(int t, int pos, int node) {
if (!t) return node;
if (pool[node].priority > pool[t].priority) {
split(t, pos, pool[node].l, pool[node].r);
update(node);
return node;
}
push(t);
int left_count = pool[pool[t].l].count;
if (pos <= left_count) {
pool[t].l = insert_node(pool[t].l, pos, node);
} else {
pool[t].r = insert_node(pool[t].r, pos - left_count - 1, node);
}
update(t);
return t;
}
int erase_node(int t, int pos) {
push(t);
int left_count = pool[pool[t].l].count;
if (pos < left_count) {
pool[t].l = erase_node(pool[t].l, pos);
update(t);
return t;
}
if (pos == left_count) {
return merge(pool[t].l, pool[t].r);
}
pool[t].r = erase_node(pool[t].r, pos - left_count - 1);
update(t);
return t;
}
int find_node(int t, int pos) {
while (t) {
push(t);
int left_count = pool[pool[t].l].count;
if (pos < left_count) {
t = pool[t].l;
} else if (pos == left_count) {
return t;
} else {
pos -= left_count + 1;
t = pool[t].r;
}
}
return 0;
}
int find_node(int t, int pos, bool reversed) const {
while (t) {
bool cur_reversed = reversed ^ pool[t].rev;
int left = cur_reversed ? pool[t].r : pool[t].l;
int right = cur_reversed ? pool[t].l : pool[t].r;
int left_count = pool[left].count;
if (pos < left_count) {
t = left;
reversed = cur_reversed;
} else if (pos == left_count) {
return t;
} else {
pos -= left_count + 1;
t = right;
reversed = cur_reversed;
}
}
return 0;
}
void dump_dfs(int t, std::vector<T>& res, bool reversed = false) const {
if (!t) return;
bool cur_reversed = reversed ^ pool[t].rev;
int left = cur_reversed ? pool[t].r : pool[t].l;
int right = cur_reversed ? pool[t].l : pool[t].r;
dump_dfs(left, res, cur_reversed);
res.push_back(pool[t].val);
dump_dfs(right, res, cur_reversed);
}
void dump_range_dfs(int t, int ql, int qr, int offset, std::vector<T>& res, bool reversed = false) const {
if (!t || qr <= offset || offset + pool[t].count <= ql) return;
bool cur_reversed = reversed ^ pool[t].rev;
int left = cur_reversed ? pool[t].r : pool[t].l;
int right = cur_reversed ? pool[t].l : pool[t].r;
int left_count = pool[left].count;
int node_pos = offset + left_count;
dump_range_dfs(left, ql, qr, offset, res, cur_reversed);
if (ql <= node_pos && node_pos < qr) {
res.push_back(pool[t].val);
}
dump_range_dfs(right, ql, qr, node_pos + 1, res, cur_reversed);
}
int clone_subtree_from(const DynamicArray& other, int t) {
if (!t) return 0;
int res = static_cast<int>(pool.size());
pool.push_back(other.pool[t]);
pool[res].l = clone_subtree_from(other, other.pool[t].l);
pool[res].r = clone_subtree_from(other, other.pool[t].r);
return res;
}
void update_dfs(int t) {
if (!t) return;
update_dfs(pool[t].l);
update_dfs(pool[t].r);
update(t);
}
int build_cartesian(int first, int last) {
if (first == last) return 0;
std::vector<int> stack;
stack.reserve(last - first);
for (int i = first; i < last; i++) {
int left_child = 0;
while (!stack.empty() && pool[stack.back()].priority < pool[i].priority) {
left_child = stack.back();
stack.pop_back();
}
pool[i].l = left_child;
if (!stack.empty()) {
pool[stack.back()].r = i;
}
stack.push_back(i);
}
int res = stack.front();
update_dfs(res);
return res;
}
int build_from_vector(const std::vector<T>& v) {
int first = int(pool.size());
pool.reserve(pool.size() + v.size());
for (const T& x : v) {
new_node(x);
}
return build_cartesian(first, int(pool.size()));
}
int build_from_vector(std::vector<T>&& v) {
int first = int(pool.size());
pool.reserve(pool.size() + v.size());
for (T& x : v) {
new_node(std::move(x));
}
return build_cartesian(first, int(pool.size()));
}
void reset_to_empty() {
pool.clear();
pool.push_back(Node());
root = 0;
}
public:
DynamicArray() : root(0), rng_state(std::uint32_t(std::chrono::steady_clock::now().time_since_epoch().count())) {
pool.push_back(Node());
if (rng_state == 0) rng_state = 1;
}
DynamicArray(const DynamicArray& other) : pool(other.pool), root(other.root), rng_state(other.rng_state) {}
DynamicArray(DynamicArray&& other) noexcept
: pool(std::move(other.pool)), root(other.root), rng_state(other.rng_state) {
other.reset_to_empty();
}
DynamicArray& operator=(const DynamicArray& other) {
if (this != &other) {
pool = other.pool;
root = other.root;
rng_state = other.rng_state;
}
return *this;
}
DynamicArray& operator=(DynamicArray&& other) noexcept {
if (this != &other) {
pool = std::move(other.pool);
root = other.root;
rng_state = other.rng_state;
other.reset_to_empty();
}
return *this;
}
explicit DynamicArray(int n) : DynamicArray(n, T()) {}
DynamicArray(int n, const T& value) : DynamicArray() {
assert(0 <= n);
pool.reserve(n + 1);
int first = int(pool.size());
for (int i = 0; i < n; i++) {
new_node(value);
}
root = build_cartesian(first, int(pool.size()));
}
explicit DynamicArray(const std::vector<T>& v) : DynamicArray() {
pool.reserve(v.size() + 1);
root = build_from_vector(v);
}
explicit DynamicArray(std::vector<T>&& v) : DynamicArray() {
pool.reserve(v.size() + 1);
root = build_from_vector(std::move(v));
}
DynamicArray(std::initializer_list<T> init) : DynamicArray() {
pool.reserve(init.size() + 1);
for (const T& x : init) push_back(x);
}
int size() const {
return pool[root].count;
}
bool empty() const {
return size() == 0;
}
void clear() {
reset_to_empty();
}
void insert(int pos, T val) {
assert(0 <= pos && pos <= size());
root = insert_node(root, pos, new_node(std::move(val)));
}
void insert(int pos, const std::vector<T>& v) {
assert(0 <= pos && pos <= size());
pool.reserve(pool.size() + v.size());
int mid = build_from_vector(v);
int l, r;
split(root, pos, l, r);
root = merge(merge(l, mid), r);
}
void insert(int pos, std::vector<T>&& v) {
assert(0 <= pos && pos <= size());
pool.reserve(pool.size() + v.size());
int mid = build_from_vector(std::move(v));
int l, r;
split(root, pos, l, r);
root = merge(merge(l, mid), r);
}
void insert(int pos, std::initializer_list<T> init) {
insert(pos, std::vector<T>(init));
}
void insert(int pos, const DynamicArray& other) {
assert(0 <= pos && pos <= size());
if (other.empty()) return;
pool.reserve(pool.size() + other.size());
int mid = clone_subtree_from(other, other.root);
int l, r;
split(root, pos, l, r);
root = merge(merge(l, mid), r);
}
void push_back(T val) {
insert(size(), std::move(val));
}
void push_front(T val) {
insert(0, std::move(val));
}
void append(const std::vector<T>& v) {
insert(size(), v);
}
void append(std::vector<T>&& v) {
insert(size(), std::move(v));
}
void append(const DynamicArray& other) {
insert(size(), other);
}
void erase(int pos) {
assert(0 <= pos && pos < size());
root = erase_node(root, pos);
}
void erase(int l, int r) {
assert(0 <= l && l <= r && r <= size());
if (l == r) return;
int a, b, c;
split(root, l, a, b);
split(b, r - l, b, c);
root = merge(a, c);
}
void pop_back() {
assert(!empty());
erase(size() - 1);
}
void pop_front() {
assert(!empty());
erase(0);
}
T& at(int pos) {
assert(0 <= pos && pos < size());
return pool[find_node(root, pos)].val;
}
const T& at(int pos) const {
assert(0 <= pos && pos < size());
return pool[find_node(root, pos, false)].val;
}
T& operator[](int pos) {
return at(pos);
}
const T& operator[](int pos) const {
return at(pos);
}
T& front() {
assert(!empty());
return at(0);
}
const T& front() const {
assert(!empty());
return at(0);
}
T& back() {
assert(!empty());
return at(size() - 1);
}
const T& back() const {
assert(!empty());
return at(size() - 1);
}
void reverse(int l, int r) {
assert(0 <= l && l <= r && r <= size());
if (l == r) return;
int a, b, c;
split(root, l, a, b);
split(b, r - l, b, c);
apply_reverse(b);
root = merge(merge(a, b), c);
}
void reverse() {
apply_reverse(root);
}
void rotate(int l, int m, int r) {
assert(0 <= l && l <= m && m <= r && r <= size());
if (l == m || m == r) return;
int a, b, c, d;
split(root, l, a, b);
split(b, m - l, b, c);
split(c, r - m, c, d);
root = merge(merge(a, c), merge(b, d));
}
T get(int pos) const {
return at(pos);
}
void set(int pos, T val) {
at(pos) = std::move(val);
}
std::vector<T> to_vector() const {
std::vector<T> res;
res.reserve(size());
dump_dfs(root, res);
return res;
}
std::vector<T> to_vector(int l, int r) const {
assert(0 <= l && l <= r && r <= size());
std::vector<T> res;
res.reserve(r - l);
dump_range_dfs(root, l, r, 0, res);
return res;
}
DynamicArray split_off(int pos) {
assert(0 <= pos && pos <= size());
int l, r;
split(root, pos, l, r);
root = l;
DynamicArray res;
res.pool.reserve(pool[r].count + 1);
res.root = res.clone_subtree_from(*this, r);
return res;
}
};
} // namespace ds
} // namespace m1une