面试题：Rust枚举类型与模式匹配基础应用

Question

在Rust中，定义一个枚举类型`TrafficLight`，包含`Red`、`Yellow`、`Green`三个变体。然后编写一个函数`match_traffic_light`，该函数接收一个`TrafficLight`类型的参数，通过模式匹配返回相应的驾驶操作建议，例如红灯返回`"Stop"`，黄灯返回`"Prepare to stop"`，绿灯返回`"Go"`。

Answer 1

// 定义枚举类型TrafficLight
enum TrafficLight {
    Red,
    Yellow,
    Green,
}

// 定义函数match_traffic_light
fn match_traffic_light(light: TrafficLight) -> &'static str {
    match light {
        TrafficLight::Red => "Stop",
        TrafficLight::Yellow => "Prepare to stop",
        TrafficLight::Green => "Go",
    }
}