Java Stream轉換Map

一、介紹

Java 8中引入了Stream API，提供了豐富的功能，方便我們對集合進行各種操作，其中一個非常常見的操作就是將Stream轉換為一個Map。

二、Stream轉換為Map的方法

Java 8為了方便我們將Stream轉換為Map，提供了兩個方法：collect()和toMap()。

三、collect()方法

collect()方法是一個非常強大的方法，可以讓我們將Stream中的元素匯總為一個集合，或者進行分組、過濾等操作，並將結果轉換為List、Set、Map等類型。

將Stream轉換為Map的語法如下：

Map<K, V> map = stream.collect(Collectors.toMap(keyMapper, valueMapper));

其中，keyMapper和valueMapper是兩個函數式接口，分別用於將Stream的每個元素映射為Map的key和value。如果key存在重複的情況，會拋出IllegalStateException，可以使用第三個參數mergeFunction來處理重複的key。

下面是一個示例代碼：

List<String> list = Arrays.asList("a", "b", "c", "d", "e");
Map<String, Integer> map = list.stream().collect(Collectors.toMap(Function.identity(), String::length));
System.out.println(map); // 輸出：{a=1, b=1, c=1, d=1, e=1}

這個例子中，我們將一個List轉換為一個String為key，Integer為value的Map，這個Map中所有的value都等於1。

四、toMap()方法

toMap()方法也是用於將Stream轉換為Map的方法，不過與collect()方法不同的是，它直接返回一個Map，而不是一個Collector。

toMap()方法的語法如下：

Map<K, V> map = stream.collect(Collectors.toMap(keyMapper, valueMapper, mergeFunction));

其中，mergeFunction是用於處理重複key的情況的函數式接口。如果key重複了，將會調用mergeFunction來處理這個key對應的value。

下面是一個示例代碼：

List<String> list = Arrays.asList("apple", "banana", "cherry", "apple");
Map<String, Integer> map = list.stream().collect(Collectors.toMap(Function.identity(), String::length, (v1, v2) -> v1 + v2));
System.out.println(map); // 輸出：{apple=10, banana=6, cherry=6}

這個例子中，我們將一個List轉換為一個String為key，Integer為value的Map。因為List中有兩個“apple”，所以mergeFunction將這兩個“apple”的value相加，得到10。

五、結論

Java Stream提供了非常方便的方法將一個Stream轉換為Map類型，我們可以使用collect()方法或toMap()方法來實現。在使用的時候，需要注意處理重複key的情況。