细品一下,发现此题和数据备份十分的相似,简单理解一下题意:
在
n
−
1
n-1
n−1 条路里选择
k
k
k 个不相邻的路段,使得路段长度总和最小。
然后你就找到了此题的多倍经验……
初看题目,你会联想到线性DP,再看一下数据范围:能水70分
为了不空间超限,我们压维(发现第一维只用记录前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,注意此题不是环状,所以两边要特判处理。
并且为了快速取得最小值,我们考虑用优先队列进行维护,选择后记录编号,为保证之后的处理:当前节点的数值变为其左端点数值加右端点数值减当前数值。
处理后重新放入队列(删除该节点)。
数值初始化一下……
于是我们有了如下代码:
#include<bits/stdc++.h>
using namespace std;
struct Q1{int val,l,r;}a[200005];
struct Q{
int val,id;
bool operator>(const Q&p)const{return val>p.val;}
};
int n,m,ans,s[200005];
bool vis[200005];
priority_queue<Q,vector<Q>,greater<Q> >q;
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(){
int T;
scanf("%d",&T);
while(T--){
ans=0;
while(q.size())q.pop();
memset(vis,0,sizeof vis);
memset(a,0,sizeof a);
scanf("%d %d %d",&n,&m,&s[0]);
for(int i=1;i<n;++i){
scanf("%d",&s[i]);
a[i].val=s[i]-s[i-1];
a[i].r=(i==n-1?0:i+1),a[i].l=(i==1?0:i-1),q.push(Q{a[i].val,i});
}
a[0].val=1e9;
q.push(Q{a[0].val,0});
while(m--){
Q zqw=q.top();q.pop();
if(vis[zqw.id]){++m;continue;}
ans+=a[zqw.id].val;
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("%d\n",ans);
}
return 0;
}