洛谷P5149 会议座位

洛谷P5149

筱柒Littleseven
QQ：3534947775

题目链接

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define N 500100
#define ll long long
using namespace std;
int tot,ans[N],n,tmp[N];
ll cnt;
struct Trie{int nex[100],num;}trie[N];
void insert(char *s,int id)
{
  int now=0,len=strlen(s+1);
  for(int i=1;i<=len;i++)
  {
  	int ch=s[i]-'A';
  	if(!trie[now].nex[ch]) trie[now].nex[ch]=++tot;
  	now=trie[now].nex[ch];
  }
  trie[now].num=id;
}
int find(char *s)
{
  int now=0,len=strlen(s+1);
  for(int i=1;i<=len;i++)
  {
  	int ch=s[i]-'A';
  	now=trie[now].nex[ch];
  }
  return trie[now].num;
}
void merge(int l,int r)
{
  int mid=(l+r)>>1;
  int i=l,j=mid+1,k=l;
  while(i<=mid&&j<=r)
  {
  	if(ans[i]<ans[j]) tmp[k++]=ans[i++];
  	else
  	{
  		tmp[k++]=ans[j++];
  		cnt+=mid-i+1;
  	}
  }
  while(i<=mid) tmp[k++]=ans[i++];
  while(j<=r) tmp[k++]=ans[j++];
  for(k=l;k<=r;k++) ans[k]=tmp[k];
}
void msort(int l,int r)
{
  if(l>=r) return;
  int mid=(l+r)>>1;
  msort(l,mid); msort(mid+1,r);
  merge(l,r);
}
int main()
{
  char c[10]; cin>>n;
  for(int i=1;i<=n;i++){ scanf("%s",c+1); insert(c,i); }
  for(int i=1;i<=n;i++){ scanf("%s",c+1); ans[i]=find(c);}
  msort(1,n);  cout<<cnt<<endl;
  return 0;
}