RustのVecで要素を追加・削除する方法：基本操作からパフォーマンス改善まで

Rustのプログラミングにおいて、可変長配列であるVec型は最も頻繁に利用されるデータ構造の一つです。

Vecはメモリのヒープ領域にデータを格納し、プログラムの実行中に要素を動的に増やしたり減らしたりすることができます。

本記事では、Rust初学者から中級者の方に向けて、Vecの基本的な操作から、パフォーマンスを意識した効率的な要素の追加・削除手法まで詳しく解説します。

Rustの所有権やメモリ管理の仕組みを理解しながら、最適なコードを書くための知識を深めていきましょう。

RustのVecにおける基本的な概念と初期化

Vecは、同じ型の値をメモリ上に連続して配置するコレクション型です。

標準ライブラリでは Vec<T> として定義されており、Tには任意の型を指定できます。

Vecを操作する上で最も重要な概念は、「サイズ(Length)」と「容量(Capacity)」の違いです。

サイズは現在格納されている要素の数を指し、容量は再メモリ割り当てを行わずに格納できる最大の要素数を指します。

まずは、Vecを作成する基本的な方法を見ていきましょう。

fn main() {
    // 空のVecを生成する
    let mut vec1: Vec<i32> = Vec::new();

    // マクロを使用して初期値を指定して生成する
    let vec2 = vec![1, 2, 3, 4, 5];

    // 容量を事前に指定して生成する
    let mut vec3 = Vec::with_capacity(10);

    println!("vec1 len: {}, cap: {}", vec1.len(), vec1.capacity());
    println!("vec2 len: {}, cap: {}", vec2.len(), vec2.capacity());
    println!("vec3 len: {}, cap: {}", vec3.len(), vec3.capacity());
}

vec1 len: 0, cap: 0
vec2 len: 5, cap: 5
vec3 len: 0, cap: 10

効率的なメモリ確保を行うwith_capacity

要素が大量に追加されることがあらかじめわかっている場合、Vec::new() ではなく Vec::with_capacity(n) を使用することが推奨されます。

通常の push 操作では、容量が不足するたびにメモリの再確保(Reallocation)が発生し、既存の要素を新しいメモリ領域へコピーするコストが生じます。

with_capacity を活用することで、この再確保の回数を最小限に抑え、プログラムの実行速度を劇的に向上させることが可能です。

要素を追加するための主要なメソッド

Vecに新しいデータを追加する方法は、用途に応じていくつか存在します。

最も一般的なのは末尾への追加ですが、特定の位置に挿入したり、他のコレクションを結合したりすることも可能です。

pushメソッドによる末尾への追加

最も頻繁に使用されるのが pushメソッド です。

これはVecの末尾に新しい要素を一つ追加します。

計算量は償却定数時間 O(1) であり、非常に高速に動作します。

fn main() {
    let mut numbers = vec![10, 20];
    
    // 末尾に要素を追加
    numbers.push(30);
    numbers.push(40);

    println!("{:?}", numbers);
}

[10, 20, 30, 40]

insertメソッドによる位置指定の挿入

特定のインデックス位置に要素を挿入したい場合は、insert メソッドを使用します。

ただし、指定した位置以降のすべての要素を右側にずらす必要があるため、計算量は O(n) となります。

Vecの先頭に近い位置に挿入を繰り返す処理は、要素数が多い場合にパフォーマンスのボトルネックとなる可能性があるため注意が必要です。

fn main() {
    let mut fruits = vec!["apple", "banana"];
    
    // インデックス1の位置に挿入
    fruits.insert(1, "orange");

    println!("{:?}", fruits);
}

["apple", "orange", "banana"]

複数の要素を一度に追加するextendとappend

複数の要素をまとめて追加したい場合には、extend または append を使用します。

extend はイテレータを引数に取り、その内容をVecにコピーします。

一方、append は別のVecを引数に取り、そのVecのすべての要素を移動(ムーブ)させます。

append を使用した後の引数側のVecは空になります。

fn main() {
    let mut v1 = vec![1, 2];
    let v2 = vec![3, 4];
    
    // イテレータから追加
    v1.extend([5, 6].iter());

    // 別のVecから追加(v2は空になる)
    let mut v3 = vec![7, 8];
    v1.append(&mut v3);

    println!("v1: {:?}", v1);
    println!("v3: {:?}", v3);
}

v1: [1, 2, 5, 6, 7, 8]
v3: []

要素を削除するための多様なアプローチ

RustのVecには、メモリの安全性とパフォーマンスのバランスを考慮した複数の削除用メソッドが用意されています。

削除する位置や、順序を維持する必要があるかどうかによって、最適なメソッドを選択しましょう。

popメソッドによる末尾の取り出し

末尾の要素を削除して、その値を Option<T> 型で返却するのが pop メソッドです。

要素が存在する場合は Some(value)、空の場合は None を返します。

この操作は他の要素を移動させる必要がないため、O(1) で動作します。

fn main() {
    let mut stack = vec![1, 2, 3];
    
    let last = stack.pop();
    
    println!("Popped: {:?}", last);
    println!("Stack: {:?}", stack);
}

Popped: Some(3)
Stack: [1, 2]

removeメソッドとインデックスによる削除

remove メソッドを使用すると、特定のインデックスにある要素を削除できます。

削除された位置を埋めるために、それ以降の要素がすべて左側に詰められます。

したがって、これも insert と同様に O(n) の計算量がかかります。

fn main() {
    let mut items = vec!["A", "B", "C", "D"];
    
    // インデックス1("B")を削除
    let removed = items.remove(1);
    
    println!("Removed: {}", removed);
    println!("Items: {:?}", items);
}

Removed: B
Items: ["A", "C", "D"]

高効率な削除を実現するswap_remove

もし要素の並び順が重要でない場合、swap_remove メソッドが非常に有用です。

これは、指定したインデックスの要素を末尾の要素と入れ替えてから削除します。

すべての要素を詰める必要がないため、どの位置の削除であっても O(1) で完了します。

大規模なVecからランダムに要素を削除する際のパフォーマンス改善に非常に効果的です。

fn main() {
    let mut values = vec!["first", "second", "third", "last"];
    
    // インデックス0をswap_removeで削除
    values.swap_remove(0);
    
    println!("Values after swap_remove: {:?}", values);
}

Values after swap_remove: ["last", "second", "third"]

条件に応じた一括処理とフィルタリング

特定の条件に合致する要素だけをVecから取り除きたい場合、個別にループを回して remove を呼ぶのは非効率です。

Rustにはそのようなユースケースに最適なメソッドが備わっています。

retainメソッドによるフィルタリング

retain メソッドは、クロージャ(匿名関数)を引数に取り、その条件が true を返す要素だけをVecに残します。

不要な要素はインプレース(その場)で削除されます。

これは、新しくVecを作成して入れ直すよりもメモリ効率が良い方法です。

fn main() {
    let mut numbers = vec![1, 2, 3, 4, 5, 6];
    
    // 偶数だけを残す
    numbers.retain(|&x| x % 2 == 0);
    
    println!("{:?}", numbers);
}

[2, 4, 6]

drainメソッドによる範囲削除と取得

drain メソッドは、指定した範囲の要素をVecから削除し、その削除された要素をイテレータとして返します。

Vecの一部を切り出して別の処理に回したい場合に非常に強力です。

fn main() {
    let mut v = vec![1, 2, 3, 4, 5];
    
    // インデックス1から3(4は含まない)までを抜き出す
    let drained: Vec<_> = v.drain(1..4).collect();
    
    println!("Drained: {:?}", drained);
    println!("Remaining: {:?}", v);
}

Drained: [2, 3, 4]
Remaining: [1, 5]

Vecのパフォーマンスを最大化するためのTips

Vecの操作をより高速にするための、実践的な最適化テクニックをまとめます。

Rustのメモリアロケータは優秀ですが、ヒープへのアクセスを減らすことが実行速度の向上に直結します。

また、可能な限り不変参照(immutable reference)を利用し、必要なときだけ mut を付与することも、コンパイラの最適化を助けることにつながります。

ループ内で push を行う場合は、事前に reserve メソッドを呼んで必要な容量を確保しておくことも検討してください。

まとめ

RustのVecは非常に柔軟で強力なデータ構造ですが、そのメソッドの選択肢は多岐にわたります。

基本的な push や pop を使いこなすことはもちろん、swap_remove や retain、with_capacity といったメソッドを状況に応じて使い分けることが重要です。

特に計算量(Time Complexity)を意識することで、要素数が増えた際にもパフォーマンスの低下を防ぐことができます。

Rustの強力な型システムとメモリ管理機能を最大限に活かし、安全かつ高速なコードを目指しましょう。

今回紹介したテクニックを日々の開発に活用して、より効率的なRustプログラミングを実践してください。