算法提高之股票买卖 V

原创于 2024-05-06 21:36:17 发布 · 327 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

该C++代码展示了如何运用状态机来计算股票买卖策略中的最优状态转换，涉及空仓、持仓和冻结期。通过动态规划求解在给定价格序列下最大收益。

算法提高之股票买卖 V

核心思想：状态机
- 一共有三种情况：空仓，持仓，冻结期
- f[i,j]表示第i天的状态j
- 状态计算：如下

  #include <iostream>
  #include <cstring>
  #include <algorithm>
  
  using namespace std;
  const int N = 100010;
  
  int f[N][3];
  int w[N];
  int n;
  
  int main()
  {
      cin>>n;
      for(int i=0;i<n;i++) cin>>w[i];
      
      f[0][0] = 0;
      f[0][1] = -w[0];
      for(int i=1;i<n;i++)
      {
          //当前空仓 = 前一次空仓/前一次冻结期
          f[i][0] = max(f[i-1][0] , f[i-1][2]);
          //当前持仓 = 前一次持仓/前一次空仓买入
          f[i][1] = max(f[i-1][1] , f[i-1][0] - w[i]);
          //当前冻结期 = 前一次持仓卖出
          f[i][2] = f[i-1][1] + w[i];
      }
      cout<<max(f[n-1][0],f[n-1][2])<<endl;
      return 0;
  }