输入一个数值n,表示一块nxn的区域,其中数值1表示信号强,0表示信号弱,例如:
3
1 0 1
1 1 1
1 1 1
寻找所有信号强的路径(不包含0)中的最短路径,并输出其最短路径的值,如果不存在最短路径(到不了右下角),则输出-1。
思路:将整个nxn数组建立成一个二维图,利用广度优先搜索算法进行搜索,直到命中右小角元素。
代码如下:
#include <iostream>
#include <vector>
#include <queue>//用队列实现广度优先搜索
using namespace std;
struct Node//图的节点
{
bool tovisit;//该节点是否将要被访问
int val, x, y;//存储该节点的值以及坐标
Node *upper,*left,*down,*right;//存储该节点周围的情况
Node():tovisit(false),val(-1),upper(NULL), left(NULL), down(NULL), right(NULL){}
};
int main()
{
int n;
cin>>n;
vector<Node> a(n);
vector<vector<Node>> A(n, a);//注意初始化第一个参数为元素个数,第二个参数为数据类型
for(int i=0; i<n; i++){
for(int j=0; j<n; j++){
cin>>A[i][j].val;
if(i-1>=0)
A[i][j].upper=&A[i-1][j];
if(j-1>=0)
A[i][j].left=&A[i][j-1];
if(i+1<=n-1)
A[i][j].down=&A[i+1][j];
if(j+1<=n-1)
A[i][j].right=&A[i][j+1];
A[i][j].x=i;
A[i][j].y=j;
}
}//给所有节点赋值,并且连接其上下左右的节点
int dis=-1;
queue<Node*> S;//S中存储节点的位置(Node*)
if(A[0][0].val==0)
return -1;
S.push(&A[0][0]);
A[0][0].tovisit=true;
while(!S.empty()){//队列非空
int length=S.size();
for(int i=0; i<length; i++){//广度优先搜索
Node* cur=S.front();
S.pop();
if(cur->x==n-1&&cur->y==n-1){
cout<<dis+1;
system("pause");
return 0;
}
/*将下一层节点加入到队列中*/
if(cur->upper&&!cur->upper->tovisit&&cur->upper->val==1){
S.push(cur->upper);
cur->upper->tovisit=true;
}
if(cur->left&&!cur->left->tovisit&&cur->left->val==1){
S.push(cur->left);
cur->left->tovisit=true;
}
if(cur->down&&!cur->down->tovisit&&cur->down->val==1){
S.push(cur->down);
cur->down->tovisit=true;
}
if(cur->right&&!cur->right->tovisit&&cur->right->val==1){
S.push(cur->right);
cur->right->tovisit=true;
}
}//遍历完每一层后
dis++;//将距离+1
}
cout<<-1;
system("pause");
return 0;
}