编辑代码

def main():
    # 提示用户输入车辆总功率
    total_power = input("请输入车辆总功率： ")
    # 显示用户输入的车辆总功率
    print(f"您输入的车辆总功率是： {total_power} kW")
    
    # 提示用户输入发动机功率
    engine_power = input("请输入发动机功率： ")
    # 显示用户输入的发动机功率
    print(f"您输入的发动机功率是： {engine_power} kW")
    
    # 提示用户输入电机功率
    motor_power = input("请输入电机功率： ")
    # 显示用户输入的电机功率
    print(f"您输入的电机功率是： {motor_power} kW")
    
    # 提示用户输入电池组模块成本价格
    battery_cost = input("请输入电池组模块成本价格： ")
    # 显示用户输入的电池组模块成本价格
    print(f"您输入的电池组模块成本价格是： {battery_cost} 元")
    
    # 提示用户输入车辆是否为增程式混合动力汽车
    is_extended_range = input("车辆是否为增程式混合动力汽车 (是/否)： ")
    # 显示用户输入的车辆是否为增程式混合动力汽车
    print(f"车辆是否为增程式混合动力汽车： {is_extended_range}")
    
    # 根据用户输入的是否为增程式混合动力汽车，计算成本减少价格
    if is_extended_range.lower() == "是":
        # 计算成本减少价格
        cost_reduction = 588.60 + (int(engine_power) * 19.53) + (((int(motor_power) - int(engine_power)) / int(motor_power)) * int(battery_cost))
        print(f"成本减少价格为： {cost_reduction} 元")
    elif is_extended_range.lower() == "否":
        print("成本减少价格为588.60元")
    else:
        print("输入无效，请输入'是'或'否'来回答是否为增程式混合动力汽车")

if __name__ == "__main__":
    main()