Poj 1708 Game【题解报告|启发式DFS】

题目链接

题目大意

给出N个点（不超过100个），每个点有一个颜色（颜色标号不超过100），有一些点的颜色可以相同。给出M条有向边，这些边也有一个颜色。给出两个棋子的初始点L,K，终点Q，问两个棋子中的任意一个是否可以移动到Q，如果可以，输出移动的最小步数。移动时遵循以下规则：

1.一个棋子可以移动到下一个点需要满足：经过的边的颜色与另一个棋子所在点的颜色相同；

2.只能沿着有向边的方向；

3.两个棋子不能同时移动到同一个点；

4.移动次序任意，没有必要交替进行；

5.其中一个棋子到达Q点时游戏结束。

思路分析

拿到题比较容易想到要搜索，但是怎么搜是一个问题，因为有两个点同时可以动，而且一个点能否前进受到另一个点的掣肘。如何建模这件事情是这道题的难点，而且搜索的图中，我们一定要避免重复状态的出现，而在这里，我们不能简单的访问过一个点就给他打上标记，因为即使在同一个点，两个pawn的位置不同得到的结果也是不同的。

关键在于pawn的位置，所以我们不妨将pawn的位置作为搜索的状态，

struct step {//dfs所用的状态
	int v1, v2, cnt;//第一个pawn和第二个pawn的状态，以及所用的步数
	step(int a = 0, int b = 0, int c = 0) { v1 = a, v2 = b, cnt = c; }
};

那么重复状态就定义为，v1,v2重复。如果此时我们访问到一个点n，那么我们有两种前进的方式，一种是第一个pawn前进，一种是第二个pawn前进，分别对这两种方式进行搜索，保证题目中的条件得到满足

#define ll long long
#define vec vector<int>
#define inf 0x3f3f3f3f
#define P pair<string,string>
#define MAX 105

struct edge {//边
	int to, c;
	edge(int a = 0, int b = 0) { to = a, c = b; }
};
struct step {//dfs所用的状态
	int v1, v2, cnt;//第一个pawn和第二个pawn的状态，以及所用的步数
	step(int a = 0, int b = 0, int c = 0) { v1 = a, v2 = b, cnt = c; }
};

vector<edge> G[MAX];//图
int N, L, K, Q, col[MAX], A, B, C, M, vis[MAX][MAX], res;

void bfs() {
	memset(vis, 0, sizeof(vis));
	vis[L][K] = 1; queue<step> q; q.push(step(L, K, 0));

	while (!q.empty()) {
		int v1 = q.front().v1, v2 = q.front().v2, num = q.front().cnt;
		q.pop();
		if ((v1 == Q || v2 == Q) && num < res)res = num;//更新答案

		//第一个点向前搜索
		for (int i = 0; i < G[v1].size(); i++) {
			edge e = G[v1][i];
			//该状态未曾访问，边的颜色与v2相等，不会到同一个点中
			if (!vis[e.to][v2] && e.c == col[v2] && e.to != v2) {
				vis[e.to][v2] = 1;
				q.push(step(e.to, v2, num + 1));
			}
		}

		//第二个点向前搜索
		for (int i = 0; i < G[v2].size(); i++) {
			edge e = G[v2][i];
			if (!vis[v1][e.to] && e.c == col[v1] && e.to != v1) {
				vis[v1][e.to] = 1;
				q.push(step(v1, e.to, num + 1));
			}
		}
	}
}

int main() {
	while (scanf("%d %d %d %d", &N, &L, &K, &Q) != EOF) {
		for (int i = 1; i <= N; i++)scanf("%d", &col[i]);
		scanf("%d", &M);
		for (int i = 0; i < M; i++) {
			scanf("%d %d %d", &A, &B, &C);
			G[A].push_back(edge(B, C));
		}
		
		res = inf; bfs();
		if (res == inf)printf("NO\n");
		else printf("YES\n%d\n", res);
	}
}