细看题目,这不就是那道数据备份(其实这类题目有4道左右)。
题目简单理解:
在
n
−
1
n-1
n−1 条路里选择
k
k
k 个不相邻的路段,使得路段长度总和最小。
初看题目,你会联想到线性DP,再看一下数据范围:能水部分分
为了不空间超限,我们压维(发现第一维只用记录前3个数值)
有了如下代码(不完整):
#include<bits/stdc++.h>
using namespace std;
int n,k,A[100005],a[100005],dp[5][500005];
int main(){
scanf("%d %d",&n,&k);
for(int i=1;i<=n;i++){
scanf("%d",&A[i]);
a[i]=A[i]-A[i-1];
}
for(int i=1;i<=k;i++)dp[2][i]=dp[1][i]=dp[0][i]=0x3f3f3f3f3f;
for(int i=2;i<=n;i++){
for(int j=0;j<=k;j++){
if(j!=0)dp[i%3][j]=min(dp[(i-2)%3][j-1]+a[i],dp[(i-1)%3][j]);
else dp[i%3][j]=dp[(i-1)%3][j];
}
}
printf("%d",dp[n%3][k]);
return 0;
}
现在我们来考虑正解,这一道题可以用贪心来解决(只不过是反悔贪心),也就是我们先选择最小值,在退回来验证选其两边数字的答案,就保证了正确性。
首先记录当前编号是否选择过,而对于删除元素操作,普通的数组太慢了,于是我们使用链表来进行维护。
首先左端点初始化为 i − 1 i-1 i−1,同理右端点初始化为 i + 1 i+1 i+1,注意此题不是环状,所以两边要特判处理。
并且为了快速取得最小值,我们考虑用优先队列进行维护,选择后记录编号,为保证之后的处理:当前节点的数值变为其左端点数值加右端点数值减当前数值。
处理后重新放入队列(删除该节点)。
其实此题还是与数据备份有一些不同的,比如“最大获利”的情况下,求解答案就有了些许的改变,见下:
ans+=zqw.val;
Ans=max(Ans,ans);
最后答案是 Ans
,而数据备份的答案是 ans
。
#include<bits/stdc++.h>
#define int long long
using namespace std;
struct Q1{int val,l,r;}a[500005];
struct Q{
int val,id;
bool operator<(Q x)const{return val<x.val;}
};
int n,m,ans;
bool vis[500005];
priority_queue<Q>q;
int Ans;
inline void remove(int x){a[x].l=a[a[x].l].l,a[x].r=a[a[x].r].r,a[a[x].l].r=x,a[a[x].r].l=x;}
signed main(){
scanf("%lld %lld",&n,&m);
for(int i=1;i<=n;++i){
scanf("%lld",&a[i].val);
a[i].l=i-1,a[i].r=i+1,q.push(Q{a[i].val,i});
}
for(int i=1;i<=m;++i){
while(vis[q.top().id])q.pop();
Q zqw=q.top();q.pop();
ans+=zqw.val;
Ans=max(Ans,ans);
vis[a[zqw.id].l]=vis[a[zqw.id].r]=1;
a[zqw.id].val=a[a[zqw.id].l].val+a[a[zqw.id].r].val-a[zqw.id].val;
q.push(Q{a[zqw.id].val,zqw.id});
remove(zqw.id);
}
printf("%lld",Ans);
return 0;
}