Skip to content

Java 與 Kotlin 中的集合

集合 (Collections) 是包含可變數量項目(可能為零)的群組,這些項目對於要解決的問題很重要且通常會被操作。 本指南解釋並比較 Java 和 Kotlin 中的集合概念與操作。 它將幫助您從 Java 遷移到 Kotlin,並以純正的 Kotlin 方式編寫程式碼。

本指南的第一部分包含 Java 和 Kotlin 中相同集合操作的快速詞彙表。 它分為 Java 和 Kotlin 中相同的操作僅存在於 Kotlin 中的操作。 本指南的第二部分,從 可變性 開始,透過觀察特定案例來解釋一些差異。

有關集合的介紹,請參閱 集合概述 或觀看 Kotlin 開發者倡導者 Sebastian Aigner 的此 影片

NOTE

以下所有範例僅使用 Java 和 Kotlin 標準函式庫 API。

Java 和 Kotlin 中相同的操作

在 Kotlin 中,許多集合操作與 Java 中對應的操作看起來完全相同。

對列表、集合 (Set)、佇列 (Queue) 和雙向佇列 (Deque) 的操作

描述常用操作更多 Kotlin 替代方案
新增一個或多個元素add(), addAll()使用 plusAssign(+=) 運算子collection += elementcollection += anotherCollection
檢查集合是否包含一個或多個元素contains(), containsAll()使用 in 關鍵字 以運算子形式呼叫 contains()element in collection
檢查集合是否為空isEmpty()使用 isNotEmpty() 來檢查集合是否不為空。
在特定條件下移除removeIf()
只保留選定的元素retainAll()
從集合中移除所有元素clear()
從集合中獲取串流stream()Kotlin 有自己處理串流的方式:序列 以及像 map()filter() 這樣的方法。
從集合中獲取迭代器iterator()

對映射 (Map) 的操作

描述常用操作更多 Kotlin 替代方案
新增一個或多個元素put(), putAll(), putIfAbsent()在 Kotlin 中,賦值 map[key] = value 的行為與 put(key, value) 相同。此外,您可以使用 plusAssign(+=) 運算子map += Pair(key, value)map += anotherMap
取代一個或多個元素put(), replace(), replaceAll()使用索引運算子 map[key] = value 代替 put()replace()
獲取元素get()使用索引運算子獲取元素:map[index]
檢查 Map 是否包含一個或多個元素containsKey(), containsValue()使用 in 關鍵字 以運算子形式呼叫 contains()element in map
檢查 Map 是否為空isEmpty()使用 isNotEmpty() 來檢查 Map 是否不為空。
移除元素remove(key), remove(key, value)使用 minusAssign(-=) 運算子map -= key
從 Map 中移除所有元素clear()
從 Map 中獲取串流stream() 於 entry、key 或 value 上

僅存在於列表的操作

描述常用操作更多 Kotlin 替代方案
獲取元素的索引indexOf()
獲取元素的最後一個索引lastIndexOf()
獲取元素get()使用索引運算子獲取元素:list[index]
取得子列表subList()
取代一個或多個元素set(), replaceAll()使用索引運算子代替 set()list[index] = value

略有差異的操作

對任何集合類型的操作

描述JavaKotlin
獲取集合大小size()count(), size
對巢狀集合元素進行扁平化存取collectionOfCollections.forEach(flatCollection::addAll)collectionOfCollections.stream().flatMap().collect()flatten()flatMap()
對每個元素應用給定的函數stream().map().collect()map()
依序將提供的操作應用於集合元素並返回累積結果stream().reduce()reduce(), fold()
根據分類器分組元素並計數stream().collect(Collectors.groupingBy(classifier, counting()))eachCount()
按條件過濾stream().filter().collect()filter()
檢查集合元素是否滿足條件stream().noneMatch(), stream().anyMatch(), stream().allMatch()none(), any(), all()
排序元素stream().sorted().collect()sorted()
取得前 N 個元素stream().limit(N).collect()take(N)
依謂詞取得元素stream().takeWhile().collect()takeWhile()
跳過前 N 個元素stream().skip(N).collect()drop(N)
依謂詞跳過元素stream().dropWhile().collect()dropWhile()
從集合元素及其相關值建構 Mapstream().collect(toMap(keyMapper, valueMapper))associate()

若要對 Map 執行上述所有操作,您首先需要取得該 Map 的 entrySet

對列表的操作

描述JavaKotlin
依自然順序排序列表sort(null)sort()
依降序排序列表sort(comparator)sortDescending()
從列表中移除元素remove(index), remove(element)removeAt(index), remove(element)collection -= element
用特定值填充列表的所有元素Collections.fill()fill()
從列表中獲取唯一元素stream().distinct().toList()distinct()

Java 標準函式庫中不存在的操作

如果您想深入了解 zip()chunked()windowed() 和其他一些操作,請觀看 Sebastian Aigner 關於 Kotlin 中進階集合操作的此影片:

可變性

在 Java 中,存在可變集合:

java
// Java
// This list is mutable!
public List<Customer> getCustomers() { ... }

部分可變的:

java
// Java
List<String> numbers = Arrays.asList("one", "two", "three", "four");
numbers.add("five"); // Fails in runtime with `UnsupportedOperationException`

以及不可變的:

java
// Java
List<String> numbers = new LinkedList<>();
// This list is immutable!
List<String> immutableCollection = Collections.unmodifiableList(numbers);
immutableCollection.add("five"); // Fails in runtime with `UnsupportedOperationException`

如果您在 IntelliJ IDEA 中編寫最後兩段程式碼,IDE 將會警告您正在嘗試修改一個不可變的物件。 這段程式碼將會編譯成功,但在執行時會因為 UnsupportedOperationException 而失敗。您無法透過查看集合的類型來判斷其是否可變。

與 Java 不同,在 Kotlin 中,您可以根據需要明確宣告可變或唯讀的集合。 如果您嘗試修改唯讀集合,程式碼將不會編譯成功:

kotlin
// Kotlin
val numbers = mutableListOf("one", "two", "three", "four")
numbers.add("five")            // This is OK
val immutableNumbers = listOf("one", "two")
//immutableNumbers.add("five") // Compilation error - Unresolved reference: add

請在 Kotlin 程式碼慣例 頁面閱讀更多關於不可變性的內容。

共變性

在 Java 中,您不能將具有子類型 (descendant type) 的集合傳遞給接受祖先類型 (ancestor type) 集合的函數。 例如,如果 Rectangle 繼承自 Shape,您不能將 Rectangle 元素的集合傳遞給接受 Shape 元素集合的函數。 為了使程式碼可編譯,請使用 ? extends Shape 類型,這樣函數就可以接受 Shape 的任何繼承者的集合:

java
// Java
class Shape {}

class Rectangle extends Shape {}

public void doSthWithShapes(List<? extends Shape> shapes) {
/* If using just List<Shape>, the code won't compile when calling
this function with the List<Rectangle> as the argument as below */
}

public void main() {
    var rectangles = List.of(new Rectangle(), new Rectangle());
    doSthWithShapes(rectangles);
}

在 Kotlin 中,唯讀集合類型是 共變的。這表示如果 Rectangle 類別繼承自 Shape 類別, 您可以在任何需要 List<Shape> 類型的地方使用 List<Rectangle> 類型。 換句話說,集合類型具有與元素類型相同的子類型關係。Map 在值類型上是共變的,但在鍵類型上不是。 可變集合不是共變的 – 這將導致執行時失敗。

kotlin
// Kotlin
open class Shape(val name: String)

class Rectangle(private val rectangleName: String) : Shape(rectangleName)

fun doSthWithShapes(shapes: List<Shape>) {
    println("The shapes are: ${shapes.joinToString { it.name }}")
}

fun main() {
    val rectangles = listOf(Rectangle("rhombus"), Rectangle("parallelepiped"))
    doSthWithShapes(rectangles)
}

在此閱讀更多關於 集合類型 的內容。

範圍與進程

在 Kotlin 中,您可以使用 範圍 來建立區間。例如,Version(1, 11)..Version(1, 30) 包含從 1.111.30 的所有版本。 您可以使用 in 運算子來檢查您的版本是否在範圍內:Version(0, 9) in versionRange

在 Java 中,您需要手動檢查 Version 是否符合兩個邊界:

java
// Java
class Version implements Comparable<Version> {

    int major;
    int minor;

    Version(int major, int minor) {
        this.major = major;
        this.minor = minor;
    }

    @Override
    public int compareTo(Version o) {
        if (this.major != o.major) {
            return this.major - o.major;
        }
        return this.minor - o.minor;
    }
}

public void compareVersions() {
    var minVersion = new Version(1, 11);
    var maxVersion = new Version(1, 31);

   System.out.println(
           versionIsInRange(new Version(0, 9), minVersion, maxVersion));
   System.out.println(
           versionIsInRange(new Version(1, 20), minVersion, maxVersion));
}

public Boolean versionIsInRange(Version versionToCheck, Version minVersion,
                                Version maxVersion) {
    return versionToCheck.compareTo(minVersion) >= 0
            && versionToCheck.compareTo(maxVersion) <= 0;
}

在 Kotlin 中,您將範圍作為一個整體物件來操作。您無需創建兩個變數並將 Version 與它們進行比較:

kotlin
// Kotlin
class Version(val major: Int, val minor: Int): Comparable<Version> {
    override fun compareTo(other: Version): Int {
        if (this.major != other.major) {
            return this.major - other.major
        }
        return this.minor - other.minor
    }
}

fun main() {
    val versionRange = Version(1, 11)..Version(1, 30)

    println(Version(0, 9) in versionRange)
    println(Version(1, 20) in versionRange)
}

一旦您需要排除其中一個邊界,例如檢查版本是否大於或等於 (>=) 最小版本且小於 (<) 最大版本,這些包含性的範圍就無法派上用場。

多重條件比較

在 Java 中,若要根據多重條件比較物件,您可以使用 Comparator 介面中的 comparing()thenComparingX() 函數。 例如,根據姓名和年齡比較人物:

java
class Person implements Comparable<Person> {
    String name;
    int age;

    public String getName() {
        return name;
    }

    public int getAge() {
        return age;
    }

    Person(String name, int age) {
        this.name = name;
        this.age = age;
    }

    @Override
    public String toString() {
        return this.name + " " + age;
    }
}

public void comparePersons() {
    var persons = List.of(new Person("Jack", 35), new Person("David", 30),
            new Person("Jack", 25));
    System.out.println(persons.stream().sorted(Comparator
            .comparing(Person::getName)
            .thenComparingInt(Person::getAge)).collect(toList()));
}

在 Kotlin 中,您只需列舉要比較的欄位:

kotlin
data class Person(
    val name: String,
    val age: Int
)

fun main() {
    val persons = listOf(Person("Jack", 35), Person("David", 30),
        Person("Jack", 25))
    println(persons.sortedWith(compareBy(Person::name, Person::age)))
}

序列

在 Java 中,您可以這樣產生一連串數字:

java
// Java
int sum = IntStream.iterate(1, e -> e + 3)
    .limit(10).sum();
System.out.println(sum); // Prints 145

在 Kotlin 中,使用 序列。序列的多步驟處理在可能的情況下會延遲執行 – 實際的計算只會在請求整個處理鏈的結果時發生。

kotlin
fun main() {
    // Kotlin
    val sum = generateSequence(1) {
        it + 3
    }.take(10).sum()
    println(sum) // Prints 145
}

序列可以減少執行某些過濾操作所需的步驟數量。 請參閱 序列處理範例,其中顯示了 IterableSequence 之間的差異。

從列表中移除元素

在 Java 中,remove() 函數接受要移除元素的索引。

當移除一個整數元素時,請使用 Integer.valueOf() 函數作為 remove() 函數的參數:

java
// Java
public void remove() {
    var numbers = new ArrayList<>();
    numbers.add(1);
    numbers.add(2);
    numbers.add(3);
    numbers.add(1);
    numbers.remove(1); // This removes by index
    System.out.println(numbers); // [1, 3, 1]
    numbers.remove(Integer.valueOf(1));
    System.out.println(numbers); // [3, 1]
}

在 Kotlin 中,有兩種元素移除類型:透過 removeAt() 根據索引移除,以及透過 remove() 根據值移除。

kotlin
fun main() {
    // Kotlin
    val numbers = mutableListOf(1, 2, 3, 1)
    numbers.removeAt(0)
    println(numbers) // [2, 3, 1]
    numbers.remove(1)
    println(numbers) // [2, 3]
}

遍歷 Map

在 Java 中,您可以透過 forEach 遍歷 Map:

java
// Java
numbers.forEach((k,v) -> System.out.println("Key = " + k + ", Value = " + v));

在 Kotlin 中,使用 for 迴圈或 forEach(類似於 Java 的 forEach)來遍歷 Map:

kotlin
// Kotlin
for ((k, v) in numbers) {
    println("Key = $k, Value = $v")
}
// Or
numbers.forEach { (k, v) -> println("Key = $k, Value = $v") }

取得可能為空的集合中的第一個和最後一個項目

在 Java 中,您可以透過檢查集合的大小並使用索引來安全地取得第一個和最後一個項目:

java
// Java
var list = new ArrayList<>();
//...
if (list.size() > 0) {
    System.out.println(list.get(0));
    System.out.println(list.get(list.size() - 1));
}

您也可以對 Deque 及其繼承者使用 getFirst()getLast() 函數:

java
// Java
var deque = new ArrayDeque<>();
//...
if (deque.size() > 0) {
    System.out.println(deque.getFirst());
    System.out.println(deque.getLast());
}

在 Kotlin 中,有特殊的函數 firstOrNull()lastOrNull()。 使用 Elvis 運算子,您可以根據函數的結果立即執行進一步的操作。 例如,firstOrNull()

kotlin
// Kotlin
val emails = listOf<String>() // Might be empty
val theOldestEmail = emails.firstOrNull() ?: ""
val theFreshestEmail = emails.lastOrNull() ?: ""

從列表中建立集合 (Set)

在 Java 中,若要從 List 建立 Set,您可以使用 Set.copyOf 函數:

java
// Java
public void listToSet() {
    var sourceList = List.of(1, 2, 3, 1);
    var copySet = Set.copyOf(sourceList);
    System.out.println(copySet);
}

在 Kotlin 中,使用函數 toSet()

kotlin
fun main() {
    // Kotlin
    val sourceList = listOf(1, 2, 3, 1)
    val copySet = sourceList.toSet()
    println(copySet)
}

分組元素

在 Java 中,您可以使用 Collectors 函數 groupingBy() 來分組元素:

java
// Java
public void analyzeLogs() {
    var requests = List.of(
        new Request("https://kotlinlang.org/docs/home.html", 200),
        new Request("https://kotlinlang.org/docs/home.html", 400),
        new Request("https://kotlinlang.org/docs/comparison-to-java.html", 200)
    );
    var urlsAndRequests = requests.stream().collect(
            Collectors.groupingBy(Request::getUrl));
    System.out.println(urlsAndRequests);
}

在 Kotlin 中,使用函數 groupBy()

kotlin
data class Request(
    val url: String,
    val responseCode: Int
)

fun main() {
    // Kotlin
    val requests = listOf(
        Request("https://kotlinlang.org/docs/home.html", 200),
        Request("https://kotlinlang.org/docs/home.html", 400),
        Request("https://kotlinlang.org/docs/comparison-to-java.html", 200)
    )
    println(requests.groupBy(Request::url))
}

過濾元素

在 Java 中,若要從集合中過濾元素,您需要使用 Stream API。 Stream API 具有 中間操作 (intermediate operation) 和 終端操作 (terminal operation)。filter() 是一個中間操作,它返回一個串流。 若要接收集合作為輸出,您需要使用終端操作,例如 collect()。 例如,只保留鍵以 1 結尾且值大於 10 的那些鍵值對:

java
// Java
public void filterEndsWith() {
    var numbers = Map.of("key1", 1, "key2", 2, "key3", 3, "key11", 11);
    var filteredNumbers = numbers.entrySet().stream()
        .filter(entry -> entry.getKey().endsWith("1") && entry.getValue() > 10)
        .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
    System.out.println(filteredNumbers);
}

在 Kotlin 中,過濾是內建在集合中的,並且 filter() 返回與被過濾的集合相同的集合類型。 因此,您只需編寫 filter() 及其謂詞 (predicate):

kotlin
fun main() {
    // Kotlin
    val numbers = mapOf("key1" to 1, "key2" to 2, "key3" to 3, "key11" to 11)
    val filteredNumbers = numbers.filter { (key, value) -> key.endsWith("1") && value > 10 }
    println(filteredNumbers)
}

在此深入了解 過濾 Map 的內容。

按類型過濾元素

在 Java 中,若要按類型過濾元素並對其執行操作,您需要使用 instanceof 運算子檢查其類型,然後進行類型轉換:

java
// Java
public void objectIsInstance() {
    var numbers = new ArrayList<>();
    numbers.add(null);
    numbers.add(1);
    numbers.add("two");
    numbers.add(3.0);
    numbers.add("four");
    System.out.println("All String elements in upper case:");
    numbers.stream().filter(it -> it instanceof String)
        .forEach( it -> System.out.println(((String) it).toUpperCase()));
}

在 Kotlin 中,您只需在集合上呼叫 filterIsInstance<NEEDED_TYPE>(),並且類型轉換會透過 智慧型轉換 完成:

kotlin
// Kotlin
fun main() {
    // Kotlin
    val numbers = listOf(null, 1, "two", 3.0, "four")
    println("All String elements in upper case:")
    numbers.filterIsInstance<String>().forEach {
        println(it.uppercase())
    }
}

測試謂詞

有些任務要求您檢查所有、沒有或任何元素是否滿足特定條件。 在 Java 中,您可以透過 Stream API 函數 allMatch()noneMatch()anyMatch() 進行所有這些檢查:

java
// Java
public void testPredicates() {
    var numbers = List.of("one", "two", "three", "four");
    System.out.println(numbers.stream().noneMatch(it -> it.endsWith("e"))); // false
    System.out.println(numbers.stream().anyMatch(it -> it.endsWith("e"))); // true
    System.out.println(numbers.stream().allMatch(it -> it.endsWith("e"))); // false
}

在 Kotlin 中,擴充函數 none()any()all() 可用於每個 Iterable 物件:

kotlin
fun main() {
// Kotlin
    val numbers = listOf("one", "two", "three", "four")
    println(numbers.none { it.endsWith("e") })
    println(numbers.any { it.endsWith("e") })
    println(numbers.all { it.endsWith("e") })
}

在此深入了解 測試謂詞

集合轉換操作

壓縮元素

在 Java 中,您可以透過同時迭代兩個集合來從相同位置的元素建立配對:

java
// Java
public void zip() {
    var colors = List.of("red", "brown");
    var animals = List.of("fox", "bear", "wolf");

    for (int i = 0; i < Math.min(colors.size(), animals.size()); i++) {
        String animal = animals.get(i);
        System.out.println("The " + animal.substring(0, 1).toUpperCase()
               + animal.substring(1) + " is " + colors.get(i));
   }
}

如果您想做比單純將元素配對列印到輸出更複雜的事情,可以使用 Records。 在上面的範例中,record 將會是 record AnimalDescription(String animal, String color) {}

在 Kotlin 中,使用 zip() 函數來做相同的事情:

kotlin
fun main() {
    // Kotlin
    val colors = listOf("red", "brown")
    val animals = listOf("fox", "bear", "wolf")

    println(colors.zip(animals) { color, animal ->
        "The ${animal.replaceFirstChar { it.uppercase() }} is $color" })
}

zip() 返回 Pair 物件的 List。

NOTE

如果集合的大小不同,zip() 的結果將以較小的大小為準。較大集合中的最後幾個元素將不會包含在結果中。

關聯元素

在 Java 中,您可以使用 Stream API 將元素與特性關聯起來:

java
// Java
public void associate() {
    var numbers = List.of("one", "two", "three", "four");
    var wordAndLength = numbers.stream()
        .collect(toMap(number -> number, String::length));
    System.out.println(wordAndLength);
}

在 Kotlin 中,使用 associate() 函數:

kotlin
fun main() {
    // Kotlin
    val numbers = listOf("one", "two", "three", "four")
    println(numbers.associateWith { it.length })
}

下一步是什麼?

如果您有喜歡的慣用法,我們邀請您透過發送 Pull Request 來分享它。