大致思路和“我要长高”没什么区别。
d
p
[
i
]
[
j
]
dp[i][j]
dp[i][j]表示第
i
i
i天手上有份
j
j
j股票。
值得注意的是通过
d
p
[
i
]
[
j
]
=
m
a
x
(
d
p
[
i
]
[
j
]
,
d
p
[
i
−
1
]
[
j
]
)
dp[i][j]=max(dp[i][j],dp[i-1][j])
dp[i][j]=max(dp[i][j],dp[i−1][j])来处理只能在
W
W
W天之后才能交易的限制。于是只用考虑
d
p
[
i
−
W
−
1
]
[
j
]
dp[i-W-1][j]
dp[i−W−1][j]的情况了。
买:
d
p
[
i
]
[
j
]
=
m
a
x
(
d
p
[
i
−
W
−
1
]
[
k
]
−
(
j
−
k
)
∗
a
p
[
i
]
,
d
p
[
i
]
[
j
]
)
(
1
<
=
k
<
j
)
dp[i][j]=max(dp[i-W-1][k]-(j-k)*ap[i],dp[i][j])(1<=k<j)
dp[i][j]=max(dp[i−W−1][k]−(j−k)∗ap[i],dp[i][j])(1<=k<j)
卖:
d
p
[
i
]
[
j
]
=
m
a
x
(
d
p
[
i
−
W
−
1
]
[
k
]
+
(
k
−
j
)
∗
b
p
[
i
]
,
d
p
[
i
]
[
j
]
)
(
j
<
k
<
m
a
x
p
)
dp[i][j]=max(dp[i-W-1][k]+(k-j)*bp[i],dp[i][j])(j<k<maxp)
dp[i][j]=max(dp[i−W−1][k]+(k−j)∗bp[i],dp[i][j])(j<k<maxp)
对于买,展开后,单调队列维护
m
a
x
(
d
p
[
i
−
W
−
1
]
[
k
]
+
k
∗
a
p
[
i
]
)
max(dp[i-W-1][k]+k*ap[i])
max(dp[i−W−1][k]+k∗ap[i]),卖同理。
最后,注意初始化。前面1~(W+1)天只能买进。
#include<bits/stdc++.h>
using namespace std;
const int maxn=2e3+10;
const int oo=1<<30;
int orz,T,maxp,W,ans,head,tail;
int ap[maxn],bp[maxn],as[maxn],bs[maxn],dp[maxn][maxn];
struct Q{
int val,num;
Q(){}
Q(int val,int num):val(val),num(num){}
}q[maxn];
inline int read(){
int x=0;char ch=getchar();
while(!isdigit(ch)) ch=getchar();
while(isdigit(ch)) x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
return x;
}
int main(){
orz=read();
while(orz--){
T=read(),maxp=read(),W=read(),ans=-oo;
for(int i=1;i<=T;++i) ap[i]=read(),bp[i]=read(),as[i]=read(),bs[i]=read();
for(int i=1;i<=T;++i){for(int j=1;j<=maxp;++j) dp[i][j]=-oo;dp[i][0]=0;}
for(int i=1;i<=W+1;++i) for(int j=0;j<=as[i];++j) dp[i][j]=-j*ap[i];
for(int i=2;i<=T;++i){
for(int j=0;j<=maxp;++j) dp[i][j]=max(dp[i-1][j],dp[i][j]);
if(i<=W+1) continue;
head=0,tail=0;
for(int j=0;j<=maxp;++j){
int f=dp[i-W-1][j]+ap[i]*j;
while(head<tail&&q[tail-1].val<f) tail--;
q[tail++]=Q(f,j);
while(head<tail&&(j-q[head].num>as[i])) head++;
dp[i][j]=max(dp[i][j],q[head].val-ap[i]*j);
}
head=0,tail=0;
for(int j=maxp;j>=0;--j){
int f=dp[i-W-1][j]+bp[i]*j;
while(head<tail&&q[tail-1].val<f) tail--;
q[tail++]=Q(f,j);
while(head<tail&&(q[head].num-j>bs[i])) head++;
dp[i][j]=max(dp[i][j],q[head].val-bp[i]*j);
}
}
for(int i=0;i<=maxp;++i) ans=max(ans,dp[T][i]);
printf("%d\n",ans);
}
}