题意:
给定长度为n的序列a和长度为m的序列b,保证序列中的数两两不同
你可以对a序列进行两种操作:
1.花费x的代价,将连续k个数消除
2.花费y的代价,选择两个相邻的数,将其中小的数消除.
问最少花费多少的代价能使得a序列变成b序列.
如果无解输出-1.
数据范围:n,m<=2e5,k<=n
解法:
根据b序列,会将a序列分成若干段,
现在就是要将每段中间的多余数删掉.
如果b中的数在a中的相对顺序不同则无解.
1.如果中间段长度<k,且最大值大于两端,那么无解.
2.如果中间段长度<k,且最大值小于两端,那么一个一个删.
3.如果中间段长度>k,且最大值大于两端,
那么至少需要用一次删除k段的,剩下的部分根据代价贪心选择就行了.
为了便于处理,在a和b序列的首位各加上一个-1.
code:
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=2e6+5;
int del[maxm];
int a[maxm];
int b[maxm];
int n,m;
int x,k,y;
//
vector<int>temp[maxm];
//
signed main(){
ios::sync_with_stdio(0);
cin>>n>>m;
cin>>x>>k>>y;
n++,m++;
a[1]=b[1]=-1;
for(int i=2;i<=n;i++){
cin>>a[i];
}
for(int i=2;i<=m;i++){
cin>>b[i];
}
n++,m++;
a[n]=b[m]=-1;
//判断相对顺序,双指针
int j=1;
for(int i=1;i<=m;i++){
while(j<=n&&a[j]!=b[i])j++;
if(j>n){
cout<<-1<<endl;return 0;
}
}
//分段
map<int,int>mark;
for(int i=1;i<=m;i++){
mark[b[i]]=1;
}
int num=0;
for(int i=1;i<=n;i++){
if(mark[a[i]]){
if(i==n)break;
num++;
temp[num].push_back(a[i]);
int j=i+1;
while(j<=n&&!mark[a[j]]){
temp[num].push_back(a[j]);
j++;
}
temp[num].push_back(a[j]);
i=j-1;
}
}
//
int ans=0;
for(int i=1;i<=num;i++){
int len=temp[i].size();
if(len==2)continue;
int lc=temp[i][0],rc=temp[i][len-1];
int ma=temp[i][1];
int cnt=len-2;
for(int j=1;j<len-1;j++){
ma=max(ma,temp[i][j]);
}
if(cnt<k&&ma>lc&&ma>rc){//无解
cout<<-1<<endl;return 0;
}
if(cnt<k){
ans+=y*cnt;
continue;
}
//cnt>=k
if(ma<lc||ma<rc){//直接贪
ans+=cnt%k*y;
cnt-=cnt%k;
ans+=min(cnt/k*x,cnt*y);
}else{//至少一次k消除
ans+=x;
cnt-=k;
ans+=cnt%k*y;
cnt-=cnt%k;
ans+=min(cnt/k*x,cnt*y);
}
}
cout<<ans<<endl;
return 0;
}