Cắt danh sách bằng take và drop

Thuộc phần Kiến thức nền tảng trong hành trình Kotlin của Coddy. Bài học 80 trên 93.

Kotlin cung cấp hai phương thức đơn giản hơn để trích xuất một phần của danh sách khi bạn chỉ cần các phần tử ở đầu hoặc cuối: take() và drop().

Phương thức take() trả về một danh sách mới chứa n phần tử đầu tiên:

val numbers = listOf(10, 20, 30, 40, 50)
val firstThree = numbers.take(3)
println(firstThree)  // [10, 20, 30]

Phương thức drop() làm điều ngược lại. Nó bỏ qua n phần tử đầu tiên và trả về phần còn lại:

val numbers = listOf(10, 20, 30, 40, 50)
val afterTwo = numbers.drop(2)
println(afterTwo)  // [30, 40, 50]

Khác với subList(), các phương thức này trả về bản sao độc lập thay vì chế độ xem. Chúng cũng dễ đọc hơn khi bạn chỉ muốn lấy “một vài phần tử đầu tiên” hoặc “mọi phần tử trừ một vài phần tử đầu tiên”.

Bạn có thể kết hợp cả hai phương thức để trích xuất phần ở giữa:

val letters = listOf("A", "B", "C", "D", "E")
val middle = letters.drop(1).take(3)
println(middle)  // [B, C, D]

Nếu bạn yêu cầu nhiều phần tử hơn số phần tử hiện có, các phương thức này chỉ trả về những phần tử sẵn có mà không gây ra lỗi:

val short = listOf(1, 2)
println(short.take(10))  // [1, 2]
println(short.drop(10))  // []

Thử thách

Trung bình

Bạn sẽ nhận được một danh sách số nguyên và hai số: số phần tử cần bỏ từ đầu danh sách và số phần tử cần lấy sau khi bỏ.

Dùng drop() và take() cùng nhau để trích xuất một phần ở giữa danh sách, sau đó in ra các phần tử đã trích xuất.

Định dạng đầu vào:

Dòng đầu tiên: một số nguyên n biểu thị số phần tử
n dòng tiếp theo: các số nguyên cần thêm vào danh sách
Dòng tiếp theo: số phần tử cần bỏ từ đầu danh sách
Dòng cuối cùng: số phần tử cần lấy sau khi bỏ

Định dạng đầu ra:

In ra danh sách kết quả theo định dạng danh sách mặc định của Kotlin.

Ví dụ:

Nếu danh sách là [10, 20, 30, 40, 50, 60], số phần tử cần bỏ là 2 và số phần tử cần lấy là 3, đầu ra sẽ là:

[30, 40, 50]

Tự làm thử

fun main() {
    // Đọc số lượng phần tử
    val n = readLine()!!.toInt()
    
    // Đọc các phần tử của danh sách
    val list = mutableListOf<Int>()
    for (i in 1..n) {
        list.add(readLine()!!.toInt())
    }
    
    // Đọc các giá trị cho drop và take
    val dropCount = readLine()!!.toInt()
    val takeCount = readLine()!!.toInt()
    
    // TODO: Viết code của bạn bên dưới
    // Dùng drop() và take() để lấy phần ở giữa
    
    
    // In ra kết quả
    println(result)
}