2022春每日一题:Day 36


题目:[JLOI2013]删除物品
直接做显然比较复杂,这个题是说对顶栈,但是可以把两个栈拼在一起,记录一下栈顶的下标,然后这样这题就可以转化为线性上的操作查询了,用树状数组简单维护一下就ok了(某个数是否存在,单点修改+区间查询)
代码:
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#define lowbit(x) x&-x
const int N=1e5+5;
using namespace std;
int n1,n2,a[N],b[N],c[N],n,dy[N];
struct pos
{
int id,v;
pos(int ii,int vv)
{
id=ii;v=vv;
}
pos(){
}
friend bool operator < (pos a,pos b)
{
return a.v<b.v;
}
}nums[N];
namespace fenwick
{
struct fw
{
int c;
}e[N];
void modify(int x,int v)
{
for(int i=x;i<=n;i+=lowbit(i))
e[i].c+=v;
}
int query(int x)
{
int ret=0;
for(int i=x;i;i-=lowbit(i))
ret+=e[i].c;
return ret;
}
}
using namespace fenwick;
int main()
{
scanf("%d %d",&n1,&n2);
for(int i=1;i<=n1;i++)
scanf("%d",&a[i]),c[n1-i+1]=a[i];
for(int i=1;i<=n2;i++)
scanf("%d",&b[i]),c[n1+i]=b[i];
n=n1+n2;
for(int i=1;i<=n;i++)
nums[i]=pos(i,c[i]);
sort(nums+1,nums+1+n);
int last=-1,cnt=1;
for(int i=1;i<=n;i++)
{
if(last!=nums[i].v)
last=nums[i].v,cnt=i;
dy[cnt]=nums[i].id;
}
for(int i=1;i<=n;i++)
modify(dy[i],1);
int now=n1;
long long ret=0;
for(int i=n;i;i--)
{
if(dy[i]<=now)
{
ret+=query(now)-query(dy[i]);
now=dy[i];
}
else
{
ret+=query(dy[i]-1)-query(now);
now=dy[i]-1;
}
modify(dy[i],-1);
}
printf("%lld\n",ret);
return 0;
}

浙公网安备 33010602011771号