我为这个问题做了这个实现: http://www.spoj.pl/problems/SHOP/
#include<iostream>
#include<stdio.h>
#include<queue>
#include<conio.h>
#include<string.h>
using namespace std;
struct node
{
int x;
int y;
int time;
};
bool operator <(const node &s,const node &r)
{
if(s.time>r.time)
return true;
else return false;
}
node beg,src,dest,tempa;
int b,a,temp;
int map[25][25];
bool vis[25][25];
int X[]={1,0,-1,0};
int Y[]={0,1,0,-1};
int djs_bfs(node src,node dest)
{
int result=0;
priority_queue<node>pq;
pq.push(src);
while(!pq.empty())
{
node top = pq.top();
pq.pop();
if(top.x==dest.x && top.y==dest.y) return result;
if(top.x<0 || top.x>=a) continue;
if(top.y<0 || top.y>=b) continue;
if(vis[top.x][top.y]) continue;
vis[top.x][top.y]=true;
result+=map[top.x][top.y];
for(int i=0;i<4;i++)
{
tempa.x=top.x+X[0];
tempa.y=top.y+Y[0];
tempa.time=map[top.x+X[0]][top.y+Y[0]];
pq.push(tempa);
}
}
return -1;
}
int main()
{
memset(vis,false,sizeof(vis));
scanf("%d %d",&a,&b);
while(a != 0)
{
for(int i=0;i<a;i++)
for(int j=0;j<b;j++)
{
scanf("%c",&temp);
if(temp=='X') {map[i][j]=0;vis[i][j]=true;}
if(temp=='S') {src.x=i;src.y=j;src.time=0;}
if(temp=='D') {dest.x=i;dest.y=j;dest.time=0;}
else map[i][j]=temp-'0';
}
cout<<djs_bfs(src,dest)<<endl;
scanf("%d %d",&a,&b);
}
return 0;
getch();
}
我不知道为什么我的代码没有为测试用例生成正确的答案。 如果有人可以帮我改进代码,请这样做:D
答案 0 :(得分:4)
首先,图解析代码不正确。第一行指定宽度和高度,其中宽度是每行的字符数,高度是行数。因此,在第一个scanf中交换&a
和&b
,或交换嵌套for
循环的顺序(但不是两者)。此外,我不得不在各个地方添加虚拟scanf("%c", &dummy);
调用来过滤掉换行符。像这样的简单转储将有助于确定您的地图是否被正确解析:
cout << "a=" << a << endl;
cout << "b=" << b << endl;
for (int i=0; i<a; i++) {
for(int j=0; j<b; j++) {
cout << (char)('0' + map[i][j]) << ",";
}
cout << endl;
}
注意:我还为{S'和'D'将map[i][j]
设置为0,同时将重复的if
语句更改为if; else if; else
链。这使得算法更加健壮,因为您可以通常从源或目标添加时间。
现在,关于算法本身......
算法的每个循环将result
增加当前地图位置权重。然而,该算法同时搜索多个路径(即,优先级队列中的条目数),因此result
最终是所有处理的节点权重的总和,而不是当前路径权重。当前路径权重为top.temp
,因此您可以删除result
变量,只需在到达目的地时返回top.temp
。
此外,正如其他答案所指出的那样,您需要在内部循环中使用X[i]
和Y[i]
,否则您只会向一个方向搜索。
现在,由于X[i]
和Y[i]
的加/减,您可能会超出范围(-1或25)访问map[][]
。因此,我建议将if
防护移动到内部for
循环以防止超出范围的访问。这也避免了填充优先队列的非法可能性。
这是我的算法版本,只有很少的更正,供参考:
priority_queue<node>pq;
pq.push(src);
while(!pq.empty())
{
node top = pq.top();
pq.pop();
if(top.x==dest.x && top.y==dest.y) return top.time;
if(vis[top.x][top.y]) continue;
vis[top.x][top.y]=true;
for(int i=0;i<4;i++)
{
tempa.x=top.x+X[i];
tempa.y=top.y+Y[i];
if(tempa.x<0 || tempa.x>=a) continue;
if(tempa.y<0 || tempa.y>=b) continue;
tempa.time=top.time + map[tempa.x][tempa.y];
pq.push(tempa);
}
}
return -1;
我希望这会有所帮助。
答案 1 :(得分:2)
您在内部循环中使用X[0]
和Y[0]
而不是X[i]
和Y[i]
。
顺便说一句,除此之外,你的Dijkstra效率非常低。首先,即使已经访问过节点,您也会将节点推送到队列中;其次,您可能在队列中有几个相同的节点,只是时间不同。最终这些都不会影响结果,但是你正在改变复杂性。
编辑:哦,tempa.time
应该等于top.time
加上边缘权重,而不仅仅是边缘权重。
答案 2 :(得分:2)
为什么你有0个索引?
tempa.x=top.x+X[0];
tempa.y=top.y+Y[0];
tempa.time=map[top.x+X[0]][top.y+Y[0]];
挑剔:
bool operator <(const node &s,const node &r)
{
if(s.time>r.time)
return true;
else return false;
}
这不是更具可读性:
bool operator <(const node &s,const node &r)
{
return (s.time>r.time);
}