Struct làm tham số cho hàm

Thuộc phần Logic và luồng chương trình trong hành trình Rust của Coddy. Bài học 24 trên 66.

Bây giờ bạn đã biết cách tạo và chỉnh sửa struct, hãy cùng tìm hiểu cách dùng chúng với hàm (function). Cũng như mọi kiểu dữ liệu khác trong Rust, bạn có thể truyền các instance của struct vào hàm và trả chúng về từ hàm.

Để truyền một struct vào hàm, bạn chỉ cần thêm nó làm tham số trong phần định nghĩa hàm:

fn calculate_area(rect: Rectangle) -> i32 {
    rect.width * rect.height
}

Hàm (function) này nhận một cấu trúc Rectangle làm tham số và trả về một i32. Bên trong hàm, bạn có thể truy cập các trường của cấu trúc bằng ký hiệu dấu chấm, giống như ở mọi nơi khác.

Bạn cũng có thể trả về các struct từ hàm bằng cách chỉ định kiểu struct làm kiểu trả về:

fn create_square(size: i32) -> Rectangle {
    Rectangle {
        width: size,
        height: size,
    }
}

Dùng struct cùng với hàm giúp code của bạn có tổ chức và ý nghĩa hơn. Thay vì truyền nhiều tham số riêng lẻ, bạn có thể nhóm các dữ liệu liên quan lại với nhau và truyền chúng dưới dạng một đơn vị duy nhất, được xác định rõ ràng.

Thử thách

Dễ

Bạn sẽ nhận được bốn đầu vào: chiều rộng của hình chữ nhật (dưới dạng số nguyên), chiều cao của hình chữ nhật (dưới dạng số nguyên), độ dài cạnh của hình vuông (dưới dạng số nguyên) và một số lựa chọn (1 hoặc 2). Định nghĩa một struct Rectangle có hai trường: width: i32 và height: i32. Tạo hai hàm: một hàm nhận một Rectangle và tính diện tích của nó, hàm còn lại nhận độ dài cạnh và trả về một Rectangle biểu thị một hình vuông. Dựa trên đầu vào lựa chọn, hãy tính và in ra diện tích tương ứng.

Yêu cầu:

Định nghĩa một struct Rectangle có các trường: width: i32 và height: i32
Tạo hàm calculate_area nhận một Rectangle làm tham số và trả về một i32 (diện tích)
Tạo hàm create_square nhận một i32 (độ dài cạnh) làm tham số và trả về một Rectangle có chiều rộng và chiều cao bằng nhau
Đọc đầu vào thứ nhất và chuyển đổi thành i32 để làm chiều rộng của hình chữ nhật
Đọc đầu vào thứ hai và chuyển đổi thành i32 để làm chiều cao của hình chữ nhật
Tạo một instance Rectangle với các kích thước này
Đọc đầu vào thứ ba và chuyển đổi thành i32 để làm độ dài cạnh của hình vuông
Đọc đầu vào thứ tư và chuyển đổi thành i32 để làm số lựa chọn (1 hoặc 2)
Nếu lựa chọn là 1: tính và in ra diện tích hình chữ nhật
Nếu lựa chọn là 2: tạo hình vuông bằng create_square, sau đó tính và in ra diện tích của nó

Đầu vào:

Dòng đầu tiên: chiều rộng hình chữ nhật dưới dạng số nguyên (ví dụ: 8)
Dòng thứ hai: chiều cao hình chữ nhật dưới dạng số nguyên (ví dụ: 5)
Dòng thứ ba: độ dài cạnh hình vuông dưới dạng số nguyên (ví dụ: 6)
Dòng thứ tư: lựa chọn dưới dạng số nguyên, là 1 hoặc 2 (ví dụ: 1)

Đầu ra:

Nếu lựa chọn là 1: Rectangle area: [area]
Nếu lựa chọn là 2: Square area: [area]

Tự làm thử

use std::io;

// TODO: Định nghĩa struct Rectangle tại đây


// TODO: Định nghĩa hàm (function) calculate_area tại đây


// TODO: Định nghĩa hàm create_square tại đây


fn main() {
    // Đọc chiều rộng hình chữ nhật
    let mut width_input = String::new();
    io::stdin().read_line(&mut width_input).expect("Failed to read line");
    let width: i32 = width_input.trim().parse().expect("Invalid input");
    
    // Đọc chiều cao hình chữ nhật
    let mut height_input = String::new();
    io::stdin().read_line(&mut height_input).expect("Failed to read line");
    let height: i32 = height_input.trim().parse().expect("Invalid input");
    
    // Đọc độ dài cạnh hình vuông
    let mut side_input = String::new();
    io::stdin().read_line(&mut side_input).expect("Failed to read line");
    let side: i32 = side_input.trim().parse().expect("Invalid input");
    
    // Đọc lựa chọn
    let mut choice_input = String::new();
    io::stdin().read_line(&mut choice_input).expect("Failed to read line");
    let choice: i32 = choice_input.trim().parse().expect("Invalid input");
    
    // TODO: Viết code bên dưới
    // Tạo một instance Rectangle, kiểm tra lựa chọn và tính diện tích tương ứng
    
    
    // In ra kết quả theo lựa chọn
    // Dùng: println!("Rectangle area: {}", area); hoặc println!("Square area: {}", area);
}