甲虫
题解
很简单的一道dp题。
很容易看出是一道区间的dp题,但由于
m
≤
1
0
6
m\leq 10^6
m≤106的数据范围限制,我们应该尽量避免让时间这一维出现在我们的dp中。
可它所有的水滴都会随着时间的发展而衰减,我们应该如何处理呢?
我们考虑一开始就将水滴加入状态中,之后再逐步递减。容易得到状态转移方程式
d
p
l
,
r
,
0
=
m
a
x
(
d
p
l
+
1
,
r
,
0
−
(
n
−
r
+
l
)
(
a
l
+
1
−
a
l
)
,
d
p
l
+
1
,
r
,
1
−
(
n
−
r
+
l
)
(
a
r
−
a
l
)
)
dp_{l,r,0}=max \left ( dp_{l+1,r,0}-(n-r+l)(a_{l+1}-a_{l}),dp_{l+1,r,1}-(n-r+l)(a_{r}-a_{l})\right )
dpl,r,0=max(dpl+1,r,0−(n−r+l)(al+1−al),dpl+1,r,1−(n−r+l)(ar−al))
可是最优解不一定是把全部
n
n
n个水滴都拿到了,这样的话有可能会拿到负数的权值。
于是我们考虑先枚举会拿到多少水滴再进行dp。
总时间复杂度
O
(
n
3
)
O\left ( n^3 \right )
O(n3)。
源码
#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<set>
using namespace std;
typedef long long LL;
typedef pair<int,int> pii;
template<typename _T>
void read(_T &x){
_T f=1;x=0;char s=getchar();
while('0'>s||s>'9'){if(s=='-')f=-1;s=getchar();}
while('0'<=s&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=getchar();}
x*=f;
}
int n,m,a[305],ans,dp[305][305][2];
int main(){
freopen("beetle.in","r",stdin);
freopen("beetle.out","w",stdout);
read(n);read(m);
for(int i=1;i<=n;i++)read(a[i]);a[++n]=0;
sort(a+1,a+n+1);int zero=0;
for(int i=1;i<=n;i++)if(a[i]==0)zero=i;
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++)
for(int k=j;k<=n;k++)
dp[j][k][0]=dp[j][k][1]=0;
dp[zero][zero][0]=(i-1)*m;
for(int j=2;j<=i;j++)
for(int l=max(1,zero-j+1);l<=min(zero,n-j+1);l++){
int r=l+j-1;
dp[l][r][0]=max(dp[l+1][r][0]-(a[l+1]-a[l])*(i-r+l),dp[l][r][0]);
dp[l][r][0]=max(dp[l+1][r][1]-(a[r]-a[l])*(i-r+l),dp[l][r][0]);
dp[l][r][1]=max(dp[l][r-1][1]-(a[r]-a[r-1])*(i-r+l),dp[l][r][1]);
dp[l][r][1]=max(dp[l][r-1][0]-(a[r]-a[l])*(i-r+l),dp[l][r][1]);
}
for(int l=max(1,zero-i+1);l<=min(zero,n-i+1);l++)
ans=max(max(dp[l][l+i-1][0],dp[l][l+i-1][1]),ans);
}
printf("%d\n",ans);
return 0;
}
谢谢!!!