tarjan缩点+LCA板子题求助,只有54pts
查看原帖
tarjan缩点+LCA板子题求助,只有54pts
365110
xuanyuan_Niubi楼主2021/2/4 17:10

代码如下,就是一个tarjan和LCA合起来,不知道为什么会错了,请求大佬指点!!

#include<cstdio>
#include<stack>
#include<algorithm>
#define max(a,b) (a>b?a:b)
#define min(a,b) (a<b?a:b)
#define swap(a,b) (a^=b^=a^=b)
using namespace std;
const int M=20005;
const int INF=0x3f3f3f3f;
struct edge{
	int v,nxt;
}ed[M<<4],ed_new[M<<4];
int n,m,head[M],cnt,head_new[M],cnt_new;
int f[M][15],deep[M];
int dfn[M],low[M],color[M],cnt_dfn,cnt_color;
bool vis[M];
stack<int>s;
inline int read(){
	char c=getchar();int x=0;
	for(;c<'0'||c>'9';c=getchar());
	for(;c<='9'&&c>='0';c=getchar())x=(x<<1)+(x<<3)+(c^48);
	return x;
}
inline void print(int x) {
    if(x<0) putchar('-'),x=-x;
    if(!x) return ;print(x/2),putchar(x%2+48);
}
inline void write(int x){//输出01串
	if(!x)putchar('0');
	else print(x);
	putchar('\n');
}
//----------------------------------
inline void add(int u,int v){
	ed[++cnt]=(edge){v,head[u]};
	head[u]=cnt;
}
inline void add_new(int u,int v){//缩点后的新图
	ed_new[++cnt_new]=(edge){v,head_new[u]};
	head_new[u]=cnt_new;
}
inline void tarjan(int now,int fa){
	dfn[now]=low[now]=++cnt_dfn;
	s.push(now);
	for(int i=head[now];i;i=ed[i].nxt){
		int v=ed[i].v;
		if(v==fa)continue;
		if(!dfn[v]){
			tarjan(v,now);
			low[now]=min(low[now],low[v]);
		}
		else{
			low[now]=min(low[now],dfn[v]);
		}
	}
	if(low[now]==dfn[now]){
		cnt_color++;
		while(s.top()!=now){
			color[s.top()]=cnt_color;
			s.pop();
		}
		color[s.top()]=cnt_color;
		s.pop();
	}
}
//-----------------tarjan---------------------
inline void dfs(int now,int fa,int dep){
	deep[now]=dep;f[now][0]=fa;
	for(int i=1;i<=15;i++){
		f[now][i]=f[f[now][i-1]][i-1];
	}
	for(int i=head_new[now];i;i=ed_new[i].nxt){//在新图上搞
		int v=ed_new[i].v;
		if(v!=fa)dfs(v,now,dep+1);
	}
}
inline int LCA(int a,int b){
	if(a==b)return a;
	if(deep[a]<deep[b])swap(a,b);
	for(int i=15;i>=0;i--){
		if(deep[f[a][i]]>=deep[b]){
			a=f[a][i];
		}
	}
	if(a==b)return a;
	for(int i=15;i>=0;i--){
		if(f[a][i]!=f[b][i]){
			a=f[a][i];
			b=f[b][i];
		}
	}
	return f[a][0];
}
inline int dis(int a,int b,int c){
	return deep[a]+deep[b]-(deep[c]<<1)+1;
}
//-------LCA板子-------------
int main(){
	n=read();m=read();
	for(int i=1;i<=m;i++){
		int u=read(),v=read();
		add(u,v);add(v,u);
	}
	for(int i=1;i<=n;i++){
		if(!dfn[i]){
			tarjan(i,0);
		}
	}
	for(int i=1;i<=n;i++){//建新图
		for(int j=head[i];j;j=ed[j].nxt){
			int v=ed[j].v;
			if(color[v]!=color[i]){
				add_new(color[v],color[i]);
			}
		}
	} 
	dfs(1,1,1);
	int q=read();
	for(int i=1;i<=q;i++){
		int a=read(),b=read();
		write(dis(color[a],color[b],LCA(color[a],color[b])));
	}
	return 0;
}

感激不尽!!!!!!

2021/2/4 17:10
加载中...