ZOJ1002栈实现

这个题很久之前做过，是N皇后问题的一个变式，之前用的是递归，前几天看数据结构，想着用栈实现看看，然后就写了一下。

大概思路是维持一个栈，和一个不断更新最大栈高的变量ans，每审查一个新元素，和栈中已经有的元素进行对比，不冲突就入栈。

当一种可能的方案运行完成后，栈顶元素出栈。跳转到另一个方案。具体代码如下：

 

#include <cstdio>
#include <iostream>
#include <cstdlib>

using namespace std;

const int N = 4;
char Matrix[N][N];
struct Block{
    int x, y;
    Block(int xx = 0, int yy = 0) : x(xx), y(yy) {};
};

bool compare(Block const p1, Block const p2){
    int max_x = p1.x > p2.x ? p1.x : p2.x;
    int max_y = p1.y > p2.y ? p1.y : p2.y;
    int min_x = p1.x < p2.x ? p1.x : p2.x;
    int min_y = p1.y < p2.y ? p1.y : p2.y;
    if(p1.y == p2.y){
        for(int i = min_x; i < max_x; i++){
            if(Matrix[i][p1.y] == 'X') return true;
        }
    }else if(p1.x == p2.x){
        for(int i = min_y; i < max_y; i++){
            if(Matrix[p1.x][i] == 'X') return true;
        }
    }
    return false;
}

bool judge(Block const p1, Block const p2){
    return (p1.x == p2.x && !compare(p1, p2))
    || (p1.y == p2.y && !compare(p1, p2)) ;
}

int collectBlock(int level){
    int ans = -1;
    int res_index = 0;
    int count = 0;
    
    Block b(0,0);
    Block res[16];
    do {
        //主要逻辑在这里 
        if(Matrix[b.x][b.y] != 'X' && b.x < level && b.y < level){
            for(int i = 0; i < res_index; ++i){                
                if(judge(res[i], b)) count++;
            }
            if(count == 0){
                res[res_index++] = b;
            }
            b.y++;
        }else{
            b.y++;
        }
        count = 0;
        if(b.y >= level){
            b.x++; b.y = 0;
        }
        if(b.x >= level){
            ans = ans > res_index ? ans : res_index;
            res_index--; //这个地方注意栈顶指针的指向，之前用的是res[res_index--]，老是出错 
            b = res[res_index];
            if(res_index < 0) break;
            b.y++;
        }
    } while(b.x >= 0 || b.y < level);
    return ans;
}

int main(){
    int level;
    while(1){
        cin >> level;
        if(level == 0) break;
        for(int i = level-1; i >= 0; --i){
            for(int j = 0; j < level; ++j){
                cin >> Matrix[i][j];
            }
        }
        int rst = collectBlock(level);
        cout << rst << endl;
    }
    return 0;
}