小米笔试题(动态规划)

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

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

 

posted @ 2015-06-13 15:09  Friday92  阅读(309)  评论(0编辑  收藏  举报