HDOJ 1180 诡异的楼梯 DFS

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1180

用DFS很容易超时，尽管题中已经考虑了剪枝，还是超了 1000MS，这题有BFS的高效解法，不过感觉时间限制太小了，1000MS就超了，上次做的题，用BFS 2000MS还能

AC啊，先上代码，思路：重点考虑到达楼梯周围四个点时候的楼梯方向和四个位置，依次枚举。

#include <iostream>
#include <string>
#include <cstdio>
#include <cmath>
#include <vector>
#include <algorithm>
#include <sstream>
#include <cstdlib>
#include <fstream>
using namespace std;
int dis(int a,int b,int x,int y){
	return abs(a-x)+abs(b-y);
}
char maze[25][25];
bool visit[25][25],flag;
int dir[4][2] = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};
int stairs;
int n,m,sx,sy,endx,endy,tsum,tmin;
void dfs(int sx,int sy){
	if(maze[sx][sy]=='T'){
		flag=1;
		if(tsum<tmin)tmin=tsum;
		return ;
	}
	if(maze[sx][sy]=='*')return ;
	
	if(sx<1||sx>m||sy<1||sy>n)return ;
	
	if(tsum>tmin)return ;
	
	
	
//	cout<<"sx: "<<sx<<" "<<"sy: "<<sy<<" "<<"tsum: "<<tsum<<endl;
	
	for(int i=0;i<4;i++)
	{
		if(sx+dir[i][0]<1||sx+dir[i][0]>m||sy+dir[i][1]<1||sy+dir[i][1]>n)continue;
		if(maze[sx+dir[i][0]][sy+dir[i][1]]=='*')continue;
		if(visit[sx+dir[i][0]][sy+dir[i][1]])continue; 
		
		if(dis(sx+dir[i][0],sy+dir[i][1],endx,endy)>dis(sx,sy,endx,endy))continue;
		//剪枝：如果方向背离目标，即距离变大，则剪掉 
		
		
		if(maze[sx+dir[i][0]][sy+dir[i][1]]=='.'||maze[sx+dir[i][0]][sy+dir[i][1]]=='T'){
			tsum++;
			visit[sx+dir[i][0]][sy+dir[i][1]]=1;
			dfs(sx+dir[i][0],sy+dir[i][1]);
			visit[sx+dir[i][0]][sy+dir[i][1]]=0;
			tsum--;
		}
		if(maze[sx+dir[i][0]][sy+dir[i][1]]=='|'||maze[sx+dir[i][0]][sy+dir[i][1]]=='-')
		{
			
				//考虑到达楼梯前的两种情况和四个方位			
				if(tsum%2==0&&stairs==1 || tsum%2==1&&stairs==0){
					if(i==0){     //楼梯在上方，此时为 ‘|’ 
						tsum++;
						visit[sx+dir[i][0]][sy+dir[i][1]]=1;
						visit[sx+dir[i][0]-1][sy+dir[i][1]]=1;
						dfs(sx+dir[i][0]-1,sy+dir[i][1]);
						visit[sx+dir[i][0]][sy+dir[i][1]]=0;
						visit[sx+dir[i][0]-1][sy+dir[i][1]]=0;
						tsum--;
					}
					if(i==1){   //楼梯在下方 
						tsum++;
						visit[sx+dir[i][0]][sy+dir[i][1]]=1;
						visit[sx+dir[i][0]+1][sy+dir[i][1]]=1;
						dfs(sx+dir[i][0]+1,sy+dir[i][1]);
						visit[sx+dir[i][0]][sy+dir[i][1]]=0;
						visit[sx+dir[i][0]+1][sy+dir[i][1]]=0;
						tsum--;
					}
					if(i==2){
						tsum+=2;
						visit[sx+dir[i][0]][sy+dir[i][1]]=1;
						visit[sx+dir[i][0]][sy+dir[i][1]-1]=1;
						dfs(sx+dir[i][0],sy+dir[i][1]-1);
						visit[sx+dir[i][0]][sy+dir[i][1]]=0;
						visit[sx+dir[i][0]][sy+dir[i][1]-1]=0;
						tsum-=2;
					}
					if(i==3){
						tsum+=2;
						visit[sx+dir[i][0]][sy+dir[i][1]]=1;
						visit[sx+dir[i][0]][sy+dir[i][1]+1]=1;
						dfs(sx+dir[i][0],sy+dir[i][1]+1);
						visit[sx+dir[i][0]][sy+dir[i][1]]=0;
						visit[sx+dir[i][0]][sy+dir[i][1]+1]=0;
						tsum-=2;
					}
				
				} 
				 //此时到达楼梯前，楼梯方向为‘-’ 
				else if(tsum%2==1&&stairs==1 ||tsum%2==0&&stairs==0 ){
					if(i==0){
						tsum+=2;
						visit[sx+dir[i][0]][sy+dir[i][1]]=1;
						visit[sx+dir[i][0]-1][sy+dir[i][1]]=1;
						dfs(sx+dir[i][0]-1,sy+dir[i][1]);
						visit[sx+dir[i][0]][sy+dir[i][1]]=0;
						visit[sx+dir[i][0]-1][sy+dir[i][1]]=0;
						tsum-=2;
					}
					if(i==1){
						tsum+=2;
						visit[sx+dir[i][0]][sy+dir[i][1]]=1;
						visit[sx+dir[i][0]+1][sy+dir[i][1]]=1;
						dfs(sx+dir[i][0]+1,sy+dir[i][1]);
						visit[sx+dir[i][0]][sy+dir[i][1]]=0;
						visit[sx+dir[i][0]+1][sy+dir[i][1]]=0;
						tsum-=2;
					}
					if(i==2){
						tsum++;
						visit[sx+dir[i][0]][sy+dir[i][1]]=1;
						visit[sx+dir[i][0]][sy+dir[i][1]-1]=1;
						dfs(sx+dir[i][0],sy+dir[i][1]-1);
						visit[sx+dir[i][0]][sy+dir[i][1]]=0;
						visit[sx+dir[i][0]][sy+dir[i][1]-1]=0;
						tsum--;
					}
					if(i==3){
						tsum++;
						visit[sx+dir[i][0]][sy+dir[i][1]]=1;
						visit[sx+dir[i][0]][sy+dir[i][1]+1]=1;
						dfs(sx+dir[i][0],sy+dir[i][1]+1);
						visit[sx+dir[i][0]][sy+dir[i][1]]=0;
						visit[sx+dir[i][0]][sy+dir[i][1]+1]=0;
						tsum--;
					}
				}
				
			
		}
		
	}
}
int main()
{
//	ifstream fin;
//	fin.open("aaa.txt");
	
	while(cin>>m>>n)
	{
		for(int i=1;i<=m;i++)
		{
			for(int j=1;j<=n;j++)
			{
				cin>>maze[i][j];
				if(maze[i][j]=='S'){
					sx=i;
					sy=j;
				}
				if(maze[i][j]=='|'){
					stairs=1;             //标记楼梯的方向 
				}
				if(maze[i][j]=='-'){
					stairs=0;
				}
				if(maze[i][j]=='T'){
					endx=i;
					endy=j;
				}
			}
		}
		memset(visit,0,sizeof(visit));
		flag=0;
		tsum=0,tmin=999999;
		visit[sx][sy]=1;
		dfs(sx,sy);
		if(flag)cout<<tmin<<endl;
		//else cout<<"no!"<<endl;
	}
	return 0;
}

来一个bfs+优先队列的解法，0MS，里面用到的是小堆优先队列：

#include <iostream>
#include <cstdio>
#include <queue>
using  namespace std;
 
struct Node {
    int x, y, step;
};
 
bool operator< ( Node n1, Node n2 ) {
    return n1.step > n2.step;
}
 
 
int n, m;
int dir[4][2] = { 0, 1, 1, 0, 0, -1, -1, 0 };
char mp[25][25];
bool mark[25][25];
Node _s;
void input() {
    int i, j;
    for( i=0; i<n; ++i ) scanf( "%s", mp[i] );
    for( i=0; i<n; ++i ) {
        for( j=0; j<n; ++j ) {
            if( mp[i][j] == 'S' ) {
                _s.x = i;
                _s.y = j;
                _s.step = 0;
            }
        }
    }
}
 
 
inline bool isBond( Node nd ) {
    if( nd.x < 0 || nd.y < 0 || nd.x >= n || nd.y >= m || mp[nd.x][nd.y] == '*' ) return 1;
    return 0;
}
 
 
int solve() {
    priority_queue<Node> Q;
    Node p, q;
    int i;
 
    Q.push( _s );
    memset(mark, 0, sizeof(mark) );
    mark[_s.x][_s.y] = 1;
    while( !Q.empty() ) {
        p = Q.top();
        Q.pop();
        if( mp[p.x][p.y] == 'T' ) return p.step; //1.放这里判断比放for里面保险，虽然多花了点时间。
        for( i=0; i<4; ++i ) {
            q.x = p.x + dir[i][0];
            q.y = p.y + dir[i][1];
            q.step = p.step + 1;                 //2.走完后步数马上加
            if( isBond( q ) || mark[q.x][q.y] ) continue;
            if( mp[q.x][q.y] == '.' || mp[q.x][q.y] == 'T' ) {
                Q.push( q );
                mark[q.x][q.y] = 1;              //3.每次push后，马上跟新mark
                continue;
            } else if( mp[q.x][q.y] == '|' ) {
                if( p.step % 2 == 1 && i % 2 == 1 ) q.step ++;
                if( p.step % 2 == 0 && i % 2 == 0 ) q.step ++;
            } else if( mp[q.x][q.y] == '-' ) {
                if( p.step % 2 == 1 && i % 2 == 0 ) q.step ++;
                if( p.step % 2 == 0 && i % 2 == 1 ) q.step ++;
            }
            q.x += dir[i][0];
            q.y += dir[i][1];
            if( isBond(q) || mark[q.x][q.y] ) continue;
            Q.push( q );
            mark[q.x][q.y] = 1;
        }
    }
    return -1;
}
 
 
int main() {
//  freopen( "c:/aaa.txt", "r", stdin );
    while( scanf("%d %d", &n, &m) == 2 ) {
        input();
        printf( "%d\n", solve() );
    }
    return 0;
}