删除物品

HYSBZ 3192:http://www.lydsy.com/JudgeOnline/problem.php?id=3192

 

题意：中文题。

题解：把两个堆头对头并起来。例如样例：5 4 1 2 7 3，然后两堆之间的设置mid，对于每一次删除最大的来说，例如，删除7，我们要知道上一次mid的位子，然后两个坐标的差值就是这一次移动的步数，前提是7到mid之间的元素，在这之前没有删除，所以要统计这两个数之间的数。可以用树状数组来维护。具体的看代码。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct Node{
  int nu;
  int data;
  bool operator<(const Node a)const{
    return data<a.data;
  }
}num[100002];
int n1,n2,n,mi;
long long ans;
int a[100002];
int c[100002];
int lowbit(int x){
  return x&(-x);
}
void add(int x){
   while(x<=n){
    c[x]+=1;
    x+=lowbit(x);
   }
}
int getsum(int x){
   int ans=0;
   while(x>=1){
      ans+=c[x];
      x-=lowbit(x);
   }
  return ans;
}
int main(){
   scanf("%d%d",&n1,&n2);
   n=n1+n2;mi=n1;ans=0;
   for(int i=n1;i>=1;i--)scanf("%d",&a[i]);
   for(int i=n1+1;i<=n1+n2;i++)scanf("%d",&a[i]);
   for(int i=1;i<=n1+n2;i++){
     num[i].nu=i;
     num[i].data=a[i];
   }
    sort(num+1,num+n+1);
   memset(c,0,sizeof(c));
   for(int i=n;i>=1;i--){
     int x=num[i].nu;
     //printf("%d\n",x);
     if(mi>=x){
       ans+=(mi-x)-(getsum(mi)-getsum(x));
       mi=x;
       add(x);
     }
     else{
      ans+=(x-1-mi)-(getsum(x-1)-getsum(mi));
       mi=x-1;
       add(x);
     }
   }
   printf("%lld\n",ans);
 
 
}