链接:登录—专业IT笔试面试备考平台_牛客网
题目描述:
与普通的dfs的区别就是,它的移动速度是不定的,所以我们不能使用单纯的queue,而可以使用优先队列,因为队列存在的意义,就是要让各个能走的点,以到起点的时间(不是距离,因为速度不定)升序依次处理,那么就应该使用优先队列,来保证队列输出的顺序,是以时间为标准的。
这是第一代代码:
#include<bits/stdc++.h>
using namespace std;
int t,n,k;
struct ty{
int x,y,step;
bool operator<(const ty& s)const{
return step>s.step;
}
};
bool vis[310][310];
ty star,ed;
int dir[][4]={{1,0},{-1,0},{0,1},{0,-1}};
char mas[310][310];
bool inmap(int x,int y){
return x>=0&&y>=0&&x<n&&y<n;
}
void bfs(){
memset(vis,0,sizeof(vis));
priority_queue<ty> q;
q.push(star);
vis[star.x][star.y]=1;
while(!q.empty()){
ty t=q.top();
q.pop();
int x=t.x,y=t.y;
if(mas[x][y]=='X'&&t.step<=k) {cout<<"YES"<<endl<<t.step<<endl;return;}
for(int i=0;i<4;++i){
int xx=x+dir[i][0];
int yy=y+dir[i][1];
if(inmap(xx,yy)&&!vis[xx][yy]&&mas[xx][yy]!='#'){
if(mas[xx][yy]=='C'){
vis[xx][yy]=1;
q.push({xx,yy,t.step+1});
}
else{
vis[xx][yy]=1;
q.push({xx,yy,t.step+2});
}
}
}
}
cout<<"NO"<<endl;
}
int main()
{ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
cin>>t;
while(t--){
cin>>n>>k;
for(int i=0;i<n;++i){
for(int j=0;j<n;++j){
cin>>mas[i][j];
if(mas[i][j]=='S') star={i,j};
if(mas[i][j]=='X') ed={i,j};
}
}
bfs();
}
return 0;
}
它wa了。。。
其实还有一个问题我们没有考虑。因为对于每个点,我们其实有两种状态,即:它是我们走过来的,或者它是我们开车过来的,这其实是可以同时存在的,因为我们的速度并非一成不变,我们完全可以先走着经过这里,拿到车之后再开过这里,虽然一个点踩了2次,但之后的路我们却有了更快的速度。当然,如果我们是开着车两次经过同一个点,那肯定还是没有必要的。所以这里的vis数组,我们应该多开一维,记录点的同时,也要记录来时的状态,是开车还是走路,当然也可以直接用速度来记录,后面也方便直接计算时间。
AC代码:
#include<bits/stdc++.h>
using namespace std;
int t,n,k;
struct ty{
int x,y,sp;
long long step;
bool operator<(const ty& s)const{
return step>s.step;
}
};
bool vis[310][310][3];
ty star,ed;
int dir[][4]={{1,0},{-1,0},{0,1},{0,-1}};
char mas[110][110];
bool inmap(int x,int y){
return x>=0&&y>=0&&x<n&&y<n;
}
void bfs(){
memset(vis,0,sizeof(vis));
priority_queue<ty> q;
q.push(star);
vis[star.x][star.y][star.sp]=1;
while(!q.empty()){
ty t=q.top();
q.pop();
int x=t.x,y=t.y,z=t.sp;
if(mas[x][y]=='X'&&t.step<=k) {cout<<"YES"<<endl<<t.step<<endl;return;}
for(int i=0;i<4;++i){
int xx=x+dir[i][0];
int yy=y+dir[i][1];
int zz=z;
int step=t.step;
//if(mas[xx][yy]=='C') zz=1;
if(mas[xx][yy]!='O'&&!vis[xx][yy][zz]&&inmap(xx,yy)){
vis[xx][yy][zz]=1;
step+=z;
if(mas[xx][yy]=='C') zz=1;
q.push({xx,yy,zz,step});
}
}
}
cout<<"NO"<<endl;
return;
}
int main()
{ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
cin>>t;
while(t--){
cin>>n>>k;
for(int i=0;i<n;++i){
for(int j=0;j<n;++j){
cin>>mas[i][j];
if(mas[i][j]=='S') star={i,j,2,0};
}
}
bfs();
}
return 0;
}