[bzoj2141][排队] (分块大法好)

Posted 2020-08-28 AronQi

Description

排排坐，吃果果，生果甜嗦嗦，大家笑呵呵。你一个，我一个，大的分给你，小的留给我，吃完果果唱支歌，大家乐和和。红星幼儿园的小朋友们排起了长长地队伍，准备吃果果。不过因为小朋友们的身高有所区别，排成的队伍高低错乱，极不美观。设第i个小朋友的身高为hi，我们定义一个序列的杂乱程度为：满足ihj的(i,j)数量。幼儿园阿姨每次会选出两个小朋友，交换他们的位置，请你帮忙计算出每次交换后，序列的杂乱程度。为方便幼儿园阿姨统计，在未进行任何交换操作时，你也应该输出该序列的杂乱程度。

Input

第一行为一个正整数n，表示小朋友的数量；第二行包含n个由空格分隔的正整数h1,h2,…,hn，依次表示初始队列中小朋友的身高；第三行为一个正整数m，表示交换操作的次数；以下m行每行包含两个正整数ai和bi¬，表示交换位置ai与位置bi的小朋友。

Output

输出文件共m行，第i行一个正整数表示交换操作i结束后，序列的杂乱程度。

Sample Input

【样例输入】

3
130 150 140
2
2 3
1 3

Sample Output

1
0
3

【样例说明】
未进行任何操作时，(2,3)满足条件；
操作1结束后，序列为130 140 150，不存在满足ihj的(i,j)对；
操作2结束后，序列为150 140 130,(1,2)，(1,3)，(2,3)共3对满足条件的(i,j)。
【数据规模和约定】
对于100%的数据，1≤m≤2*103，1≤n≤2*104，1≤hi≤109，ai≠bi，1≤ai,bi≤n。

Solution

分块啊，用块套树状数组，分块处理逆序对

考虑交换，不完整块就暴力枚举更新完整的就用树状数组直接更新

//Kaiba_Seto 20170116
//otz cjkmao
#include <math.h>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#define MaxN 20010
#define MaxBuf 1<<22
#define RG register
#define inline __inline__ __attribute__((always_inline))
#define Blue() (((S == T)&&(T=(S=B)+fread(B,1,MaxBuf,stdin),S == T)) ? 0 : *S++)

char B[MaxBuf],*S=B,*T=B;

inline void Rin(RG int &x) {
    x=0;RG int c=Blue(),f=1;
    for(; c<48||c>57; c=Blue())
        if(c==45)f=-1;
    for(; c>47&&c<58; c=Blue())
        x=(x<<1)+(x<<3)+c-48;
    x*=f; }

struct Pr{
    int fir,sec;
    bool operator < (const Pr &other) const {
        return fir < other.fir; } }b[MaxN];

int n,m,a[MaxN],block_size,c1[MaxN],c2[200][MaxN],ans;

inline void exc(RG int &x,RG int &y) {
    x^=y;
    y^=x;
    x^=y; }

inline void modify(RG int *C,RG int x,RG int d) {
    for(; x<=n; x+=x&(-x))
        C[x]+=d; }

inline int sum(RG int *C,RG int x) {
    RG int res=0;
    for(; x; x-=x&(-x))
        res+=C[x];
    return res; }

int main() {
    Rin(n);
    block_size=static_cast<int>(sqrt(n)+1e-6);
    for(RG int i=1; i<=n; i++)
        Rin(b[i].fir),b[i].sec=i;
    std::sort(b+1,b+1+n);
    for(RG int i=1; i<=n; i++) {
        static int top=0;
        if(b[i].fir != b[i-1].fir)
            ++top;
        a[b[i].sec]=top; }
    for(RG int i=n; i; i--)
        ans+=sum(c1,a[i]-1),modify(c1,a[i],1);
    for(RG int i=1; i<=n; i++)
        modify(c2[(i-1)/block_size],a[i],1);
    printf("%d\n",ans);
    Rin(m);
    while(m--) {
        RG int x,y,l,r,i;
        Rin(x),Rin(y);
        if(x > y)exc(x,y);
        l=(x-1)/block_size+1;
        r=(y-1)/block_size-1;
        if(l <= r) {
            for(i=l; i<=r; i++)
                ans=ans-sum(c2[i],a[x]-1)+sum(c2[i],n)-sum(c2[i],a[x])+sum(c2[i],a[y]-1)-sum(c2[i],n)+sum(c2[i],a[y]);
            for(i=x+1; i<=l*block_size; i++) {
                if(a[i] < a[x])ans--;
                if(a[i] > a[x])ans++;
                if(a[i] > a[y])ans--;
                if(a[i] < a[y])ans++; }
            for(i=(r+1)*block_size+1; i<y; i++) {
                if(a[i] < a[x])ans--;
                if(a[i] > a[x])ans++;
                if(a[i] > a[y])ans--;
                if(a[i] < a[y])ans++; }
        }
        else {
            for(i=x+1; i<y; i++) {
                if(a[i] < a[x])ans--;
                if(a[i] > a[x])ans++;
                if(a[i] > a[y])ans--;
                if(a[i] < a[y])ans++; } }
        if(a[x] < a[y])ans++;
        else if(a[x] > a[y])ans--;
        printf("%d\n",ans);
        modify(c2[(x-1)/block_size],a[x],-1);
        modify(c2[(x-1)/block_size],a[y],1);
        modify(c2[(y-1)/block_size],a[x],1);
        modify(c2[(y-1)/block_size],a[y],-1);
        exc(a[x],a[y]); }
    fclose(stdin);
    return 0; }

orz cjkmao/Mr.cjk.cat