【动态规划案例】详细介绍投资问题

目录

投资问题描述

动态规划分析

案例分析

第一阶段

第二阶段

第三阶段

第四阶段

代码实现

---

投资问题描述

现有万元资金，个项目投资，假设：为分配给第i个项目的资金；为第i个项目投资得到的收益（万元）。问：如何投资使得总效益最大？

动态规划分析

根据动态规划的一般性步骤，首先建立目标函数和约束条件，根据上面假设得到：

目标函数：;

约束条件：;

假设为万元钱投资给前个项目获得的最大收益，用动态规划思路求解，问题就转成求。

当时，也就是说只投一个项目时最大收益为：

；

当时，即至少投资两个项目以上时，设（）为分配给第个项目的资金，此时还剩下的资金可分配给前个项目，则获得的最大收益为：

根据上面的分析，可以得到转移方程：

案例分析

以上得到的转移方程分析比较理论话，下面用一个例子来做一下案例分析，比较直观和容易理解。

设某机构有60万元准备投资给4个项目，每个项目获得的收益和投资金额如下所示：

 根据问题要求，求出。

第一阶段

当时，只投资一个项目，那么显然得到，当投资金额时分别计算得到如下结果： 

最优策略展示表格如下：

投资金额x	0	10	20	30	40	50	60
最优策略	（0)	(10)	(20)	(30)	(40)	(50)	(60)

此时投资金额60万元的最优投资策略为:，即(60)。

第二阶段

当时，此时需要考虑前两个项目投资情况下获得的最大收益，即，下面公式展现了当投资金额时求最优策略过程：

 ，投资金额的最优投资策略为(40,20)。同理，

，投资金额的最优投资策略为(30,20)。

，投资金额的最优投资策略为(20,20)。

依此类推，可以求得。最终展示如下：

对应的最优投资策略如下表格所示：

投资金额x	0	10	20	30	40	50	60
最优策略	（0,0)	(10,0) (0,10)	(20,0)	(20,10)	(20,20)	(30,20)	(40,20)

第三阶段

当时，此时就需要考虑前三个项目都投资情况下获得的最大收益，即，当投资金额时求最优策略：

，投资金额的最优投资策略为(20,10,30)。

依此类推，可以求得，最终结果如下： 

对应的最优投资策略如下表格所示：

投资金额x	0	10	20	30	40	50	60
最优策略	（0,0,0)	(0,0,10)	(0,0,20)	(0,0,30)	(20,0,20)	(20,0,30)	(20,10,20)

第四阶段

当时，此时4个项目获得的最大收益，即，当投资金额时求最优策略：

 

当投资金额为60时，投资4个项目最大收益为160万元，最优投资策略为（20,0,30,10）。

代码实现

var M = 60; // 投资总金额
var N = 4; // 投资项目数量
var LR = [ // 投资4个项目的收益
    [0, 20, 50, 65, 80, 85, 85],
    [0, 20, 40, 50, 55, 60, 65],
    [0, 25, 60, 85, 100, 110, 115],
    [0, 25, 40, 50, 60, 65, 70]
];
function dptz() {
    var F = [];
    var m = LR[0].length;
    var n = N;
    for (var i = 1; i <= n; i++) { // 按照项目循环
        F[i - 1] = [];
        for (var j = 0; j < m; j++) { // 按照总投资金额
            if (i == 1) {
                F[i - 1][j] = LR[i - 1][j];
            } else {
                var tmp = 0;
                for (var k = 0; k <= j; k++) {
                    tmp = Math.max(tmp, F[i - 2][k] + LR[i - 1][j - k]);
                }
                F[i - 1][j] = tmp;
            }
        }
    }
    console.log(F[N-1][len-1]); // 160
}
dptz();

通过代码可以看出有三个for循环，所以时间复杂度为。求出最大收益没什么难度，关键是如何输出最大收益策略呢？代码优化如下：

var M = 60; // 投资总金额
var N = 4; // 投资项目数量
var LR = [ // 投资4个项目的收益
    [0, 20, 50, 65, 80, 85, 85],
    [0, 20, 40, 50, 55, 60, 65],
    [0, 25, 60, 85, 100, 110, 115],
    [0, 25, 40, 50, 60, 65, 70]
];
function dptz() {
    var F = [];
    var mem = []; // 保存当前总金额下，投资项目数的最优策略
    var m = LR[0].length;
    var n = N;
    for (var i = 1; i <= N; i++) {
        F[i - 1] = [];
        mem[i - 1] = [];
        for (var j = 0; j < m; j++) {
            if (i == 1) {
                F[i - 1][j] = LR[i - 1][j];
                mem[i - 1][j] = [j * 10];
            } else {
                var tmp = 0;
                var f = 0; // 当前项目分配的金额
                var TF = 0; // 剩余金额投资最优解的下标
                for (var k = 0; k <= j; k++) {
                    // tmp = Math.max(tmp, F[i - 2][k] + LR[i - 1][j - k]);
                    // var Fcur = k * 10;
                    // var fcur = M - Fcur;
                    if (F[i - 2][k] + LR[i - 1][j - k] >= tmp) {
                        tmp = F[i - 2][k] + LR[i - 1][j - k];
                        f = (j - k) * 10; // 当前项目分配的金额
                        TF = k; // 保存剩余金额k*10（通过j*10-f得到）时的最优解下标
                    }
                }
                F[i - 1][j] = tmp;
                mem[i - 1][j] = [];
                mem[i - 2][TF].forEach(el => { // 将剩余金额TF*10下的最优解赋值给当前项目下的最优解
                    mem[i - 1][j].push(el);
                })
                // mem[i - 1][j] = [...mem[i - 2][TF], f];
                // 将当前最优解保存起来，f就是当前项目i的投资金额数。
                mem[i - 1][j].push(f);
            }
        }
    }
    console.log(F[N - 1][m - 1]);
    console.log(mem[N - 1][m - 1])
}
dptz();

运行得出了最大收益和最优策略方案：