小米笔试题（动态规划）

风口之下，猪都能飞。当今中国股市牛市，真可谓“错过等七年”。 给你一个回顾历史的机会，已知一支股票连续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>
#include <vector>

using namespace std;

int calculateMax(vector<int> prices);

int main()
{
    int b[6] = {3,8,5,1,7,8};
    vector<int> a(b, b + 6);

    cout << calculateMax(a) << endl;
    return 0;
}

int calculateMax(vector<int> prices) {

    int length = prices.size();
    int maxM = 0;
    int i,j,counts;
    //dynamic programming is create a form
    //just like sub[][]=0
    vector<vector<int> >suboptimization(length ,vector<int>(3, 0));

    for(i = 1; i < length; i ++){

            for(counts = 1; counts < 3; counts ++){
                    maxM = suboptimization[i - 1][counts];

                for(j = 0; j < i; j++ ){
                  maxM = max(maxM, suboptimization[j][counts - 1] + prices[i] - prices[j]);
                }
                suboptimization[i][counts] = maxM;
            }

    }


    return suboptimization[length - 1][2];
}

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