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;
}
标签:10,const,int,排位赛,ll,long,QDU,len,第三场 来源: https://blog.csdn.net/birdmanqin/article/details/110120614