华为OD机试统一考试 - 虚拟理财游戏

题目描述

在一款虚拟游戏中生活，你必须进行投资以增强在虚拟游戏中的资产以免被淘汰出局。

现有一家Bank，它提供有若干理财产品 m 个，风险及投资回报不同，你有 N（元）进行投资，能接收的总风险值为X。

你要在可接受范围内选择最优的投资方式获得最大回报。

备注：

• 在虚拟游戏中，每项投资风险值相加为总风险值；
• 在虚拟游戏中，最多只能投资2个理财产品；
• 在虚拟游戏中，最小单位为整数，不能拆分为小数；
• 投资额*回报率=投资回报

输入描述

第一行：

• 产品数（取值范围[1,20]）
• 总投资额（整数，取值范围[1, 10000]）
• 可接受的总风险（整数，取值范围[1,200]）

第二行：产品投资回报率序列，输入为整数，取值范围[1,60]

第三行：产品风险值序列，输入为整数，取值范围[1, 100]

第四行：最大投资额度序列，输入为整数，取值范围[1, 10000]

输出描述

每个产品的投资额序列

用例

题目解析

第一眼看这题有点像二维费用背包，但是本题备注中有一个关键限制：

在虚拟游戏中，最多只能投资2个理财产品；

那么本题其实就变成了：m个理财产品中选1个或2个，所选产品风险值之和 ≤ x，投资额之和 ≤ n，并且最终所选产品的投资回报之和最大。

由于 m 的数量级不大，取值范围[1,20]，因此可以考虑暴力破解。

import Foundation

func ODTest_7() {
    print("第一行：")
    print("产品数（取值范围[1,20]）")
    print("总投资额（整数，取值范围[1, 10000]）")
    print("可接受的总风险（整数，取值范围[1,200]）")
    let tmp = readLine()?.split(separator: " ").map { Int($0) ?? 0 } ?? []

    let m = tmp[0] // 产品数
    let n = tmp[1] // 总投资
    let x = tmp[2] // 总风险

    print("第二行：产品投资回报率序列，输入为整数，取值范围[1,60]")
    // 产品回报率序列
    let back = readLine()?.split(separator: " ").map { Int($0) ?? 0 } ?? []

    print("第三行：产品风险值序列，输入为整数，取值范围[1, 100]")
    // 产品风险值序列
    let risk = readLine()?.split(separator: " ").map { Int($0) ?? 0 } ?? []

    print("第四行：最大投资额度序列，输入为整数，取值范围[1, 10000]")
    // 产品投资额序列
    let invest = readLine()?.split(separator: " ").map { Int($0) ?? 0 } ?? []