问题
设m元钱,n项投资,函数f[i][x]表示将x元投入第i项项目所产生的效益,i=1,2,…,n.问:如何分配这m元钱,使得投资的总效益最高。
解析
对于动态规划的一般问题,首先要列出递推方程:
设F[k][x]表示x万元投给前k个项目的最大效益,k=1,2,…,n,x=1,2,…,m。
分析可得:
递推方程:F[k][x]=max{f[k][j]+F[k-1][x-j]}, 0<=j<=x , k=1,2,…,n
边界条件:F[1][x]=f[1][x] , x=0,1,…,m ; F[k][0]=0 , k=1,2,…,n
说明:第k步,共分配x万元
- 其中分配给第k个项目为j万元;
- 分配给前k-1的项目为x-j万元。
满足优化原则:
优化原则:一个最优决策序列的任何子序列本身一定是相对于子序列的初始和结束状态的最优决策序列。
已知这个序列L1时最优决策序列,那么这个序列任何子序列本身一定是相对于子序列的初始和结束状态的最优决策序列。
设计
For k=1,2,3,…,n //第k个项目
For x=1,2,3…,m //k个项目共分配x万元
For xk=0,1,2,3,…,x //第k个项目分配xk
分析
k∈[0,j],共j+1项,f[i][k]+F[i-1][j-k]有j+1项,因此有j+1次加法,j+1项比较出最大值。
for i = 1 ~ n
for j = 0 ~ m
for k= 0 ~ j
加法与比较的复杂度均为(n-1)m(m+3)/2.
因此该算法时间复杂度为O(n·m^2)
源码
#include<bits/stdc++.h>
using namespace std;
const int maxn=1010;
int f[maxn][maxn],val[maxn][maxn],x[maxn][maxn];
int n,m;
int main(){
scanf("%d %d",&n,&m);
for(int i=0;i<=m;++i){
for(int j=1;j<=n;++j){
scanf("%d",&val[j][i]);
}
}
for(int i=1;i<=n;++i){
for(int j=0;j<=m;++j){
for(int k=0;k<=j;++k){
if(f[i-1][j-k]+val[i][k]>f[i][j]){
f[i][j]=f[i-1][j-k]+val[i][k];
x[i][j]=k;
}
}
}
}
for(int i=1;i<=m;++i){
for(int j=1;j<=n;++j){
printf("%d %d%c",f[j][i],x[j][i],"\n "[j!=n]);
}
}
printf("%d\n",f[n][m]);
}
/*
4 5
0 0 0 0
11 0 2 20
12 5 10 21
13 10 30 22
14 15 32 23
15 20
github:https://github.com/Geedhayb/Geed/blob/master/DPinvest.cpp