[校内模拟][Kruscal重构树][线段树合并] 201031 NOIP T3 Garden

题意

一张图，点有颜色，边有边权，查询从一个点开始，每次经过边权不大于某值的边，求经过的颜色种经过次数最多且最小的

30pts暴力

这种值最多且最小的问题，最近写线段树合并写多了就看着很熟
但是我不会kruscal重构树，就只写了$O(n^2)$的丑陋代码
30pts

#define FILE(x) freopen(x".in","r",stdin);freopen(x".out","w",stdout);
#include<bits/stdc++.h>
using namespace std;
#define in Read()
int in{
	int i=0,f=1;char ch=0;
	while(!isdigit(ch)&&ch!='-') ch=getchar();
	if(ch=='-') ch=getchar();
	while(isdigit(ch)) i=(i<<1)+(i<<3)+ch-48,ch=getchar();
	return i*f;
}

const int N=4e5+5,M=N*25;
int n,m,val[N],type;
int tot,first[N],nxt[N<<1],aim[N<<1],wei[N<<1];
int sz,col[M],lch[M],rch[M],num[M];
int tim,vis[N],root[N];

void new_node(int p){
	lch[p]=rch[p]=0;
	num[p]=col[p]=0;
	return;
}

void ljb(int u,int v,int w){
	++tot;
	nxt[tot]=first[u];
	first[u]=tot;
	wei[tot]=w;
	aim[tot]=v;
	return;
}

void push_up(int p){
	if(num[lch[p]]>=num[rch[p]]){
		num[p]=num[lch[p]];
		col[p]=col[lch[p]];
		return;
	}else{
		num[p]=num[rch[p]];
		col[p]=col[rch[p]];
		return;
	}
}

void update(int &p,int l,int r,int pos,int v){
	if(!p) new_node(p=++sz);
	if(l==r){
		col[p]=l;
		num[p]+=v;
		return;
	}
	
	int mid=l+r>>1;
	if(pos<=mid) update(lch[p],l,mid,pos,v);
	else update(rch[p],mid+1,r,pos,v);
	push_up(p);
	return;
}

int merge(int p,int rt,int l,int r){
	if(!p||!rt) return p|rt;
	if(l==r){
		col[p]=l;
		num[p]+=num[rt];
		return p;
	}
	
	int mid=l+r>>1;
	lch[p]=merge(lch[p],lch[rt],l,mid);
	rch[p]=merge(rch[p],rch[rt],mid+1,r);
	push_up(p);
	return p;
}

void DFS(int u,int lim){
	
	vis[u]=tim;
	update(root[u],1,n,val[u],1);
	
	for(int e=first[u];e;e=nxt[e]){
		int v=aim[e];
		if(vis[v]==tim) continue;
		if(wei[e]>lim) continue;
		DFS(v,lim);
		root[u]=merge(root[u],root[v],1,n);
	}
	return;
}

int main(){
	FILE("garden");
	
	n=in,m=in,type=in-1;
	for(int i=1;i<=n;++i) val[i]=in;
	for(int i=1;i<=m;++i){
		int u=in,v=in,w=in;
		ljb(u,v,w);
		ljb(v,u,w);
	}
	
	int las=0,q=in;
	for(int i=1;i<=q;++i){
		int s=in,l=in;
		if(type) s^=las,l^=las;
		
		sz=0,++tim;
		memset(root,0,sizeof root);
		
		DFS(s,l);
		printf("%d\n",las=col[root[s]]);
	}
	
	return 0;
}

正解

Kruscal重构树

重构树，倍增查询，线段树合并预处理
启发式合并跑的飞快但是不会不想学线段树合并够了
其实启发式合并的思想掌握了应该能够YY出来

#include<bits/stdc++.h>
using namespace std;
#define in Read()
int in{
	int i=0,f=1;char ch=0;
	while(!isdigit(ch)&&ch!='-') ch=getchar();
	if(ch=='-') ch=getchar();
	while(isdigit(ch))
		i=(i<<1)+(i<<3)+ch-48,ch=getchar();
	return i*f;
}

const int N=2e5+5;
int n,m,q,type,tot,las=0;
int c[N],fa[N],val[N];
vector<int>G[N];
int f[N][25],root[N];
int mst[N],pos[N],num[N];
int lch[N],rch[N],sz;

struct node{
	int u,v,w;
	
	friend inline bool operator < 
	(const node &a,const node &b)
	{return a.w<b.w;}
	
}edge[N<<1];

int get(int x){
	return x==fa[x]?x:fa[x]=get(fa[x]);
}

void push_up(int p){
	if(num[lch[p]]>=num[rch[p]]){
		num[p]=num[lch[p]];
		pos[p]=pos[lch[p]];
	}else{
		num[p]=num[rch[p]];
		pos[p]=pos[rch[p]];
	}
	return;
}

void update(int &p,int l,int r,int pnt,int v){
	if(!p) p=++sz;
	if(l==r){
		pos[p]=pnt;
		num[p]+=v;
		return;
	}
	
	int mid=l+r>>1;
	if(pnt<=mid) update(lch[p],l,mid,pnt,v);
	else update(rch[p],mid+1,r,pnt,v);
	push_up(p);
	return;
}

int merge(int p,int rt,int l,int r){
	if(!p||!rt) return p|rt;
	if(l==r){
		pos[p]=l;
		num[p]+=num[rt];
		return p;
	}
	
	int mid=l+r>>1;
	lch[p]=merge(lch[p],lch[rt],l,mid);
	rch[p]=merge(rch[p],rch[rt],mid+1,r);
	push_up(p);
	return p;
}

void DFS(int u,int dad){
	f[u][0]=dad;
	for(int l=1;l<=20;++l)
		f[u][l]=f[f[u][l-1]][l-1];
	if(!G[u].size())
		update(root[u],1,n,c[u],1);
	for(int v:G[u]){
		DFS(v,u);
		root[u]=merge(root[u],root[v],1,n);
	}
	mst[u]=pos[root[u]];
	return;
}

int query(int u,int lim){
//	cout<<u<<" ";
	for(int l=20;l>=0;--l){
		if(f[u][l]==tot||f[u][l]==0) continue;
		if(val[f[u][l]]<=lim) u=f[u][l];
	}
//	cout<<u<<","<<mst[u]<<endl;
	return mst[u];
}

int main(){
	
	freopen("1.in","r",stdin);
	
	tot=n=in,m=in,type=in-1;
	for(int i=1;i<=n;++i) c[i]=in;
	
	for(int i=1;i<=m;++i){
		edge[i].u=in;
		edge[i].v=in;
		edge[i].w=in;
	}
	sort(edge+1,edge+m+1);
	for(int i=1;i<=n;++i) fa[i]=i;
	for(int i=1;i<=m;++i){
		int u=get(edge[i].u),v=get(edge[i].v);
		if(u==v) continue;
		++tot;
		fa[tot]=fa[u]=fa[v]=tot;
		val[tot]=edge[i].w;
		G[tot].push_back(u);
		G[tot].push_back(v);
	}
	++tot;
	for(int i=1;i<=tot;++i)
		if(fa[i]==i)
			G[tot].push_back(i);
	
	DFS(tot,0);
	
	q=in;
	for(int i=1;i<=q;++i){
		int s=in,l=in;
		if(type) s^=las,l^=las;
		printf("%d\n",las=query(s,l));
	}
	
//	for(int i=1;i<=tot;++i)
//		printf("%d ",mst[i]);
	
	return 0;
}

认真看我程序的同学应该发现了，我数组开小了😭
线段树合并的空间和主席树等动态开点的树据结构差不多，要*20