投资问题（动态规划）

问题

设m元钱，n项投资，函数fi(x)表示将x元投入到第i项项目中所产生的效益，i=1,2,3…n

问：如何分配这m元钱，使得投资的总效益最高?

解析

假设分配给第 i 个项目的钱数是 xi，问题描述为：
目标函数：max{f1(x1)+f2(x2)+…+fn(xn)}
约束条件：x1+x2+…+xn=m,xi∈N；

设Fk(x)表示x元投给前k个项目的最大效益，k=1,2,…,n，x=1,2,…,m
递推方程：Fk(x)=max{fk(xk)+Fk-1(x-xk)}(0≤xk≤x)，k=2,3,…,n
边界条件：F1(x)=f1(x),Fk(0)=0,k=1,2,…,n

*说明：第k步，前后共分配x万元，

• 分配给第k个项目xk；
• x-xk万元，分配给前k-1个项目；

设计

For k=1,2,…,n //第k个项目
 For x=0,1,2,…,m //k个项目共分配x元
  For xk=0,1,2,…,x //第k个项目分配xk
   Fk(x)=max{fi(xk)+Fk-1(x-xk)}//递推公式

核心代码

	for (i = 1; i < n; i++) {//项目编号，从1开始
		for (j = 0; j < m + 1; j++) {//钱数,从0开始
			for (tmp_m = 0; tmp_m <= j; tmp_m++) {
				//递推公式
				tmp_F = F[i - 1][j - tmp_m] + f[i][tmp_m];
				//取最大值
				if (tmp_F > F[i][j]) {
					F[i][j] = tmp_F;
				}	
			}
		}
	}

分析

xk∈[0,…,x],共x+1项，fk(xk)+Fk-1(x-xk)有x+1项，因此有x+1次加，x项比较出最大值。
For k=1,2,…,n
For x=1,2,…,m
加法
比较

复杂度 W(n,m)=O(nm2)

github源码地址