股市编程问题的解决方案

题目描述

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

输入例子:

3,8,5,1,7,8

输出例子:

12

题目分析

对手中的一支股票要求先买入后卖出

某一时刻股票有两种状态：持一股；未持股（不可能出现同时持有两股：上一股未卖出，下一股已买入）

需要求出收益最大的max=买卖第一股获得的最大收益max1+买卖第二股的最大收益max2

那么我们只需要划分两个股票区间即可

#include<iostream>
#include<vector>
using namespace std;
 
class Solution {
	
	public:
	    /**
	     * 计算你能获得的最大收益
	     * 
	     * @param prices Prices[i]即第i天的股价
	     * @return 整型
	     */
	    int calculateMax(vector<int> prices) {
	            int len=prices.size();
	                int sum=0;
	            for(int i=1;i<len;i++){
			int temp=getmax(prices,0,i)+getmax(prices,i,len-1);
	              	if(temp>sum)
	                    sum=temp;
		    }
	            return sum;
	    }
	    
	   int getmax(vector<int> p,int a,int b){
	       int maxnum=-1;
	       if(a>=b)
	           return 0;
	     
	       for(int i=a;i<b;i++)
	           for(int j=i+1;j<=b;j++)
	               maxnum=max(maxnum,p[j]-p[i]);
	               
	       return maxnum;
	   }
};

int main(){
	vector<int> arr;
	arr.push_back(3);
	arr.push_back(8);
	arr.push_back(5);
	arr.push_back(1);
	arr.push_back(7);
	arr.push_back(8);
	
	Solution a;
	cout<<a.calculateMax(arr);
	return 0;
}