BZOJ 2789 letters（树状数组）

Posted 2020-09-27 forever97‘s blog

 

【题目链接】 http://www.lydsy.com/JudgeOnline/problem.php?id=2789

 

【题目大意】

　　给出两个字符串，通过A字符串相邻之间字符的交换得到B字符串，
　　求最小的交换次数

 

【题解】

　　最小交换则对于同个字符来说前后顺序不变，我们得到B序列的字符在A序列中的位置，
　　顺序插入计算后缀和即可。

 

【代码】

#include <cstdio>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long LL;
const int N=1000010;
int c[N],pos[N],n;
queue<int> q[30];
LL ans;
char A[N],B[N];
void add(int x,int val){while(x)c[x]+=val,x-=x&-x;}
int query(int x){int res;while(x<=n)res+=c[x],x+=x&-x;return res;}
int main(){
    scanf("%d %s %s",&n,A+1,B+1);
    for(int i=1;i<=n;i++)q[A[i]-‘A‘].push(i);
    for(int i=1;i<=n;i++){
        pos[i]=q[B[i]-‘A‘].front();
        q[B[i]-‘A‘].pop();
    }ans=0;
    for(int i=1;i<=n;i++){
        ans+=query(pos[i]);
        add(pos[i],1);
    }printf("%lld\n",ans);
    return 0;
}