hdoj1175 连连看（dfs+剪枝）

处理连连看问题。

要求拐弯方向不多于两次。剪枝很重要！！！

用dir记录当前方向。Orz，居然没想到。

#include<iostream>
#include<cstring>
#include<algorithm>
#define maxn 1005
using namespace std;
const int dx[]={-1,1,0,0},dy[]={0,0,-1,1};
int map[maxn][maxn],V[maxn][maxn];
int n,m,flag,x1,x2,y1,y2;
void dfs(int x,int y,int num,int dir){
    if (flag || x<0 || x>=n || y<0 || y>=m || V[x][y]) return ;
    if (num==2 && x!=x2 && y!=y2) return ;//当拐弯数为2，但与终点不在同一方向 
    if (num>2) return ;
    if (num<=2 && x==x2 && y==y2){
        flag=1;
        return ;
    } 
    if (map[x][y]!=0){
        if (x==x1 && y==y1) ;
        else return ;
    }//当前不为零且不为起点,即路径上有棋子 
    V[x][y]=1;
    for (int i=0;i<4;i++){
        int xx=x+dx[i],yy=y+dy[i];
        if (i==dir) dfs(xx,yy,num,i);
        else dfs(xx,yy,num+1,i);
    }
    V[x][y]=0;
    return ;
}
int main(){
    int t;
    while (cin >> n >> m && n+m){
        memset(map,0,sizeof(map));
        for (int i=0;i<n;i++){
            for (int j=0;j<m;j++){
                cin >> map[i][j];
            }
        }
        cin >> t;
        while (t--){
            cin >> x1 >> y1 >> x2 >> y2;
            x1--,y1--,x2--,y2--;
            if (x1==x2 && y1==y2){
                cout << "NO\n";
                continue;
            }//起点与终点相同不能消去
            if (map[x1][y1]!=map[x2][y2] || !map[x1][y1] || !map[x2][y2]){
                cout << "NO\n";
                continue;
            }//起点与终点不同，或起点或终点位置没有棋子
            if (x1<0 || x1>=n || y1<0 || y1>=m || x2<0 || x2>=n || y2<0 || y2>=m){
                cout << "NO\n";
                continue;
            }//所给坐标超出当前范围
            flag=0;
            memset(V,0,sizeof(V));
            for (int i=0;i<n;i++){
                dfs(x1+dx[i],y1+dy[i],0,i); //从一个点的四个方向开始 ,拐弯数 ，当前方向 
            } 
            if (flag) cout << "YES\n";
            else cout << "NO\n"; 
        }
    }
    return 0;
}