2015小米暑期实习笔试题_风口的猪-中国牛市(dp)

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

3,8,5,1,7,8

输出例子:

12

代码：

#include <iostream>
#include <cmath>
#include <vector>using namespace std;vector<int> prices;int calculateMax(vector<int> prices)
{int i,nmin,nmax,len,ans;len=prices.size();int *lhs=new int[len];int *rhs=new int[len];lhs[0]=0;nmin=prices[0];for(i=1;i<len;i++){lhs[i]=max(lhs[i-1],prices[i]-nmin);if(prices[i]<nmin)nmin=prices[i];
//		cout<<"lhs "<<i<<" "<<lhs[i]<<endl;}rhs[len-1]=0;nmax=prices[len-1];for(i=len-2;i>=0;i--){rhs[i]=max(rhs[i+1],nmax-prices[i]);if(prices[i]>nmax)nmax=prices[i];
//		cout<<"rhs "<<i<<" "<<rhs[i]<<endl;}ans=0;for(i=0;i<len;i++)if((lhs[i]+rhs[i])>ans)ans=lhs[i]+rhs[i];return ans;
}int main()
{int m,len;while(cin>>m)prices.push_back(m);cout<<calculateMax(prices)<<endl;return 0;
}

下面是去掉注释的测试截图，可以直观一点

语言表达能力太差啦，以后发文会注意写注释，亲们多多提建议哈~

#include <iostream>
#include <cmath>
#include <vector>using namespace std;vector<int> prices;int calculateMax(vector<int> prices)
{int i,nmin,nmax,len,ans;len=prices.size();int *lhs=new int[len];int *rhs=new int[len];lhs[0]=0;	//下标为0的时候卖出，利润为0nmin=prices[0];		//假定下标为0的的价格为最低for(i=1;i<len;i++){lhs[i]=max(lhs[i-1],prices[i]-nmin);cout<<"lhs 下标 "<<i<<" : "<<nmin<<" 时买入 "<<prices[i]<<" 时还没卖出，则能获得的最大利润是 "<<lhs[i]<<endl;if(prices[i]<nmin)nmin=prices[i];}rhs[len-1]=0;nmax=prices[len-1];for(i=len-2;i>=0;i--){rhs[i]=max(rhs[i+1],nmax-prices[i]);cout<<"rhs 下标 "<<i<<" : "<<prices[i]<<" 时买入 "<<nmax<<" 时还没卖出，则能获得的最大利润是 "<<rhs[i]<<endl;if(prices[i]>nmax)nmax=prices[i];}ans=0;for(i=0;i<len;i++){cout<<"下标 "<<i<<" : "<<prices[i]<<" 时能得到的最大利润为 "<<lhs[i]<<" 之后又能获得利润 "<<rhs[i]<<" 此时总共能获得利润 "<<lhs[i]+rhs[i]<<endl;if((lhs[i]+rhs[i])>ans)ans=lhs[i]+rhs[i];}return ans;
}int main()
{int m,len;while(cin>>m)prices.push_back(m);cout<<calculateMax(prices)<<endl;return 0;
}

再插个图片