题意:给你一个图,图上每个点的信息表示该点的水流方向,若下一步走的方向与水流方向一致,则消耗能量为零,否则消耗能量为1,这样,给定起点和终点,问最少消耗的能量。
分析:比赛的时候,一开始交了一个思路很简单的代码,就是用优先队列实现,主要就是标记点的问题了,因为即使使用优先队列,如果当初将一个状态标记为0和1,那么有可能在后面存在“已访问的状态消耗的能量大于即将到达那个状态所消耗的能量”,这样,就无法保证消耗的能量最少了。所以,初始的时候,将每一个状态消耗的能量初始为无穷大,之后,当遇到到达该状态消耗更少能量的,则压入队列。这样就保证了消耗能量最少的问题了。但是,但我把这份很容易敲完的代码叫上去之后,2986ms,这时间虽然是没有TLE,但也差不多算没过了。
该怎么去优化呢?
想了好久,换了一下思路,将代码改了。
这样的,在每次压入一个点的是,我们可以顺便把与该点消耗能量相同的点(就是按照该状态下的水流方向走的点)压入队列,就能保证是先压入的点消耗的能量最少了,同时也就不需要用优先队列了,标记状态也只需要标记为0和1 就够了。
那怎么把能量相同点同时也压入队列呢,其实只需要深搜一下就行了。687ms就过了,但是,如果在找出那些点的过程中不用栈去保存,而是用数组的话,可以再快些。
#include<iostream>
#include<algorithm>
#include<queue>
#include<stack>
using namespace std;
struct node
{int x,y,energy;
};
bool vis[1001][1001];
char map[1001][1001];
int n,m,ans,si,sj,ei,ej;
int dir[8][2]={{-1,0},{-1,1},{0,1},{1,1},{1,0},{1,-1},{0,-1},{-1,-1}};
stack<node> ST;
void DFS(node temp)
{int k=map[temp.x][temp.y]-'0';int i=temp.x+dir[k][0];int j=temp.y+dir[k][1];node cur;if(i<0||i>=n||j<0||j>=m)return;if(!vis[i][j]){vis[i][j]=true;cur.x=i;cur.y=j;cur.energy=temp.energy;if(i==ei-1 && j==ej-1){ans=temp.energy;return ;}ST.push(cur);DFS(cur);}
}
void BFS()
{queue<node> Q;node f,temp,cur;f.x=si-1;f.y=sj-1;vis[f.x][f.y]=true;f.energy=0;Q.push(f);while(!Q.empty()){temp=Q.front();Q.pop();DFS(temp);while(!ST.empty()){Q.push(ST.top());ST.pop();}if(ans!=-1)return ;for(int k=0;k<8;k++){int i=temp.x+dir[k][0];int j=temp.y+dir[k][1];if(k==map[temp.x][temp.y]-'0')continue;if(i<0||i>=n||j<0||j>=m)continue;if(vis[i][j])continue;vis[i][j]=true;if(i==ei-1 && j==ej-1){ans=temp.energy+1;return ;}cur.x=i;cur.y=j;cur.energy=temp.energy+1;Q.push(cur);DFS(cur);while(!ST.empty()){Q.push(ST.top());ST.pop();}if(ans!=-1)return ;}}
}
int main()
{int k;while(scanf("%d %d",&n,&m)==2){for(int i=0;i<n;i++)scanf("%s",map[i]);scanf("%d",&k);while(k--){memset(vis,0,sizeof(vis));scanf("%d %d %d %d",&si,&sj,&ei,&ej);if(si==ei&&sj==ej){printf("0\n");continue;}ans=-1;BFS();printf("%d\n",ans);}}return 0;
}