1. 程式人生 > >【bzoj2789】 Letters 樹狀陣列

【bzoj2789】 Letters 樹狀陣列

又是一道樹狀陣列求逆序對的題目。

這一題我們可以將第二個串中的每一個字母,與第一個串中的字母做兩兩匹配,令第二個串第i個字母的值id[i]為該字母與第一個串中的字母匹配到的位置。

然後考慮到所求答案為最小的移動次數,那麼這個最小的移動次數顯然為序列id的逆序對個數。

樹狀陣列求一求就沒了。

我一看時間只剩15min了本來想看下能否打完的,沒想到3min打完了。。。。

 1 #include<bits/stdc++.h>
 2 #define M 2000005
 3 #define L long long
 4 #define lowbit(x) (x)&(-x)
 5
using namespace std; 6 7 char c[M]={0}; 8 int n,a[M]={0}; 9 void add(int x,int k){for(;x<=n;x+=lowbit(x)) a[x]+=k;} 10 int sum(int x){int k=0;for(;x;x-=lowbit(x)) k+=a[x]; return k;} 11 priority_queue<int> q[26]; 12 int main(){ 13 scanf("%d",&n); 14 scanf("%s",c+1);
15 for(int i=1;i<=n;i++) 16 q[c[i]-'A'].push(-i); 17 scanf("%s",c+1); 18 L ans=0; 19 for(int i=1;i<=n;i++){ 20 int id=-q[c[i]-'A'].top(); q[c[i]-'A'].pop(); 21 ans+=sum(n)-sum(id); 22 add(id,1); 23 } 24 cout<<ans<<endl; 25
}