luoguP3531 [POI2012]LIT-Letters

(https://www.luogu.org/problem/P3531)

注意编号

#include
#include
#include
#include
#include
using namespace std;
const int MAX = 1001001;

int n;
long long ans;
int p[MAX], num[MAX], t[MAX];
string a,b;
vector  v[27];

int add(int x, int k) {
    while(x <= n) {
        t[x] += k;
        x += x&(-x);
    }
}

int query(int x) {
    int res = 0;
    while(x) {
        res += t[x];
        x -= x&(-x);
    }
    return res;
}

int main() {
    scanf("%d",&n);
    cin>>a>>b;
    for(int i = 0; i < n; i++) v[b[i]-'A'+1].push_back(i+1);//对b编号
    for(int i = 0; i < n; i++) p[i+1] = v[a[i]-'A'+1][num[a[i]-'A'+1]++];//对a编号 
//  for(int i = 1; i <= n; i++) printf("%d ", p[i]);
//裸求逆序对（因为a中字母在b中的位置固定，对于相同的字母，一定是a中第一个位置的 放在 b中第一个位置， 第二个放第二个...）
    for(int i = 1; i <= n; i++) {
        add(p[i], 1);
        ans += (long long)i - query(p[i]);
    }
    printf("%lld\n",ans);
}