挑战ACM迷宫(DFS or BFS+优先队列(A*))
题目描述
如下图所示的是一个由程序设计题目组成的ACM迷宫。迷宫的左上角是入口,右下角是出口。迷宫中每一个格子都有一个程序设计题目,挑战者要AC该题目后才能通过,大于0的数字表示AC该题目所需的最短时间。数字如果是0表示是陷阱,进去了就出不来。现在的问题是:求挑战者从入口到出口所需的最短时间。
输入
有多组测试实例。
对于每组测试实例,先输入一个数字n(1<n<=100),然后输入n*n个数字表示迷宫中的数字(数字的值不超过1000)。
输出
对应输出挑战者从入口到出口所需的最短时间。
样例输入
10
1 0 2 1 2 3 4 2 2 5
2 1 0 1 3 2 5 7 2 1
1 1 1 0 1 1 1 3 2 3
1 2 1 1 0 1 1 1 2 3
1 1 2 0 2 0 2 3 2 3
2 2 2 2 3 2 0 2 3 2
3 2 2 2 1 1 1 0 1 1
0 1 1 3 0 1 1 2 3 2
2 0 1 1 2 2 2 2 2 2
3 2 3 2 3 2 3 2 3 2
5
1 2 1 2 5
1 3 2 4 5
2 1 0 2 5
2 1 1 2 1
2 4 1 1 2
样例输出
min=29
min=11
思路来源:https://blog.csdn.net/xMaii/article/details/75429921
首先使用单纯的DFS,很显然会超时,贴出代码。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<queue>
#include<cmath>
#include<algorithm>
using namespace std;
int n,cnt;
int visit[101][101];
int acm[101][101];
int fx[4][2] = { {1,0},{-1,0},{0,1},{0,-1} };
int check(int a, int b)
{
if (a > 0 && a <= n && b > 0 && b <= n)
return 1;
else
return 0;
}
void DFS(int a,int b,int time)
{
if (a == n && b == n)
{
cnt = min(time + acm[a][b], cnt);
return;
}
if (time >= cnt)
return;
for (int i = 0; i < 4; i++)
{
if (check(a + fx[i][0], b + fx[i][1]) && acm[a + fx[i][0]][b + fx[i][1]] && !visit[a + fx[i][0]][b + fx[i][1]])
{
visit[a + fx[i][0]][b + fx[i][1]] = 1;
DFS(a + fx[i][0], b + fx[i][1], time+acm[a][b]);
visit[a + fx[i][0]][b + fx[i][1]] = 0;
}
}
}
int main()
{
while (scanf("%d", &n) != EOF)
{
memset(visit, 0, sizeof(visit));
cnt = 9999999;
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
scanf("%d", &acm[i][j]);
visit[1][1] = 1;
DFS(1,1,0);
printf("min=%d\n",cnt);
}
return 0;
}
于是通过DFS+剪枝
具体思路请看:https://blog.csdn.net/xMaii/article/details/75429921
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<queue>
#include<cmath>
#include<algorithm>
using namespace std;
int n,cnt;
int visit[101][101];
int acm[101][101];
int time1[101][101];
int fx[4][2] = { {1,0},{-1,0},{0,1},{0,-1} };
int check(int a, int b)
{
if (a > 0 && a <= n && b > 0 && b <= n)
return 1;
else
return 0;
}
void DFS(int a,int b,int time)
{
if (a == n && b == n)
{
cnt = min(time + acm[a][b], cnt);
return;
}
if (time >= cnt||time>=time1[a][b])
return;
time1[a][b] = time;
for (int i = 0; i < 4; i++)
{
if (check(a + fx[i][0], b + fx[i][1]) && acm[a + fx[i][0]][b + fx[i][1]] && !visit[a + fx[i][0]][b + fx[i][1]])
{
visit[a + fx[i][0]][b + fx[i][1]] = 1;
DFS(a + fx[i][0], b + fx[i][1], time+acm[a][b]);
visit[a + fx[i][0]][b + fx[i][1]] = 0;
}
}
}
int main()
{
while (scanf("%d", &n) != EOF)
{
memset(visit, 0, sizeof(visit));
memset(time1, 9999999, sizeof(time1));
cnt = 9999999;
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
scanf("%d", &acm[i][j]);
visit[1][1] = 1;
DFS(1,1,0);
printf("min=%d\n",cnt);
}
return 0;
}
能够缩减一部分时间,但是,依旧超时,从超时百分之40变成超时百分之10.
所以,改用第二种方法,BFS+优先队列(A*)算法。
思路来源:https://blog.csdn.net/deepseazbw/article/details/75650752
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<queue>
#include<cmath>
#include<algorithm>
using namespace std;
int n, cnt;
int visit[101][101];
int acm[101][101];
int fx[4][2] = { {1,0},{-1,0},{0,1},{0,-1} };
struct Node
{
int x, y, sum;
bool operator<(const Node& a)const
{
return sum > a.sum;
}
};
int check(int a, int b)
{
if (a > 0 && a <= n && b > 0 && b <= n&&acm[a][b])
return 1;
else
return 0;
}
void BFS(int a, int b)
{
priority_queue<Node> Q;
Node xw,qw;
xw.x = a; xw.y = b; xw.sum = acm[1][1];
Q.push(xw);
visit[1][1] = 1;
while (!Q.empty())
{
xw = Q.top(); Q.pop();
if (xw.x == n && xw.y == n)
{
cnt = xw.sum;
return;
}
for (int i = 0; i < 4; i++)
{
if (check(xw.x + fx[i][0], xw.y + fx[i][1]) && !visit[xw.x + fx[i][0]][xw.y + fx[i][1]])
{
qw.x = xw.x + fx[i][0], qw.y = xw.y + fx[i][1], qw.sum = xw.sum+ acm[xw.x + fx[i][0]][xw.y + fx[i][1]];
Q.push(qw);
visit[xw.x + fx[i][0]][xw.y + fx[i][1]] = 1;
}
}
}
}
int main()
{
while (scanf("%d", &n) != EOF)
{
memset(visit, 0, sizeof(visit));
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
scanf("%d", &acm[i][j]);
BFS(1, 1);
printf("min=%d\n", cnt);
}
}