请按照下方输入/输出格式实现 solve。
len(values) = len(rewards) + 1 0 <= gamma, lambda <= 1。
示例 1
输入
{"rewards":[1,0,2],"values":[0,0,1,0],"gamma":1,"lambda":1}预期输出
[3.0,2.0,1.0]
示例 2
输入
{"rewards":[0,1],"values":[0,0,0],"gamma":0.9,"lambda":0.5}预期输出
[0.45,1.0]
运行会检查可见示例，提交会检查完整测试集。
登录并提交后，你的提交记录会显示在这里。