题目:https://codeforces.com/contest/1095/problem/D
题意:
n个人参加舞会,每个人知道自己后面两个人是谁,求这个循环舞会的环顺序。
题解:DFS。
第一眼看到这个题目想到了暴力搜索,当遇到访问过的数字再剪枝,后来发现n的数量级为10的五次方,时间上来说应该是不过关的。
因为每个人知道自己后面的两个人是谁,举个例子,我们从1开始搜索,不妨设1知道的第一个人为x,第二个人为y,(此时不知道xy的顺序),如果x没被访问过且x是y的父节点(x知道y),就继续把x当下一个节点搜,否则把y当下一个节点搜。
以第一个测试例子作图,图中可以看到第一步从1开始,然后从3知道的两个人中找不到5,因此来到第二步,5为1的第一顺位,然后继续搜索,2知道的人没有3,来到第三步,目前的顺序为153,继续搜索,2知道的人中有4,便把2作为父节点,4作为子节点,来到第四步,目前的顺序为1532,继续走下去结果就出来啦!
当搜到只剩下一个节点的时候,要选取那个没有访问过的节点做最后一个节点,可以int c[maxn]作为一个标记数组,当访问过的节点设为1,没访问过的设为0。
#include<iostream> //P
#include<algorithm>
#include<string>
#include<cstring>
const int maxn=2e5+5;
using namespace std;
int a[maxn],b[maxn],c[maxn];
void DFS(int n,int m);
int p;
int main()
{
std::ios::sync_with_stdio(false);
memset(c,0,sizeof(c));
cin>>p;
for(int i=1;i<=p;i++){
cin>>a[i]>>b[i];
}
c[1]=1;
cout<<"1 ";
DFS(1,1);
}
void DFS(int n,int m)
{
if(n==p) return ;
int x=a[m];
int y=b[m];
if(!c[x]&&(a[x]==y||b[x]==y)){
c[x]=1;
cout<<x<<" ";
DFS(n+1,x);
}
else{
c[y]=1;
cout<<y<<" ";
DFS(n+1,y);
}
}