BFS 吝啬的国度

题意

 

 

输入

第一行输入一个整数M表示测试数据共有M(1<=M<=5)组
每组测试数据的第一行输入一个正整数N(1<=N<=100000)和一个正整数S(1<=S<=100000)，N表示城市的总个数，S表示参观者所在城市的编号
随后的N-1行，每行有两个正整数a,b(1<=a,b<=N)，表示第a号城市和第b号城市之间有一条路连通。

输出

每组测试数据输N个正整数，其中，第i个数表示从S走到i号城市，必须要经过的上一个城市的编号。（其中i=S时，请输出-1）

样例输入

1
10 1
1 9
1 8
8 10
10 3
8 6
1 2
10 4
9 5
3 7

样例输出

-1 1 10 10 9 8 3 1 1 8

描述

在一个吝啬的国度里有N个城市，这N个城市间只有N-1条路把这个N个城市连接起来。现在，Tom在第S号城市，他有张该国地图，他想知道如果自己要去参观第T号城市，必须经过的前一个城市是几号城市（假设你不走重复的路）。

思路

BFS 超时了，，，，，，

code

#include<stdio.h>
#include<string.h>
#include<queue>
using namespace std;

int a[100005][2];
int maze[100005][2];

int N,S;
        //  i     S
int bfs(int n,int s)
{
	queue<int> Q;
	
	Q.push(s);  // 1

	for(int i=0;i<N;i++)
	{
		if(a[i][0]==s)
		maze[i][0]=-1;
		else if(a[i][1]==s)
		maze[i][1]=-1;
	}
	
	while(!Q.empty())
	{
		int t;
		int flag=0;
		t = Q.front(); Q.pop();
		
		for(int i=0;i<N;i++)
		{
			if(t==a[i][0] && maze[i][1]!=-1)
			{
				Q.push(a[i][1]);
				maze[i][1]=-1;
				
				if(a[i][1]==n)
				{
					return t;
					flag=1;
					break;
				}
			}
			else if(t==a[i][1] && maze[i][0]!=-1)
			{
				Q.push(a[i][0]);
				maze[i][0]=-1;
				
				if(a[i][0]==n)
				{
					return t;
					flag=1;
					break;
				}
			}
		}
		
		if(flag==1)
		{
			break;
		}
	}
	while(!Q.empty()) Q.pop();
}

int main()
{
	int M;
	
	while(scanf("%d",&M)!=EOF)
	{
		while(M--)
		{
		    scanf("%d%d",&N,&S);
			
			for(int i=0;i<N-1;i++)
			{
				scanf("%d%d",&a[i][0],&a[i][1]);
		    }
		    
		    for(int i=1;i<=N;i++)
		    {
		    	memset(maze,0,sizeof(maze));
		    	
		    	if(i==S)
		    	printf("-1 ");
		    	else
		    	printf("%d ",bfs(i,S));
			}
			printf("\n");
		}
	}
	return 0;
}

换了个写法，超时，，，，，，

#include<stdio.h>
#include<queue>
#include<string.h>
using namespace std;

int pre[100005];
int a[100005][2];
int maze[100005][2];
int N,S;

void preing()
{
	for(int i=1;i<=N;i++)
	{
		pre[i]=i;
	}
}

void init(int n)
{
	queue<int> Q;
	Q.push(n);
	
	for(int i=1;i<N;i++)
	{
		if(a[i][0]==n)
		maze[i][0]=-1;
		else if(a[i][1]==n)
		maze[i][1]=-1;
	}
	
	while(!Q.empty())
	{
		int t;
		t=Q.front();Q.pop();
				
		for(int i=1;i<N;i++)
	    {   
		    if(a[i][0]==t && maze[i][1]!=-1)
		    {
			    Q.push(a[i][1]);
			    maze[i][1]=-1;
			    pre[a[i][1]]=a[i][0];
				
				for(int j=1;j<N;j++)
	            {            
		            if(a[j][0]==a[i][1])
		            maze[j][0]=-1;
		            else if(a[j][1]==a[i][1])
		            maze[j][1]=-1;
	            }
				
			//	printf("%d *** %d *** %d 上\n",a[i][0],pre[a[i][1]],a[i][1]);
		    }
		    else if(a[i][1]==t && maze[i][0]!=-1)
		    {
			    Q.push(a[i][0]);
			    maze[i][0]=-1;
			    pre[a[i][0]]=a[i][1];   
				
				for(int j=1;j<N;j++)
	            {            
		            if(a[j][0]==a[i][0])
		            maze[j][0]=-1;
		            else if(a[j][1]==a[i][0])
		            maze[j][1]=-1;
	            }
				
			//	printf("%d *** %d *** %d 下\n",a[i][1],pre[a[i][0]],a[i][0]);
		    }
	    }
	}
	while(!Q.empty()) Q.pop();
}

int find(int n)
{
	return pre[n];
}

int main()
{
	int M;
	
	while(scanf("%d",&M)!=EOF)
	{
		while(M--)
		{
			scanf("%d%d",&N,&S);
			
			for(int i=1;i<N;i++)
			{
				scanf("%d%d",&a[i][0],&a[i][1]);
			}
			
			preing();
			memset(maze,0,sizeof(maze));
			init(S);
			
			for(int i=1;i<=10;i++)
			{
				
				if(i==S)
				printf("-1 ");
				else
				{
					printf("%d ",find(i));
				} 			
			}
			printf("\n");
		}
	}
	return 0;
}
/*
1
10 1
1 9
1 8
8 10
10 3
8 6
1 2
10 4
9 5
3 7
2 *** 2 *** 1 ?
1 *** 1 *** 9 ?
1 *** 1 *** 8 ?
9 *** 9 *** 1 ?
9 *** 9 *** 5 ?
8 *** 8 *** 1 ?
8 *** 8 *** 10 ?
8 *** 8 *** 6 ?
5 *** 5 *** 9 ?
10 *** 10 *** 8 ?
10 *** 10 *** 3 ?
10 *** 10 *** 4 ?
6 *** 6 *** 8 ?
3 *** 3 *** 10 ?
3 *** 3 *** 7 ?
4 *** 4 *** 10 ?
7 *** 7 *** 3 ?
2
*/

心怀希望，就能永远有希望。