洛谷 P5149 会议座位（归并排序，trie树）

传送门

---

解题思路

纯属为了想练trie树

其实map更简单

注意答案开longlong，数组开5*maxn

AC代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<iomanip>
#include<cmath>
using namespace std;
const int maxn=100005;
int n,ch[5*maxn][105],num[5*maxn],cnt;
long long ans;
string s,a[maxn],f[maxn];
void insert(string s,int numm){
    int len=s.length();
    int now=1;
    for(int i=0;i<len;i++){
        if(!ch[now][s[i]-'A']) ch[now][s[i]-'A']=++cnt;
        now=ch[now][s[i]-'A'];
    }
    num[now]=numm;
}
int getnum(int i){
    int len=a[i].length();
    int now=1;
    for(int j=0;j<len;j++){
        now=ch[now][a[i][j]-'A'];
    }
    return num[now];
}
void divide(int l,int r){
    if(l==r) return;
    int mid=(l+r)/2;
    divide(l,mid);
    divide(mid+1,r);
    int i=l,j=mid+1,k=l;
    while(i<=mid&&j<=r){
        if(getnum(i)>getnum(j)){
            ans+=mid-i+1;
            f[k++]=a[j++];
        }else{
            f[k++]=a[i++];
        }
    }
    while(i<=mid) f[k++]=a[i++];
    while(j<=r) f[k++]=a[j++];
    for(int i=l;i<=r;i++) a[i]=f[i];
}
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>s;
        insert(s,i);
    }
    for(int i=1;i<=n;i++) cin>>a[i];
    divide(1,n);
    cout<<ans;
    return 0;
}

啊啊啊，快上课了，赶紧跑回教室去啊啊啊