AcWing 1057. 股票买卖 IV
状态图
有:手中有股票
无:手中无股票
#include <stdio.h>
#include <cstring>
#include <algorithm>
#include <vector>
#include <stack>
#include <queue>
#include <iostream>
#include <map>
#include <cmath>
#include <set>
#define go(i, l, r) for(int i = (l), i##end = (int)(r); i <= i##end; ++i)
#define god(i, r, l) for(int i = (r), i##end = (int)(l); i >= i##end; --i)
#define ios ios_base::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define debug_in freopen("in.txt","r",stdin)
#define debug_out freopen("out.txt","w",stdout);
#define pb push_back
#define all(x) x.begin(),x.end()
#define fs first
#define sc second
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
const ll maxM = 1e6+10;
const ll inf_int = 1e8;
const ll inf_ll = 1e17;
template<class T>void read(T &x){
T s=0,w=1;
char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
x = s*w;
}
template<class H, class... T> void read(H& h, T&... t) {
read(h);
read(t...);
}
void pt(){ cout<<'\n';}
template<class H, class ... T> void pt(H h,T... t){ cout<<" "<<h; pt(t...);}
//--------------------------------------------
const int maxn = 1e5+10;
int N,K;
int a[maxn];
ll f[110][2],tmp[110][2];//用了拷贝数组,空间上减少一个维度
int main() {
// debug_in;
// debug_out;
read(N,K);
for(int i = 1;i<=N;i++) read(a[i]);
memset(f,-0x3f,sizeof f);
ll ans = 0;
f[0][0] = 0;
for(int i = 1;i<=N;i++){
memcpy(tmp,f,sizeof tmp);
for(int j = 1;j<=K;j++){
f[j][1] = max(tmp[j][1],tmp[j-1][0] - a[i]);
f[j][0] = max(tmp[j][1] + a[i],tmp[j][0]);
ans = max(ans,f[j][0]);
}
}
printf("%lld\n",ans);
return 0;
}