bzoj 2789 [Poi2012]Letters 树状数组

最新推荐文章于 2024-06-03 17:17:41 发布

原创最新推荐文章于 2024-06-03 17:17:41 发布 · 372 阅读

0 ·

CC 4.0 BY-SA版权

POI 同时被 2 个专栏收录

5 篇文章

订阅专栏

树状数组

2 篇文章

订阅专栏

本文介绍了一种使用树状数组进行优化的算法题目解决方案。该算法通过维护字符串A中每个字符往后第一个与其相等的字符位置，并利用树状数组来记录每个位置的排名，实现了高效地求解字符交换次数的过程。

第i次将A中大于等于i的第一个等于bi的位置换到第i个。
维护一下A中每个点往后第一个和他相等的字符。用树状数组维护每个点当前的排名。

#include <bits/stdc++.h>
using namespace std;
#define N 1100000
#define ll long long
int n;
char s1[N],s2[N];
int tr[N],nex[N],pos[26];
ll ans;
void insert(int x,int v)
{
    for(int i=x;i<=n;i+=i&-i)
        tr[i]+=v;
}
int query(int x)
{
    int ret=0;
    for(int i=x;i;i-=i&-i)
        ret+=tr[i];
    return ret;
}
int main()
{
    //freopen("tt.in","r",stdin);
    scanf("%d%s%s",&n,s1+1,s2+1);
    for(int i=1;i<=n;i++)
        insert(i,1);
    for(int i=0;i<26;i++)pos[i]=n+1;
    for(int i=n;i>=1;i--)
    {
        nex[i]=pos[s1[i]-'A'];
        pos[s1[i]-'A']=i;
    }
    for(int i=1;i<=n;i++)
    {
        int t=pos[s2[i]-'A'];
        ans+=query(t)-1;
        pos[s2[i]-'A']=nex[t];
        insert(t,-1);
    }
    printf("%lld\n",ans);
    return 0;
}