QDU校内排位赛第三场 D 于老师发生什么事了 （字典树+逆序对）

题链：https://pintia.cn/problem-sets/1330210570443206656/problems/1330211663055204354

题目：

思路1：字典树思想，插入一个数时，如果当前这一位是0，那么，之前插入的这一位是1的数都与他构成逆序对。插入时，统计一下每一位为1的数量即可。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 2e7+10;
const int M = 1e5+10;
int trie[N][2];
int num[N],len[M];
int n,id=1,limit=0;
ll ans=0;
char str[M][151],s[151];
void insert(){
	int root=0,nex;
	for(int i=0;i<limit;i++){
		nex=s[i]-'0';
		if(nex==0) ans+=num[root];
		if(!trie[root][nex]) trie[root][nex]=id++;
		if(nex==1) num[root]++;
		root=trie[root][nex];
	}
}
int main(void){
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%s",str[i]+1);
		len[i]=strlen(str[i]+1);
		if(limit<len[i]) limit=len[i];
	}
	s[limit]='\0';
	for(int i=1;i<=n;i++){
		int num0=limit-len[i];
		for(int j=0;j<num0;j++)	
			s[j]='0';
		for(int j=0;j<len[i];j++)
			s[j+num0]=str[i][j+1];	
		insert();
	}	
	printf("%lld",ans);
	return 0;	
} 

思路2：归并排序即可，因为数据范围原因，不会超时，范围再大点就会了。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e5+10;
string a[N],b[N];
int n,len[N],c[N];
ll ans=0;
bool big(int x,int y){
	if(len[x]==len[y]){
		for(int i=0;i<len[x];i++)
			if(a[x][i]>a[y][i]) return 1;
			else if(a[x][i]<a[y][i]) return 0;
		return 0;
	}else
		return len[x]>len[y];
}
void f(int l,int r){
	if(l==r) return ;
	int m=(l+r)>>1;
	f(l,m);
	f(m+1,r);
	int i=l,j=m+1,k=l;
	while(i<=m&&j<=r){
		if(big(i,j)){
			ans+=m-i+1;
			b[k]=a[j],c[k++]=len[j++];	
		}else
			b[k]=a[i],c[k++]=len[i++];
	}
	while(i<=m){
		b[k]=a[i];
		c[k++]=len[i++];	
	} 
	while(j<=r){
		b[k]=a[j];
		c[k++]=len[j++];		
	} 
	for(i=l;i<=r;i++)
		a[i]=b[i],len[i]=c[i];
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		char s[200];
		scanf("%s",s+1);
        a[i]=s+1;
        len[i]=strlen(s+1);
	}
	f(1,n);
	cout<<ans<<endl;
	return 0;
}