【UVALive 7364】Robots（逆向思维+搜索）

【UVALive 7364】Robots（逆向思维+搜索）

题目大意：

n个机器人在坐标0~n-1上。

有两个按钮，可以让每个位置上的机器人到该按钮指定的点去（∈[0,n−1])。

问是否可以经过几次操作，让所有机器人到一个点上。

考虑最终状态是所有机器人在一个点，考虑两两机器人相互关系，从状态(i,i)开始逆向搜索，搜到的状态(a,b)经过多次A或B操作能达到(i,i)。

那么遍历所有机器人对，如果两两可以到达同一个点，最终一定有方式全部到一个点。

所以就BFS一下就行了。。一开始用的DFS结果TLE了……

代码如下：

#include <iostream>
#include <cmath>
#include <vector>
#include <cstdlib>
#include <cstdio>
#include <climits>
#include <ctime>
#include <cstring>
#include <queue>
#include <stack>
#include <list>
#include <algorithm>
#include <map>
#include <set>
#define LL long long
#define Pr pair<int,int>
#define fread(ch) freopen(ch,"r",stdin)
#define fwrite(ch) freopen(ch,"w",stdout)

using namespace std;
const int INF = 0x3f3f3f3f;
const int mod = 1e9+7;
const double eps = 1e-8;
const int maxn = 1123;

vector <int> vc[2][maxn];
bool vis[maxn][maxn];

void bfs(int a,int b)
{
queue <Pr> q;
q.push(Pr(a,b));

Pr p;
vis[a][b] = vis[b][a] = 1;

while(!q.empty())
{
p = q.front();
q.pop();

a = p.first;
b = p.second;

for(int i = 0; i < 2; ++i)
for(int j = 0; j < vc[i][a].size(); ++j)
for(int k = 0; k < vc[i][b].size(); ++k)
if(!vis[vc[i][a][j]][vc[i][b][k]])
{
vis[vc[i][a][j]][vc[i][b][k]] = vis[vc[i][b][k]][vc[i][a][j]] = 1;
q.push(Pr(vc[i][b][k],vc[i][a][j]));
}
}

}

bool solve(int n)
{
int x;
for(int j = 0; j < 2; ++j)
for(int i = 0; i < n; ++i)
vc[j][i].clear();

for(int j = 0; j < 2; ++j)
for(int i = 0; i < n; ++i)
{
scanf("%d",&x);
vc[j][x].push_back(i);
}

memset(vis,0,sizeof(vis));
for(int i = 0; i < n; ++i)
bfs(i,i);

for(int i = 0; i < n; ++i)
for(int j = 0; j < n; ++j)
if(!vis[i][j])
return false;

return true;
}

int main()
{
//fread("");
//fwrite("");

int t,k,n;

scanf("%d",&t);

while(t--)
{
scanf("%d%d",&k,&n);
printf("%d %s\n",k,solve(n)? "YES": "NO");
}

return 0;
}