风口的猪-中国牛市(小米2016校招)
1、题目描述
风口之下,猪都能飞。当今中国股市牛市,真可谓“错过等七年”。 给你一个回顾历史的机会,已知一支股票连续n天的价格走势,以长度为n的整数数组表示,数组中第i个元素(prices[i])代表该股票第i天的股价。 假设你一开始没有股票,但有至多两次买入1股而后卖出1股的机会,并且买入前一定要先保证手上没有股票。若两次交易机会都放弃,收益为0。 设计算法,计算你能获得的最大收益。 输入数值范围:2<=n<=100,0<=prices[i]<=100
输入例子:
3,8,5,1,7,8
输出例子:
12
2、代码:
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int calculateMax(vector<int> prices) {
int len = prices.size();
vector<int> maxLeft(len,0);
//Left to right
for (int i = 1, minPos = 0;i < len;++i)
{
if (prices[i] > prices[i - 1])
{
maxLeft[i] = max(maxLeft[i - 1], prices[i] - prices[minPos]);
}
else
{
maxLeft[i] = maxLeft[i - 1];
if (prices[i] < prices[minPos])
{
minPos = i;
}
}
}
//Right to left
vector<int> maxRight(len, 0);
maxRight[len - 1] = 0;
for (int i = len - 2, maxPos = len - 1;i >= 0;--i)
{
if (prices[i] < prices[i + 1])
{
maxRight[i] = max(maxRight[i + 1], prices[maxPos] - prices[i]);
}
else
{
maxRight[i] = maxRight[i + 1];
if (prices[i] > prices[maxPos])
{
maxPos = i;
}
}
}
int res=0;
for (int i = 0;i < len;++i)
{
res = max(res, maxLeft[i] + maxRight[i]);
}
return res;
}
int main()
{
int ia;
vector<int> v;
while (cin >> ia)
{
v.push_back(ia);
}
cout << calculateMax(v) << endl;
system("pause");
return 0;
}
3、总结:
参考了两个解答写出,实在感叹自身实力不足。
A、从左至右扫描得到每天能收益的最大值,记数组maxLeft,此时我们要以最小值为基准;从右至左扫描得到每天能收益的最大值,记数组maxRight,此时我们要以最大值为基准。返回每天两数组和的最大值。
注意点:结果中Left部分相当于是第一次获得的最大收益,Right则是第二次。这个想法对目前的我太难。这个解答方式是我主要参考的思路。
B、这个思路很自然,但是我竟然没有想到。以i=1开始,求i前面最大收益,求i后面最大收益,记录最大和。然后++i,不断更新最大和,结束时返回最大和。求最大收益:i=3,k=1,j=0,更新max,k=2,j=0,1,更新max……不断求取前后最大,满足手里始终只有一支股。