Description
对于序列A,它的逆序对数定义为满足i< j,且Ai>Aj的数对(i,j)的个数。给1到n的一个排列,按照某种顺序依次删除m个元素,你的任务是在每次删除一个元素之前统计整个序列的逆序对数。
Input
输入第一行包含两个整数n和m,即初始元素的个数和删除的元素个数。以下n行每行包含一个1到n之间的正整数,即初始排列。以下m行每行一个正整数,依次为每次删除的元素。
Output 输出包含m行,依次为删除每个元素之前,逆序对的个数。
解法二树套树见【这里】。
用树状数组正着倒着分别扫一遍就可以求出每个元素所在的逆序对数。但是删除这个元素的时候,还要把之间已经被删掉也就是被减掉两次的加回来。也就是说需要对每个元素求出在他之前被删除的逆序对个数。这样的话按删除时间进行CDQ分治,分治的每一层还是在归并的时候用树状数组正着倒着扫两遍。
#include<cstdio>
#include<algorithm>
using namespace std;
#define LL long long
int rd()
{
int x=0;
char c=getchar();
while (c<'0'||c>'9') c=getchar();
while (c>='0'&&c<='9')
{
x=x*10+c-'0';
c=getchar();
}
return x;
}
struct str
{
int v,p,t,cnt;
}a[50010],b[50010];
int n,m,f[100010],g[50010],cnt[100010],s[100010],num[50010],pos[100010];
int qry(int p)
{
int ret=0;
for (;p;p-=p&-p) ret+=s[p];
return ret;
}
void inc(int p)
{
for (;p<=n;p+=p&-p) s[p]++;
}
void dec(int p)
{
for (;p<=n;p+=p&-p) s[p]--;
}
void solve(int l,int r)
{
if (l==r) return;
int mid=l+r>>1;
solve(l,mid);
solve(mid+1,r);
for (int i=mid,j=r;i>=l||j>mid;)
{
if (i>=l&&(j==mid||a[i].p>a[j].p))
{
inc(a[i].v);
i--;
}
else
{
a[j].cnt+=qry(a[j].v);
j--;
}
}
for (int i=l;i<=mid;i++) dec(a[i].v);
for (int i=l,j=mid+1,now=l;i<=mid||j<=r;)
{
if (i<=mid&&(j>r||a[i].p<a[j].p))
{
inc(n-a[i].v+1);
b[now++]=a[i];
i++;
}
else
{
a[j].cnt+=qry(n-a[j].v+1);
b[now++]=a[j];
j++;
}
}
for (int i=l;i<=mid;i++) dec(n-a[i].v+1);
for (int i=l;i<=r;i++) a[i]=b[i];
}
int main()
{
LL ans=0;
n=rd();
m=rd();
for (int i=1;i<=n;i++) pos[f[i]=rd()]=i;
for (int i=1;i<=m;i++) g[i]=rd();
for (int i=1;i<=n;i++)
{
cnt[i]+=qry(n-f[i]+1);
inc(n-f[i]+1);
}
for (int i=1;i<=n;i++) s[i]=0;
for (int i=n;i;i--)
{
cnt[i]+=qry(f[i]);
inc(f[i]);
}
for (int i=1;i<=n;i++) s[i]=0;
for (int i=1;i<=n;i++) ans+=cnt[i];
ans/=2;
for (int i=1;i<=m;i++) a[i]=(str){g[i],pos[g[i]],i,0};
solve(1,m);
for (int i=1;i<=m;i++) num[a[i].t]=a[i].cnt;
for (int i=1;i<=m;i++)
{
printf("%lld\n",ans);
ans-=cnt[pos[g[i]]]-num[i];
}
}