【bzoj2789】[Poi2012]Letters

Description

给出两个长度相同且由大写英文字母组成的字符串A、B，保证A和B中每种字母出现的次数相同。

现在每次可以交换A中相邻两个字符，求最少需要交换多少次可以使得A变成B。

Input

第一行一个正整数n (2<=n<=1,000,000)，表示字符串的长度。

第二行和第三行各一个长度为n的字符串，并且只包含大写英文字母。

Output

一个非负整数，表示最少的交换次数。
Sample Input

3

ABC

BCA

Sample Output

2
HINT

ABC -> BAC -> BCA

题解
按顺序对应每个字母，用树状数组模拟交换。

代码

#include<bits/stdc++.h>
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
typedef long long ll;
char s1[1000005],s2[1000005];
int n,last[27],Next[1000005],now[27],t[1000005];
bool flag[27];
ll ans;
inline int lowbit(int x){return x&-x;}
int query(int x)
{
    int ans=0;
    for (int i=x;i;i-=lowbit(i))
        ans+=t[i];
    return ans;
}
inline void update(int x)
{
    for (int i=x;i<=n;i+=lowbit(i))
        t[i]++;
}
int main()
{
    n=read();
    scanf("%s",s1+1);
    for (int i=n;i;i--)
    {
        int x=s1[i]-'A';
        Next[i]=last[x];
        last[x]=i;
    }
    for (int i=1;i<=n;i++)
    {
        int x=s1[i]-'A';
        if (flag[x]) continue;
        now[x]=i;flag[x]=1;
    }
    scanf("%s",s2+1);
    for (int i=1;i<=n;i++)
    {
        int x=s2[i]-'A';
        ans+=abs(query(n-now[x])+now[x]-i);
        update(n-now[x]+1);
        now[x]=Next[now[x]];
    }
    cout<<ans;
    return 0;
}