【华为机试真题 Python】能量消耗

目录

题目描述

参考示例 

参考代码

机试介绍

写在最后

---

题目描述

给出一批用户，每个用户有 3 种选择 A\B\C，但是价值不同，相临的用户不能选同一个，求
出所有用户选择后总价值最大。

参考示例 

输入：
3
30 8 4
50 20 9
11 7 6

输出：
65

说明：因为选 8 50 7

参考代码

def calc():
    global n, nums, dfs
    n = int(input())
    nums = [list(map(int, input().split())) for _ in range(n)]

    def dfs(idx, com, val):
        # idx 用户id
        # com 三个策略
        # val 资源消耗
        if idx == n - 1:
            return val
        idx += 1
        return max([dfs(idx, c, nums[idx][c]) + val for c in [0, 1, 2] if c != com])

    dp = []
    for c in [0, 1, 2]:
        dp.append(dfs(0, c, nums[0][c]))
    print