Problem【★★★】
有一串序列,你要使得最终的序列正负交替,有两个操作
1.用c的代价使任意部分元素+1或-1。
2.用r代价删除一个数。
问你最后用的代价最少是多少。
Solution【★★★★】
在k*c的代价下一部分元素变的可正可负,可正可负用?表示,假设一个子序列+??+?+,前两个+显然是不合法的,需要删一个,无论删哪一个都不会影响后面序列的合法性(删除数量)。这个性质使得我们可以在按权值从小到大枚举变成可负可正的点的时候可以O(1)计算此时产生的总代价。
0的情况比较特殊,因为对于0来说只会出现全部删除或者全部变成可正可负的情况,其他情况不会比这两种最优的那个优。
Code【★★】
链表维护还没有变成可正可负的元素,详见注释。
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cmath>
using namespace std;
const int maxn=5e5+5;
long long n,c,r,a[maxn];
long long minc=1e15,ans;
int lef[maxn],rig[maxn];
struct node{
long long x;
int id;
}s[maxn];
bool cmp(const node &a,const node &b)
{
return abs(a.x)<abs(b.x);
}
long long numcut(int i,int j)//计算两个元素间需删除的数量
{
if(i==0||i==n+1||j==0||j==n+1) return 0;
int x=(j-i)%2;
if(a[i]*a[j]>0&&!x) return 0;
if(a[i]*a[j]<0&&x) return 0;
return 1;
}
int main()
{
scanf("%lld%lld%lld",&n,&c,&r);
for(int i=1;i<=n;i++)
{
scanf("%lld",&s[i].x);
a[i]=s[i].x;
s[i].id=i;
}
sort(s+1,s+1+n,cmp);
for(int i=1;i<=n;i++)//初始化链表
{
lef[i]=i-1;
rig[i]=i+1;
}
long long nw=0;
int k=1,last=0;
//计算全部删除0的情况
while(k<=n&&a[k]==0)
{
nw+=r;
k++;
}
for(int i=k+1;i<=n;i++)
{
if(a[i]==0) nw+=r;
else if(a[k]*a[i]>0)
nw+=r;
else
k=i;
}
minc=nw;
//计算把0全部变成可正可负元素的代价
nw=0;
for(int i=1;i<=n;i++)
{
if(a[i])
{
nw+=numcut(last,i)*r;
last=i;
}
else{//更新链表
rig[lef[i]]=rig[i];
lef[rig[i]]=lef[i];
}
}
minc=min(minc,nw+c);
//按权值从小到大枚举
for(int i=1;i<=n;i++)
{
if(s[i].x==0) continue;
//计算把当前元素变成可正可负后新的删除元素总数
nw-=(numcut(lef[s[i].id],s[i].id)+numcut(s[i].id,rig[s[i].id])-numcut(lef[s[i].id],rig[s[i].id]))*r;
minc=min(minc,nw+(abs(s[i].x)+1)*c);
rig[lef[s[i].id]]=rig[s[i].id];//更新链表
lef[rig[s[i].id]]=lef[s[i].id];
}
cout<<minc;
return 0;
}