ABC,D莫名TLE求调
  • 板块学术版
  • 楼主T_TLucas_Yin
  • 当前回复1
  • 已保存回复1
  • 发布时间2024/8/24 22:13
  • 上次更新2024/8/24 22:15:27
查看原帖
ABC,D莫名TLE求调
835809
T_TLucas_Yin楼主2024/8/24 22:13

rt,后来用另一种方法过了,但有没有大佬能帮忙看看这个做法为什么会 TLE 两个点?

#include<bits/stdc++.h>
using namespace std;
int n,m,x,y,a[1000005],fa[1000005],d[1000005];
int head[1000005],ne[1000005],ver[1000005],tot;
void add(int x,int y){
	ver[++tot]=y,ne[tot]=head[x],head[x]=tot;
}
int sum;
void dfs(int x,int f,int k){
	fa[x]=f,d[x]=k;
	for(int i=head[x];~i;i=ne[i]) if(ver[i]!=f) dfs(ver[i],x,k+1);
}
bool flag[1000005];
void merge(int x,int y){
	flag[x]=1,flag[y]=1;
	while(x!=y){
		if(d[x]<d[y]) swap(x,y);
		x=fa[x],flag[x]=1;
	}
}
int main(){
	cin>>n>>m;
	memset(head,-1,sizeof(head));
	for(int i=1;i<n;i++){
		cin>>x>>y;
		add(x,y),add(y,x);
	}
	for(int i=1;i<=m;i++) cin>>a[i];
	dfs(1,0,1);
	if(m==1){
		cout<<1;
		return 0;
	}
	for(int i=1;i<m;i++) merge(a[i],a[i+1]);
	for(int i=1;i<=n;i++) sum+=flag[i];
	cout<<sum;
	return 0;
}
2024/8/24 22:13
加载中...