洛谷p5149(字典树+逆序对)

P5149 会议座位 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

根据题意可以得到，题目要求的就是一个逆序对的问题。只要把排队后的数列求出来，再求一下逆序对就能解决问题了。因为名字是字符串，很容易想到字典树，用字典树求出数列，然后归并排序或者树状数组求一下逆序对就能得到答案了。

#include<cstdio>
#include<cstring>
int cnt,nex[500010][60],f[500010],a[500010],b[500010];
long long ans;
void merge(int l,int r)
{
    if (r-l<=0) return;
    int mid=(l+r)>>1;
    merge(l,mid);
    merge(mid+1,r);
    int p=l,q=mid+1,s=l;
    while (p<=mid || q<=r)
    {
        if ((p>mid) || (q<=r && a[p]>a[q]))
        {
            b[s++]=a[q++];
            ans+=mid-p+1;
        }
        else 
        {
            b[s++]=a[p++];
        }
    }
    for (int i=l;i<=r;i++)
        a[i]=b[i];
}
void trie(char st[100],int x)
{
    int p=0,c;
    for (int i=0;i<strlen(st);i++)
    {
        c=st[i]-65;
        if (!nex[p][c]) nex[p][c]=++cnt;
        p=nex[p][c];
    }
    f[p]=x;
}
int find(char st[100])
{
    int p=0,c;
    for (int i=0;i<strlen(st);i++)
    {
        c=st[i]-65;
        p=nex[p][c];
    }
    return f[p];
}
int main()
{
    int n,t;char st[100];
    cnt=0;ans=0;
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
    {
        scanf("%s",st);
        trie(st,i);
    }
    for (int i=1;i<=n;i++)
    {
        scanf("%s",st);
        t=find(st);
        a[i]=t; 
    }    
    merge(1,n);
    printf("%lld\n",ans);
}