I
题意
要求:交易一次
(交易的定义为一次买入和一次卖出,并且进行下一次交易前上一次交易必须结束,且同一天不能同时进行买入和卖出的操作)
思路
求个前缀最小值和后缀最大值,枚举中间点
时间复杂度:
O(n)
Code
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n = prices.size();
if (n == 0) return 0;
int* minn = new int[n], *maxx= new int[n];
minn[0] = prices[0]; maxx[n-1] = prices[n-1];
for (int i = 1; i < n; ++i) {
minn[i] = min(minn[i-1], prices[i]);
}
for (int i = n-2; i >= 0; --i) {
maxx[i] = max(maxx[i+1], prices[i]);
}
int ans = 0;
for (int i = 0; i < n-1; ++i) {
ans = max(ans, maxx[i+1]-minn[i]);
}
return ans;
}
};
II
题意
要求:交易不限次数
思路
考虑一个单调增的序列,显然不会从中间交易,而是取头和尾。
所以就是把序列分为若干个尽量长的单调增序列,每个都取头和尾。
时间复杂度:
O(n)
Code
class Solution {
public:
int maxProfit(vector<int>& prices) {
int i = 0, j, n = prices.size(), ans = 0;
for (; i < n; i = j+1) {
for (j = i; j < n-1; ++j) if (prices[j+1] <= prices[j]) break;
ans += prices[j] - prices[i];
}
return ans;
}
};
III
题意
要求:两次交易
思路
dp[i][0]
表示第
i
天及以前第一次买入,
转移方程十分显然(主要是要想到这么表示= =)
Code
class Solution {
const int inf = 0x3f3f3f3f;
public:
int maxProfit(vector<int>& prices) {
int n = prices.size();
if (n == 0) return 0;
int dp[2][4];
vector<int>& v = prices;
dp[0][0] = -v[0], dp[0][1] = dp[0][2] = dp[0][3] = -inf;
int ans = 0;
for (int i = 1; i < n; ++i) {
dp[i&1][0] = max(dp[(i-1)&1][0], -v[i]);
dp[i&1][1] = max(dp[(i-1)&1][1], dp[(i-1)&1][0] + v[i]);
dp[i&1][2] = max(dp[(i-1)&1][2], dp[(i-1)&1][1] - v[i]);
dp[i&1][3] = max(dp[(i-1)&1][3], dp[(i-1)&1][2] + v[i]);
ans = max(ans, max(dp[i&1][3], dp[i&1][1]));
}
return ans;
}
};
IV
题意
要求:
思路
同上题
Code
class Solution {
const int inf = 0x3f3f3f3f;
public:
int maxProfit(int k, vector<int>& prices) {
int n = prices.size();
if (n == 0) return 0;
k = min(n>>1, k);
k <<= 1;
int** dp = new int*[2];
for (int i = 0; i < 2; ++i) dp[i] = new int[k];
vector<int>& v = prices;
dp[0][0] = -v[0];
for (int i = 1; i < k; ++i) dp[0][i] = -inf;
int ans = 0;
for (int i = 1; i < n; ++i) {
dp[i&1][0] = max(dp[(i-1)&1][0], -v[i]);
for (int j = 1; j < k; ++j) {
dp[i&1][j] = max(dp[(i-1)&1][j], dp[(i-1)&1][j-1]+((j & 1) ? v[i] : -v[i]));
if (j&1) ans = max(ans, dp[i&1][j]);
}
}
return ans;
}
};