P5318 【深基18.例3】查找文献

这是洛谷上的一个图遍历的题，对于结果分别使用dfs和bfs输出，对于每一层要求从最小的开始，所以我们需要先排序

刚开始是一边搜索，一边排序，不出意外，TLE四个点
代码如下：

#include<bits/stdc++.h>

using namespace std;

const int N=1000010;
int h[N],ne[N],e[N],idx;
int n,m;
int q[N],d[N];
int Q[N],T=-1;
bool st[N];
int ans1[N],ans2[N];
int res1[N],res2[N];
int num1=0,num2=0;

void add(int a,int b)
{
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

void dfs(int u,int uu)
{
	if(st[u]) return ;
	st[u]=true;
	ans1[++num1]=u;
	int num=0;
	for(int i=h[u];i!=-1;i=ne[i])
	{
		int j=e[i];
		res2[++num]=j;
	}
	sort(res2+1,res2+num+1);
	
	for(int i=1;i<=num;i++) Q[++T]=res2[i];
	
	while(uu<=T)
	{
		if(!st[Q[uu++]]) dfs(Q[uu-1],num+uu-1);
	}
	return ;
}

void bfs()
{
	memset(d,-1,sizeof d);
	d[1]=0;
	ans2[++num2]=1;
	
	int hh=0,tt=0;
	q[hh]=1;
	
	while(hh<=tt)
	{
		int t=q[hh++];
		int num=0;
		for(int i=h[t];i!=-1;i=ne[i])
		{
			int j=e[i];
			if(d[j]==-1) 
			{
				d[j]=0;
				res1[++num]=j;
			}
		}
		sort(res1+1,res1+num+1);
		for(int i=1;i<=num;i++)
		{
			q[++tt]=res1[i];
			ans2[++num2]=res1[i];
		}
	}
	return ;
}

int main()
{
	scanf("%d%d",&n,&m);
	memset(h,-1,sizeof h);
	for(int i=1;i<=m;i++)
	{
		int a,b;
		scanf("%d%d",&a,&b);
		add(a,b);
	}
	dfs(1,0);
	bfs();
	
	for(int i=1;i<=n;i++) printf("%d ",ans1[i]);
	puts("");
	for(int i=1;i<=n;i++) printf("%d ",ans2[i]);
	return 0;
}

然后，我更改了代码，先排序，后搜索
本来是WA了第三个点的，在输出点的时候并不是所有的点都输出，之后就成功了
AC代码如下：

#include<bits/stdc++.h>

using namespace std;

const int N=1000010;
int h[N],ne[N],e[N],idx;
int n,m;
int q[N],d[N];
bool st[N];
int ans1[N],ans2[N];
int res1[N],res2[N];
int num1=0,num2=0;
vector<pair<int,int>> vv;

bool cmp(pair<int,int> a,pair<int,int> b)
{
	if(a.first!=b.first) return a.first>b.first;
	return a.second>b.second;
}

void add(int a,int b)
{
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

void dfs(int u)//int uu
{
	if(st[u]) return ;
	st[u]=true;
	ans1[++num1]=u;
//	int num=0;
	for(int i=h[u];i!=-1;i=ne[i])
	{
		int j=e[i];
//		res2[++num]=j;
		if(!st[j]) dfs(j);
	}
//	sort(res2+1,res2+num+1);
	
//	for(int i=1;i<=num;i++) Q[++T]=res2[i];
	
//	while(uu<=T)
//	{
//		if(!st[Q[uu++]]) dfs(Q[uu-1],num+uu-1);
//	}
	return ;
}

void bfs()
{
	memset(d,-1,sizeof d);
	d[1]=0;
	ans2[++num2]=1;
	
	int hh=0,tt=0;
	q[hh]=1;
	
	while(hh<=tt)
	{
		int t=q[hh++];
		int num=0;
		for(int i=h[t];i!=-1;i=ne[i])
		{
			int j=e[i];
			if(d[j]==-1) 
			{
				d[j]=0;
			//	res1[++num]=j;
				q[++tt]=j;
				ans2[++num2]=j;
			}
		}
//		sort(res1+1,res1+num+1);
//		for(int i=1;i<=num;i++)
//		{
//			q[++tt]=res1[i];
//			ans2[++num2]=res1[i];
//		}
	}
	return ;
}

int main()
{
	scanf("%d%d",&n,&m);
	memset(h,-1,sizeof h);
	for(int i=1;i<=m;i++)
	{
		int a,b;
		scanf("%d%d",&a,&b);
		vv.push_back({a,b});
	//	add(a,b);
	}
	sort(vv.begin(),vv.end(),cmp);
	for(auto it:vv)
	{
		add(it.first,it.second);
	}
	dfs(1);//0
	bfs();
	
	for(int i=1;i<=num1;i++) printf("%d ",ans1[i]);
	puts("");
	for(int i=1;i<=num2;i++) printf("%d ",ans2[i]);
	return 0;
}



#include<bits/stdc++.h>

using namespace std;

const int N=1000010;
int h[N],ne[N],e[N],idx;
int n,m;
int q[N],d[N];
bool st[N];
vector<pair<int,int>> vv;

bool cmp(pair<int,int> a,pair<int,int> b)
{
	if(a.first!=b.first) return a.first>b.first;
	return a.second>b.second;
}

void add(int a,int b)
{
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

void dfs(int u)
{
	if(st[u]) return ;
	st[u]=true;
	printf("%d ",u);
	for(int i=h[u];i!=-1;i=ne[i])
	{
		int j=e[i];
		if(!st[j]) dfs(j);
	}
	return ;
}

void bfs()
{
	memset(d,-1,sizeof d);
	d[1]=0;
	printf("1 ");
	int hh=0,tt=0;
	q[hh]=1;
	
	while(hh<=tt)
	{
		int t=q[hh++];
		int num=0;
		for(int i=h[t];i!=-1;i=ne[i])
		{
			int j=e[i];
			if(d[j]==-1) 
			{
				d[j]=0;
				q[++tt]=j;
				printf("%d ",j);
			}
		}
	}
	return ;
}

int main()
{
	scanf("%d%d",&n,&m);
	memset(h,-1,sizeof h);
	for(int i=1;i<=m;i++)
	{
		int a,b;
		scanf("%d%d",&a,&b);
		vv.push_back({a,b});
	}
	sort(vv.begin(),vv.end(),cmp);
	for(auto it:vv) add(it.first,it.second);
	dfs(1);
	puts("");
	bfs();
	return 0;
}