(状态机dp)股票买卖 IV
作者:互联网
题目:
给定一个长度为 N 的数组,数组中的第 i 个数字表示一个给定股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润,你最多可以完成 k 笔交易。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。一次买入卖出合为一笔交易。
输入格式
第一行包含整数 N 和 k,表示数组的长度以及你可以完成的最大交易数量。
第二行包含 N 个不超过 10000 的正整数,表示完整的数组。
输出格式
输出一个整数,表示最大利润。
数据范围
1≤N≤1e5,
1≤k≤100
输入样例1:
3 2
2 4 1
输出样例1:
2
输入样例2:
6 2
3 2 6 5 0 3
输出样例2:
7
思路:
最开始没有考虑全初始化的问题,很难受。
AcCode:
#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
typedef long long ll;
typedef pair<int, int> PII;
const int mod = 1e9 + 7;
const int inf = 0x3f3f3f3f;
const int N = 1e5 + 10;
int n, k, m, w[N];
int dp[N][110][2];
int main(void){
ios::sync_with_stdio(false);
cin.tie(0);
cin >> n >> k;
for(int i = 1; i <= n; i++)
cin >> w[i];
memset(dp, -0x3f, sizeof dp); //就这个
for(int i = 1; i <= n; i++) //还有这个初始化,下次一定想仔细一点
dp[i][0][0] = 0, dp[i][0][1] = -w[i];
for(int i = 2; i <= n; i++){
for(int j = 0; j <= k; j++){
if(j >= 1) dp[i][j][0] = max(dp[i - 1][j - 1][1] + w[i], dp[i - 1][j][0]);
else dp[i][j][0] = dp[i - 1][j][0];
dp[i][j][1] = max(dp[i - 1][j][0] - w[i], dp[i - 1][j][1]);
}
}
int res = 0;
for(int i = 0; i <= k; i++)
res = max(res, dp[n][i][0]);
cout << res << endl;
return 0;
}
标签:输出,const,int,样例,IV,状态机,数组,dp 来源: https://www.cnblogs.com/ReSakura/p/16218227.html