Description
对于序列A,它的逆序对数定义为满足i
Input
输入第一行包含两个整数n和m,即初始元素的个数和删除的元素个数。
以下n行每行包含一个1到n之间的正整数,即初始排列。
以下m行每行一个正整数,依次为每次删除的元素。
N<=100000 M<=50000
Output
输出包含m行,依次为删除每个元素之前,逆序对的个数。
Sample Input
5 4
1
5
3
4
2
5
1
4
2
Sample Output
5
2
2
1
样例解释
(1,5,3,4,2)(1,3,4,2)(3,4,2)(3,2)(3)。
Solution
这是一道类似三维偏序的题目
我们先给每个数设定一个t值,即最先被删除的t值最大,没有被删除的点的t值按照位置从小到大拍
那么我们现在得到了一个三元组(t,x,pos)
我们要求的就是t
Code
#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
const int N=100005;
typedef long long ll;
struct edge{
ll x,y,z;
} a[N];
ll n,m,i,cnt,x,ans[N],tr[N];
ll read(){
ll sum=0;
char c=getchar();
while (c<'0'||c>'9') c=getchar();
while (c>='0'&&c<='9'){
sum=sum*10+c-'0';
c=getchar();}
return sum;
}
bool cmp1(edge a,edge b){
return a.z<b.z;
}
bool cmp2(edge a,edge b){
return a.x>b.x;
}
void add(ll x,ll y){
while (x<=n){
tr[x]+=y;
x+=x&(-x);}
}
ll query(ll x){
ll s=0;
while (x>0) {
s+=tr[x];
x-=x&(-x);}
return s;
}
void cdq(ll l,ll r){
if (l==r) return;
ll mid=(l+r)>>1;
cdq(l,mid),cdq(mid+1,r);
ll pl,pr;
pl=l,pr=mid+1;
sort(a+l,a+pr,cmp2);
sort(a+pr,a+r+1,cmp2);
while (pr<=r){
while (pl<=mid&&a[pr].x<a[pl].x)
add(a[pl].y,1),pl++;
ans[a[pr].z]+=query(a[pr].y);
pr++;
}
// if (pl>mid) pl--;
while (--pl>=l) add(a[pl].y,-1);
pl=mid,pr=r;
while (pr>=mid+1){
while (pl>=l&&a[pr].x>a[pl].x)
add(a[pl].y,1),pl--;
ans[a[pr].z]+=query(n)-query(a[pr].y);
pr--;
}
// if (pl<l) pl++;
while (++pl<=mid) add(a[pl].y,-1);
}
int main(){
//freopen("1.in","r",stdin);
//freopen("1.out","w",stdout);
n=read(),m=read(),cnt=1;
fo(i,1,n) x=read(),a[x].x=x,a[x].y=i;
fo(i,1,m) {
x=read();
a[x].z=n-i+1;}
fo(i,1,n)
if (!a[i].z) a[i].z=cnt++;
sort(a+1,a+n+1,cmp1);
cdq(1,n);
fo(i,2,n) ans[i]=ans[i]+ans[i-1];
fd(i,n,n-m+1) printf("%lld\n",ans[i]);
}