其他分享
首页 > 其他分享> > (状态机dp)股票买卖 IV

(状态机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