小米笔试题（动态规划）

风口之下，猪都能飞。当今中国股市牛市，真可谓“错过等七年”。 给你一个回顾历史的机会，已知一支股票连续n天的价格走势，以长度为n的整数数组表示，数组中第i个元素（prices[i]）代表该股票第i天的股价。 假设你一开始没有股票，但有至多两次买入1股而后卖出1股的机会，并且买入前一定要先保证手上没有股票。若两次交易机会都放弃，收益为0。 设计算法，计算你能获得的最大收益。 输入数值范围：2<=n<=100,0<=prices[i]<=100 

输入例子:

3,8,5,1,7,8

输出例子:

12

 1 #include <iostream>
 2 #include <vector>
 3 
 4 using namespace std;
 5 
 6 int calculateMax(vector<int> prices);
 7 
 8 int main()
 9 {
10     int b[6] = {3,8,5,1,7,8};
11     vector<int> a(b, b + 6);
12 
13     cout << calculateMax(a) << endl;
14     return 0;
15 }
16 
17 int calculateMax(vector<int> prices) {
18 
19     int length = prices.size();
20     int maxM = 0;
21     int i,j,counts;
22     //dynamic programming is create a form
23     //just like sub[][]=0
24     vector<vector<int> >suboptimization(length ,vector<int>(3, 0));
25 
26     for(i = 1; i < length; i ++){
27 
28             for(counts = 1; counts < 3; counts ++){
29                     maxM = suboptimization[i - 1][counts];
30 
31                 for(j = 0; j < i; j++ ){
32                   maxM = max(maxM, suboptimization[j][counts - 1] + prices[i] - prices[j]);
33                 }
34                 suboptimization[i][counts] = maxM;
35             }
36 
37     }
38 
39 
40     return suboptimization[length - 1][2];
41 }

用动态规划做的具体不解释。