2021.08.22 字典树

2021.08.22 字典树

P2580 于是他错误的点名开始了 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;

const int N=5e5+10;
int n,m,cnt=1,end[N],tree[N][30],vis[N*30];
char s[50];

inline int read(){
	int s=0,w=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){
		if(ch=='-')w=-1;
		ch=getchar();
	}
	while(ch<='9'&&ch>='0'){
		s=s*10+ch-'0';
		ch=getchar();
	}
	return s*w;
}
void insert(char *ch,int len){
	int x=0;
	for(int i=0;i<len;i++){
		int xi=ch[i]-'a';
		if(!tree[x][xi])tree[x][xi]=++cnt;
		x=tree[x][xi];
	}
	end[x]=1;
}
int find(char *ch,int len){
	int x=0;
	for(int i=0;i<len;i++){
		int xi=ch[i]-'a';
		if(!tree[x][xi])return 0;
		x=tree[x][xi];
	}
	if(!vis[x])return vis[x]=1,1;
	else return 2;
}

int main(){
	n=read();
	for(int i=1;i<=n;i++){
		scanf("%s",s);
		insert(s,strlen(s));
		//cout<<s<<" "<<strlen(s)<<endl;
	}
	m=read();
	for(int i=1;i<=m;i++){
		scanf("%s",s);
		int x=find(s,strlen(s));
		if(!x)cout<<"WRONG"<<endl;
		else if(x==1)cout<<"OK"<<endl;
		else if(x==2)cout<<"REPEAT"<<endl;
	}
	return 0;
}

重点：

1.逆序对的两种写法

1.树状数组、线段树

2.归并排序

注：trie每个点都还有延伸的点，cnt有可能爆int，记得开足空间。

//重点：逆序对！ 
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long ll;
const int N=1e5+10;
int n,trie[N*5][100],end[N*5],tree[N*5],a[N*5];
ll cnt,ans;
char s[10];

inline int read(){
	int s=0,w=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){
		if(ch=='-')w=-1;
		ch=getchar();
	}
	while(ch<='9'&&ch>='0'){
		s=s*10+ch-'0';
		ch=getchar();
	}
	return s*w;
}
void insert(char *ch,int len,int id){
	int x=0;
	for(int i=0;i<len;i++){
		int xi=ch[i]-'A';
		if(!trie[x][xi])trie[x][xi]=++cnt;
		x=trie[x][xi];
	}
	end[x]=id;
}
int find(char *ch,int len){
	int x=0;
	for(int i=0;i<len;i++){
		int xi=ch[i]-'A';
		if(!trie[x][xi])return 0;
		x=trie[x][xi];
	}
	return end[x];
}
int lowbit(int x){
	return x&-x;
}
void add(int x,int k){
	for(int i=x;i<=n;i+=lowbit(i))tree[i]+=k;
}
int query(int x){
	int fin=0;
	for(int i=x;i;i-=lowbit(i))fin+=tree[i];
	return fin;
}

int main(){
	n=read();
	for(int i=1;i<=n;i++){
		scanf("%s",s);
		insert(s,strlen(s),i);
	}
	for(int i=1;i<=n;i++){
		scanf("%s",s);
		a[i]=find(s,strlen(s));
	}
	for(int i=1;i<=n;i++){
		add(a[i],1);
		ans+=(ll)(i-query(a[i]));
	}
	cout<<ans;
	return 0;
}