洛谷P1967

怒敲170+行的代码……


好吧,题目见此:点击看题

Kruskal最大生成树+树上倍增

很明显了,如果是要两点间路径承载的重量尽可能大且两点间有较多路径的话,就一定会只连可承载重量较大的边。于是我们想到用Kruskal来做最大生成树,尽管不一定所有的点之间两两联通,但是连出来的边承载重量一定是最大的。

接下来讲讲怎么求任意两点间最大承载重量的问题,也就是找两点路径当中的权值最小的边。由于用kruskal之后得到的一定是一片森林,所以可以用倍增lca来查询两个点是否在同一棵树里面。如果在同一棵树里面,则求出lca,并求两个点到lca的最小权值边。如果不在一棵树则输出-1。

代码见此:

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
#define rep(i,l,r) for(int i=l;i<=r;i++)
#define repu(i,r,l) for (int i=r;i>=l;i--)
#define N 10100
#define M 50500
using namespace std;
struct data
{
	int u,v,w;
}edge[M];
struct way
{
	int v,w,next;
};
int n,m,q,p[N],dep[N],parent[N][17],afford[N][17],x,y,z,head[N],cnt;
bool vis[N];
way map[2*M];
bool cmp(const data &a,const data &b)
{
	return a.w>b.w;
}
void ufsets()
{
	rep(i,1,n)
	p[i]=-1;
}
void addedge(int x,int y,int value)
{
	way e;
	cnt++;
	e.v=y;
	e.w=value;
	e.next=head[x];
	head[x]=cnt;
	map[cnt]=e;
	cnt++;
	e.v=x;
	e.next=head[y];
	head[y]=cnt;
	map[cnt]=e;
}
int find(int x)
{
	return p[x]<0?x:p[x]=find(p[x]);
}
void uni(int x,int y)
{
	if (p[x]<=p[y])
	{
		p[x]+=p[y];
		p[y]=x;
	}
	else
	{
		p[y]+=p[x];
		p[x]=y;
	}
}
void dfs(int x,int pa,int d)
{
	parent[x][0]=pa;
	dep[x]=d;
	vis[x]=true;
	for (int i=head[x];i;i=map[i].next)
	{
		int t=map[i].v;
		if ((!vis[t])&&(t!=x))dfs(t,x,d+1);
	}
}
int neiaff(int x,int y)
{
	for (int i=head[x];i;i=map[i].next)
	if (map[i].v==y) return map[i].w;
}
void makemap()
{
	memset(vis,0,sizeof(vis));
	memset(parent,0,sizeof(parent));
	int tree=0;
	rep(i,1,n)
	{
		if (!vis[i])
		{
			tree--;
			dfs(i,tree,0);
		}
	}
	rep(j,1,16)
	{
		rep(i,1,n)
		{
			if (parent[i][j-1]<0) parent[i][j]=parent[i][j-1];
			parent[i][j]=parent[parent[i][j-1]][j-1];
		}
	}
	rep(i,1,n)
	{
		afford[i][0]=neiaff(i,parent[i][0]);
	}
	rep(j,1,16)
	{
		rep(i,1,n)
		{
			if (parent[i][j-1]<0)continue;
			afford[i][j]=min(afford[i][j-1],afford[parent[i][j-1]][j-1]);
		}
	}
}
int getaff(int x,int y)
{
	int t=dep[x]-dep[y];
	int ans=1234567890;
	rep(i,0,16)
	if ((t>>i)&1)
	{	
		ans=min(ans,afford[x][i]);
		x=parent[x][i];	
	}	
	return ans;
}
int lca(int x,int y)
{
	if (dep[x]<dep[y]) swap(x,y);
	int t=dep[x]-dep[y];
	rep(i,0,16)
	if (t&(1<<i)) x=parent[x][i];
	repu(i,15,0)
	if (parent[x][i]!=parent[y][i])
	{
		x=parent[x][i];
		y=parent[y][i];
	}
	if (x==y) return x;
	else return parent[x][0];
}
int main()
{
	scanf("%d%d",&n,&m);
	rep(i,1,m)
	scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
	sort(edge+1,edge+m+1,cmp);
	cnt=0;
	ufsets();
	memset(head,0,sizeof(head));
	rep(i,1,m)
	{
		x=find(edge[i].u);
		y=find(edge[i].v);
		if (x!=y)
		{
			uni(x,y);
			addedge(x,y,edge[i].w);
		}
	}
	makemap();
	scanf("%d",&q);
	rep(i,1,q)
	{
		scanf("%d%d",&x,&y);
		if (find(x)!=find(y))
		{
			printf("-1\n");
			continue;
		}
		else
		{
			int anc=lca(x,y);
			printf("%d\n",min(getaff(x,anc),getaff(y,anc)));
		}
	}
	return 0;
}
PS 近期会将模版代码打包,专题存起来,方便复习,毕竟NOIP快到了,要干点正事了……

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值