一、题目要求
约翰的农场有 n 头奶牛,每一头奶牛都有一个正整数编号。
不同奶牛的编号不同。
现在,这 n头牛按某种顺序排成一队,每头牛都拿出一张纸条写下了其前方相邻牛的编号以及其后方相邻牛的编号。
注意:
- 这些奶牛并没有记下自己的编号。
- 位于队首的奶牛前方没有牛,所以它在前方相邻牛处写下的是数字 0。
- 位于队尾的奶牛后方没有牛,所有它在后方相邻牛处写下的是数字 0。
将所有奶牛写下的纸条收集起来并打乱顺序后交给你。
你的任务是根据这些纸条信息,推导出完整的奶牛队列。
输入格式
第一行包含整数 n,表示奶牛数量。
接下来 n行,每行包含两个整数 ai,bi,表示其中一头奶牛写下的其前方相邻牛的编号以及其后方相邻牛的编号。
注意,ai或 bi 可能为 0,这表示该奶牛没有前方相邻牛或后方相邻牛。
输出格式
输出共一行,n 个整数,按照从前到后的顺序输出队列中每头奶牛的编号。
数据范围
前 55 个测试点满足 2≤n≤5。
所有测试点满足 2≤n≤2×10^5,0≤ai,bi≤10^6。
输入样例:
4
92 31
0 7
31 0
7 141
输出样例:
92 7 31 141
二、思路
1.利用并查集
看成两条链,0是偶数位置的开头,u是奇数位置的开头
0<-7<-141;//0偶数位置开头
92<-31<-0;//u奇数位置开头
2.利用dfs();
这道题可以看成两条链,一条奇数链,一条偶数链,分别从每条链的根节点深搜
三、代码
1.并查集
#include<bits/stdc++.h>
#define endl '\n'
#define int long long
#define IOS ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
using namespace std;
const int N=1e6+10;
const int inf=0x3f3f3f3f;
int n;
int a[N],f[N],h[N];
bool k[N];
int find(int x)//方便查找头结点
{
if(x==f[x])
return x;
return f[x]=find(f[x]);
}
void solve()
{
cin>>n;
int i,j,u;
for(i=0;i<N;i++)
{
f[i]=i;
}
for(i=1;i<=n;i++)
{
int l,r;
cin>>l>>r;
k[l]=true,k[r]=true;
h[l]=r;
int x=find(l);
int y=find(r);
f[y]=x;
}
for(j=2,i=h[0];i!=0;i=h[i],j+=2)//偶数位置
{
a[j]=i;
}
for(i=1;i<=1e6;i++)
{
if(k[i]==true&&find(i)!=0)
{
u=find(i);//找第一个奇数位置上的数字
}
}
for(j=1,i=u;i!=0;i=h[i],j+=2)//奇数位置
{
a[j]=i;
}
for(i=1;i<=n;i++)
{
cout<<a[i]<<" ";
}
cout<<endl;
}
signed main()
{
int t=1;
IOS;
while(t--)
{
solve();
}
return 0;
}
/*
看成两条链
92<-31<-0;//92是奇数位置的开头
0<-7<-141;//0是偶数位置的开头
*/
2.dfs
#include<bits/stdc++.h>
#define endl '\n'
#define int long long
#define IOS ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
using namespace std;
const int N=1e6+10;
int n;
int a[N],b[N];
int h[N],e[N],ne[N],idx;
int x[N],c[N],ans=2;
void add(int a,int b)
{
e[idx]=b;
ne[idx]=h[a];
h[a]=idx++;
}
void dfs(int u,int fa)
{
int i;
for(i=h[u];~i;i=ne[i])
{
int j=e[i];
//cout<<"j="<<j<<" "<<"u="<<u<<endl;
if(j==fa||j==0)//j==0,是偶数链从0断开
continue;
c[ans]=j;
ans+=2;
dfs(j,u);
}
}
void solve()
{
cin>>n;
memset(h,-1,sizeof(h));
int i;
for(i=1;i<=n;i++)
{
cin>>a[i]>>b[i];
add(a[i],b[i]);
x[a[i]]++;
x[b[i]]++;
}
dfs(0,-1);//偶数链根节点从0开始
for(i=1;i<=n;i++)
{
if(x[a[i]]==1)//寻找另一个根节点,这道题给出的a,b两点都是从前指向后的,所以根节点一定在a上
{ //x[a]==1时,只有一条边相连即为根节点
{
c[1]=a[i];
ans=3;
dfs(a[i],-1);
break;
}
}
for(i=1;i<=n;i++)
{
cout<<c[i]<<' ';
}
cout<<endl;
}
signed main()
{
int t=1;
IOS;
while(t--)
{
solve();
}
return 0;
}
/*
ne[0]=1 ne[1]=2 ne[2]=3 ne[3]=4;
e[0]=31 e[1]=7 e[2]=0 e[3]=141
h[92]=1 h[0]=2 h[31]=3 h[7]=4
*/
ps:以上代码,来自其他大佬题解