CCF 201312-5 I’m stuck!

试题编号：	201312-5
试题名称：	I’m stuck!
时间限制：	1.0s
内存限制：	256.0MB
问题描述：	问题描述 　　给定一个R行C列的地图，地图的每一个方格可能是'#', '+', '-', '|', '.', 'S', 'T'七个字符中的一个，分别表示如下意思： 　　'#': 任何时候玩家都不能移动到此方格； 　　'+': 当玩家到达这一方格后，下一步可以向上下左右四个方向相邻的任意一个非'#'方格移动一格； 　　'-': 当玩家到达这一方格后，下一步可以向左右两个方向相邻的一个非'#'方格移动一格； 　　'|': 当玩家到达这一方格后，下一步可以向上下两个方向相邻的一个非'#'方格移动一格； 　　'.': 当玩家到达这一方格后，下一步只能向下移动一格。如果下面相邻的方格为'#'，则玩家不能再移动； 　　'S': 玩家的初始位置，地图中只会有一个初始位置。玩家到达这一方格后，下一步可以向上下左右四个方向相邻的任意一个非'#'方格移动一格； 　　'T': 玩家的目标位置，地图中只会有一个目标位置。玩家到达这一方格后，可以选择完成任务，也可以选择不完成任务继续移动。如果继续移动下一步可以向上下左右四个方向相邻的任意一个非'#'方格移动一格。 　　此外，玩家不能移动出地图。 　　请找出满足下面两个性质的方格个数： 　　1. 玩家可以从初始位置移动到此方格； 　　2. 玩家不可以从此方格移动到目标位置。 输入格式 　　输入的第一行包括两个整数R 和C，分别表示地图的行和列数。(1 ≤ R, C ≤ 50)。 　　接下来的R行每行都包含C个字符。它们表示地图的格子。地图上恰好有一个'S'和一个'T'。 输出格式 　　如果玩家在初始位置就已经不能到达终点了，就输出“I'm stuck!”（不含双引号）。否则的话，输出满足性质的方格的个数。 样例输入 5 5 --+-+ ..|#. ..|## S-+-T ####. 样例输出 2 样例说明 　　如果把满足性质的方格在地图上用'X'标记出来的话，地图如下所示： 　　--+-+ 　　..|#X 　　..|## 　　S-+-T 　　####X

关键词：dfs深度优先遍历+栈，正序建立树，倒序建立树

#include<iostream>
#include<vector>
#include<stack>

using namespace std;
bool isleft(int i,int j,vector<vector<char> > &f);
bool isright(int i,int j,vector<vector<char> > &f);
bool isup(int i,int j,vector<vector<char> > &f);
bool isdown(int i,int j,vector<vector<char> > &f);

void dfs(vector<vector<char> > &f,stack<pair<int,int> > &s,vector<char> &sf,int i,int j);
void dfs2(vector<vector<char> > &f,stack<pair<int,int> > &e,vector<char> &sf,int i,int j);
int r,c;

int main(){
    // 0 1 2 3 up down left right
    //freopen("in2.txt","r",stdin);
    cin >> r;
    cin >> c;
    int si,sj;
    int ei,ej;    
    vector<vector<char> > f;
    vector<char> fb;
    for(int i = 0;i<r;i++){
        fb.clear();
        for(int j = 0;j<c;j++){
            char buf;
            cin>>buf;
            if(buf != '+' 
                && 
                buf != '-' 
                &&
                buf != '|' 
                && 
                buf != '.'
                && 
                buf != '#' 
                && 
                buf != 'S'
                &&
                buf != 'T'){
                    continue;
            }
            if(buf == 'S'){
                si = i;
                sj = j;
            }
            if(buf == 'T'){
                ei = i;
                ej = j;
            }
            fb.push_back(buf);
        }
        f.push_back(fb);
    }
    vector<char> sf;//1 : 可以从起点到达的点
    for(int i = 0;i<r*c;i++){
        sf.push_back(0);
    }
    stack<pair<int,int> > s;
    dfs(f,s,sf,si,sj);
    
    vector<char> ef;//1 : 可以到达终点的点
    for(int i = 0;i<r*c;i++){
        ef.push_back(0);
    }
    stack<pair<int,int> > e;
    dfs2(f,e,ef,ei,ej);

    int suc = 0;
    for(int i = 0;i<r;i++){
        for(int j = 0;j<c;j++){
            if(sf[c*i+j] == 1 && ef[c*i+j] == 0){
                suc++;
            }
        }
    }
    if(sf[c*ei+ej] == 0){
        cout<<"I'm stuck!";
    }
    else{
        cout<<suc;
    }

    return 0;
}
void dfs2(vector<vector<char> > &f,stack<pair<int,int> > &e,vector<char> &ef,int i,int j){
    pair<int,int> m = make_pair(i,j);
    e.push(m);
    ef[i*c+j] = 1;
    if(isdown(i-1,j,f)){
        if(ef[c*(i-1)+j] == 0){
            dfs2(f,e,ef,i-1,j);
        }
    }
    if(isup(i+1,j,f)){
        if(ef[c*(i+1)+j] == 0){
            dfs2(f,e,ef,i+1,j);
        }
    }
    if(isright(i,j-1,f)){
        if(ef[c*i+j-1] == 0){
            dfs2(f,e,ef,i,j-1);
        }
    }
    if(isleft(i,j+1,f)){
        if(ef[c*i+j+1] == 0){
            dfs2(f,e,ef,i,j+1);
        }
    }
    e.pop();
}
void dfs(vector<vector<char> > &f,stack<pair<int,int> > &s,vector<char> &sf,int i,int j){
    pair<int,int> m = make_pair(i,j);
    s.push(m);
    sf[i*c+j] = 1;
    if(isup(i,j,f)){
        if(sf[c*(i-1)+j] == 0){
            dfs(f,s,sf,i-1,j);
        }
    }
    if(isdown(i,j,f)){
        if(sf[c*(i+1)+j] == 0){
            dfs(f,s,sf,i+1,j);
        }
    }
    if(isleft(i,j,f)){
        if(sf[c*i+j-1] == 0){
            dfs(f,s,sf,i,j-1);
        }
    }
    if(isright(i,j,f)){
        if(sf[c*i+j+1] == 0){
            dfs(f,s,sf,i,j+1);
        }
    }
    s.pop();
}


bool isup(int i,int j,vector<vector<char> > &f){
    if(i <= 0
            ||
        i >= r
            ||
        j < 0
            ||
        j >= c
            ||
        f[i][j] == '-'
            ||
        f[i][j] == '.'
            ||
        f[i][j] == '#'){
        return false;
    }
    if(f[i-1][j] == '#'){
        return false;
    }
    return true;
}
bool isdown(int i,int j,vector<vector<char> > &f){
    if(i >= r-1
            ||
        i < 0
            ||
        j < 0
            ||
        j >= c
            ||
        f[i][j] == '-'
            ||
        f[i][j] == '#'){
            return false;
    }
    if(f[i+1][j] == '#'){
        return false;
    }
    return true;
}
bool isleft(int i,int j,vector<vector<char> > &f){
    if(j <= 0
            ||
        j >= c
            ||
        i < 0
            ||
        i >= r
            ||
        f[i][j] == '|'
            ||
        f[i][j] == '.'
            ||
        f[i][j] == '#'){
            return false;
    }
    if(f[i][j-1] == '#'){
        return false;
    }
    return true;
}
bool isright(int i,int j,vector<vector<char> > &f){
    if(j >= c-1
            ||
        j < 0
            ||
        i < 0
            ||
        i >= r
            ||
        f[i][j] == '|'
            ||
        f[i][j] == '.'
            ||
        f[i][j] == '#'){
            return false;
    }
    if(f[i][j+1] == '#'){
        return false;
    }
    return true;
}