import java.util.*; public class Solution { public double get_max_profit (double M, int n, double[] historyPrices, int k) { double[][][] dp=new double[n+1][k+1][2]; //dp[i][j][0]--> 不持股,当前资金 dp[i][j][1]-->持股,股票数 for(int i=0;i<=k;i++) dp[0][i][0]=M; /* dp[i][j][hold]-> 第i天交易了j次的持股数/资金数 hold=0 资金数。今天不持股,可能一直不持股或者卖了股票 dp[i][j][0]=Math.max(dp[i-1][j][0],dp[i-1][j+1][1]*historyPrices[i-1]) hold=1 股票数。今天持股,可能一直持股或者买了股票 dp[i][j][1]=Math.max(dp[i-1][j][1],dp[i-1][j][0]/historyPrices[i-1]) */ for(int i=1;i<=n;i++){ for(int j=0;j<=k;j++){ if(j<k) dp[i][j][0]=Math.max(dp[i-1][j+1][1]*historyPrices[i-1],dp[i-1][j][0]); else dp[i][j][0]=dp[i-1][j][0]; dp[i][j][1]=Math.max(dp[i-1][j][0]/historyPrices[i-1],dp[i-1][j][1]); } } return dp[n][0][0]-M; } }
点赞 评论

相关推荐

牛客网
牛客网在线编程
牛客网题解
牛客企业服务