第二章.数据结构(二)

知识点 : Trie树 ,并查集,堆的操作

Trie树

高效地储存和查找字符串集合的数据结构

const int N = 100010;
int son[N][26],cnt[N],idx;

//插入 
void insert(char str[]){
	int p = 0;
	for(int i=0;str[i];i++){
		int u = str[i] - 'a';
		if(!son[p][u])son[p][u] = ++idx;
		p = son[p][u];
	}
	cnt[p]++;  //以这个点结尾的字符数 
}

//查询
int query(char str[]){
	int p = 0;
	for(int i=0;str[i];i++){
		int u = str[i] - 'a';
		if(!son[p][u])return 0;
		p = son[p][u];
	}
	return cnt[p];
}

并查集

实现两个集合的合并,询问两个元素是否在一个集合中.几乎 O(1);
原理 : 每一个集合用一棵树表示,每个结点储存它的父亲结点,p[x] 表示x的父亲结点.
1.判断树根 : p[x] = x;
2.求x的集合编号 : while(p[x] != x) x = p[x]; 此时进行优化,路径压缩
3.合并 :
注意题目要维护的是啥…


维护size

int p[N],size[N];  //根结点的元素个数

void init(int n){
	for(int i=1;i<=n;i++){
		p[i] = i;
		size[i] = 1; 
	}
}
//返回x的祖宗 
int find_set(int x){
	if(p[x] != x)p[x] = find_set(p[x]);
	return p[x]; 
}
//合并 
int merge_set(int x,int y){
	x = find_set(x);
	y = find_set(y);
	p[x] = y;
	size[y] += size[x];
}

维护与根结点的距离

int p[N], d[N];
//p[]存储每个点的祖宗节点, d[x]存储x到p[x]的距离
int find(int x){
	if (p[x] != x){
 	   int u = find(p[x]);
       d[x] += d[p[x]];
       p[x] = u;
    }
    return p[x];
}
// 初始化，假定节点编号是1~n
    for (int i = 1; i <= n; i ++ ){
        p[i] = i;
        d[i] = 0;
    }
    // 合并a和b所在的两个集合：
    p[find(a)] = find(b);
    d[find(a)] = distance; // 根据具体问题，初始化find(a)的偏移量

堆

插入:,求集合中的最小值,删除最小值.
删除任意一个元素 … 修改任意一个元素…

本质是二叉树.
用数组模拟 : a[1]表示根节点,x 的左儿子是 2x , 右儿子是 2x + 1.
up 操作 : 每次找到两个儿子最小的,与父亲交换,直到不用交换.

//插入一个数
heap[++sizee] = x;up(sizee);
//删除最小值
heap[1] = heap[sizee];sizee–;down(1);
//删除任意一个元素
heap[k] = heap[sizee];sizee–;;down(k), up(k);
//修改任意一个元素
heap[k] = x;down(k), up(k);

int n,m;
int h[N],sizee;
int ph[N],hp[N];
//ph[]储存第k个插入的点在堆中的位置
//hp[]储存下标是k的点是第几个插入的

void h_swap(int a,int b){
	 swap(ph[hp[a]],ph[hp[b]]);
	 swap(hp[a],hp[b]);
	 swap(a,b);
}
 
void down(int u){
	int t = u;
	if(2*u <= sizee && h[2*u] < h[t])t = 2*u;
	if(2*u+1 <= sizee && h[2*u+1] < h[t])t = 2*u+1;
	if(u != t){
		h_swap(u,t);
		down(t);
	}
}

void up(int u){
	while(u/2 && h[u/2] > h[u]){
		h_swap(u/2,u);
		u /= 2;
	}
}

int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		cin >> h[i];
	}	
	sizee = n;
	//将h[]按照小根堆排好 
	for(int i = n/2;i;i--){
		down(i);
	}
	return 0;
}	

//删除最小的数	
{
    h_swap(1,sizee);
	sizee--;
	down(1);
}

//删除第k个数
{
	k = ph[k]; //第k个点的位置
	h_swap(k,sizee);
	sizee --;
	down(k),up(k);
} 

//修改第k个数为x
{
	k = ph[k];
	h[k] = x;
	down(k),up(k);
}	

//插入一个数
{
	sizee++;
	m++;  //第几个插入的数 
	ph[m] = sizee,hp[sizee] = m;
	h[sizee] = x;
	up(sizee);
}