[BZOJ2789][Poi2012]Letters（树状数组求逆序对）

题目描述

传送门

题解

预处理之后求逆序对。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define LL long long
const int max_n=1e6+5;

char a[max_n],b[max_n];
int n,last[100],next[max_n],st[100],bit[max_n];
LL C[max_n],ans=0;

inline void add(int loc,int val){
    for (int i=loc;i<=n;i+=i&(-i)) C[i]+=val;
}
inline int query(int loc){
    int ans=0;
    for (int i=loc;i>=1;i-=i&(-i)) ans+=C[i];
    return ans;
}
int main(){
    scanf("%d\n",&n);
    gets(a); gets(b);
    for (int i=0;i<n;++i) next[i]=n;
    memset(last,-1,sizeof(last));
    for (int i=0;i<n;++i){
        if (last[a[i]]!=-1) next[last[a[i]]]=i;
        else st[a[i]]=i;
        last[a[i]]=i;
    }
    for (int i=0;i<n;++i){
        bit[i+1]=st[b[i]]+1;
        st[b[i]]=next[st[b[i]]];
    }
    for (int i=n;i>=1;--i){
        ans+=query(bit[i]);
        add(bit[i],1);
    }
    printf("%lld\n",ans);
}