bzoj2789 [Poi2012]Letters 树状数组

最新推荐文章于 2022-01-31 13:47:28 发布

原创最新推荐文章于 2022-01-31 13:47:28 发布 · 395 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#bzoj

收录于

c++ 同时被 2 个专栏收录

1072 篇文章

订阅专栏

树状数组

29 篇文章

订阅专栏

本文介绍了一种算法，用于计算将一个字符串通过交换相邻字符变为另一个字符串所需的最小交换次数。字符串长度可达1,000,000，且两字符串中每种字母出现次数相同。采用树状数组优化逆序对计算过程。

Description

给出两个长度相同且由大写英文字母组成的字符串A、B，保证A和B中每种字母出现的次数相同。
现在每次可以交换A中相邻两个字符，求最少需要交换多少次可以使得A变成B。

2<=n<=1,000,000
只包含大写英文字母。

Solution

每个字母都会移动到距离最近的相同字母位置，把这样一个关系看成是边求相交边的数量即可
可以发现这是一个逆序对问题，用树状数组就行了

Code

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)
#define lowbit(x) ((x)&(-(x)))

typedef long long LL;
const int N=2000005;

char a[N],b[N];

int next[N],last[N],d[N],n;
LL c[N];

void add(int x,int v) {
    for (;x<=n;x+=lowbit(x)) c[x]+=v;
}

LL get(int x) {
    LL ret=0;
    for (;x;x-=lowbit(x)) ret+=c[x];
    return ret;
}

int main(void) {
    scanf("%d",&n);
    scanf("%s%s",a+1,b+1);
    rep(i,1,n) {
        next[i]=last[b[i]-'A'];
        last[b[i]-'A']=i;
    }
    drp(i,n,1) {
        d[i]=last[a[i]-'A'];
        last[a[i]-'A']=next[last[a[i]-'A']];
    }
    LL ans=0;
    rep(i,1,n) {
        ans+=get(n)-get(d[i]);
        add(d[i],1);
    }
    printf("%lld\n", ans);
    return 0;
}