Java 8でインデックスを使ってStreamを反復処理する方法

Java 8のStream APIには、要素と同時にインデックスを取得する機能が標準では用意されていません。しかし、AtomicIntegerクラスを活用することで、各要素にインデックスを付けながらStreamを反復処理できます。以下に具体的な実装例を示します。

サンプルコード

import java.util.stream.IntStream;
import java.util.*;
import java.util.concurrent.atomic.AtomicInteger;
public class Demo{
    public static void main(String[] args){
        String[] my_array = { "T", "h", "i", "s", "s","a", "m", "p", "l", "e" };
        AtomicInteger my_index = new AtomicInteger();
        System.out.println("The elements in the string array are :");
        Arrays.stream(my_array).map(str -> my_index.getAndIncrement() + " -> " + str).forEach(System.out::println);
    }
}

出力結果

The elements in the string array are :
0 -> T
1 -> h
2 -> i
3 -> s
4 -> s
5 -> a
6 -> m
7 -> p
8 -> l
9 -> e

コードの解説

まず、Demoという名前のクラス内にmainメソッドが定義されています。このメソッドでは、文字列型の配列my_arrayを宣言し、続いてAtomicIntegerクラスのインスタンスmy_indexを生成しています。

Arrays.stream()メソッドで配列からStreamを作成し、map()操作の中でgetAndIncrement()関数を呼び出します。この関数は現在の値を返した後、カウンターを1つ増やす動作をするため、各要素に対して0から始まる連番を自動的に付与できます。最終的に、インデックスと要素が「0 -> T」のような形式で結合され、forEach(System.out::println)によってコンソールへ1行ずつ出力されます。

なぜAtomicIntegerを使うのか

ラムダ式やStreamの内部から参照されるローカル変数は、事実上final（effectively final）である必要があります。通常のint変数は再代入できないためカウンターとして利用できませんが、AtomicIntegerはオブジェクトの内部状態を更新できるうえ、スレッドセーフに動作するため、並列ストリームでも安全にインデックスを管理できます。

別のアプローチ：IntStreamを活用する方法

IntStream.range()を使えば、インデックスを明示的に生成しながら配列要素へアクセスすることも可能です。逐次処理の場合はこちらの方が直感的に理解しやすいでしょう。

IntStream.range(0, my_array.length)
         .forEach(i -> System.out.println(i + " -> " + my_array[i]));