C++のSTL関数が返すポインタからインデックスを計算する方法

本記事では、C++においてSTL関数が返すポインタ（イテレータ）からインデックスを求める方法を詳しく解説します。

C++には、目的の要素が格納されているメモリ上のアドレスを指すポインタを返す組み込み関数が数多く存在します。しかし、これらの関数が返すのはあくまでメモリ上の位置であり、コンテナ内での実際のインデックスとは直接対応していません。

典型的な例がstd::max_element()関数です。この関数はコンテナ内の最大要素を検索できますが、戻り値として返すのはその要素のインデックスではなく、メモリ上のアドレス（イテレータ）です。場合によっては、このアドレスからインデックスを取得したい場面があります。ここでは、その具体的な手法を2つ紹介します。

方法1：先頭イテレータとの減算

最もシンプルな方法は、container.begin()メソッドで取得できる先頭位置のアドレスを利用することです。返されたアドレスから先頭アドレスを減算すると、その差分＝インデックスを求めることができます。

これは、ランダムアクセスイテレータ（vectorなど）がポインタのように減算をサポートしているため実現できるテクニックです。

サンプルコード

#include <bits/stdc++.h>
using namespace std;
int main(){
    vector<int> vec = { 10, 40, 50, 60, 30};
    cout << "Max element is : " << (*max_element(vec.begin(), vec.end())) << endl;
    cout << "The index of maximum element : ";
    cout << max_element(vec.begin(), vec.end()) - vec.begin();
}

実行結果

Max element is : 60
The index of maximum element : 3

この例では、最大値「60」の位置がベクターの先頭から3番目であるため、インデックス「3」が出力されます。

方法2：std::distance()を使用する

もう一つの方法として、std::distance()関数を使う方法があります。この関数は、2つのイテレータ間の距離（要素数）を計算してくれるため、減算と同様にインデックスを求められます。

std::distance()の利点は、listやsetのような双方向イテレータしか持たないコンテナでも動作する点です（ただし、その場合は線形時間O(n)で計算される点に注意が必要です）。

サンプルコード

#include <bits/stdc++.h>
using namespace std;
int main(){
    vector<int> vec = { 10, 40, 50, 60, 30};
    cout << "Max element is : " << (*max_element(vec.begin(), vec.end())) << endl;
    cout << "The index of maximum element : ";
    cout << distance(vec.begin(), max_element(vec.begin(),
    vec.end()));
}

実行結果

Max element is : 60
The index of maximum element : 3

まとめ

STL関数が返すイテレータからインデックスを取得するには、以下の2つの方法が有効です。

• begin()との減算：vectorなどのランダムアクセスイテレータ向け。定数時間O(1)で計算可能。
• std::distance()の使用：より汎用的で、listやsetなどにも対応。ただしコンテナの種類によっては計算量が増える。

用途や使用しているコンテナの種類に応じて、適切な方法を選択しましょう。