C++基于prim实现迷宫生成

#include <iostream>
#include <vector>
#include <cstdio>
#include <ctime>
#include <cstdlib>
using namespace std;
#define m 51//row
#define n 51//column
#define down 1
#define right 2
#define left 4
#define up 8
vector <int> block_row;
vector <int> block_column;
vector <int> block_direct;
typedef struct point
{
 int x;
 int y;
}Point;
Point start,end;
int x_num=1;
int y_num=1;
int a[m+2][n+2];
void init(){//将地图全部置1=wall
 for(int i=0;i<=m+1;i++){
  for(int j=0;j<=n+1;j++){
   a[i][j]=1;//wall
  }
 }
}
void push(int x,int y,int direct){//把一组数据存放进三个vector容器中
 block_row.push_back(x);
 block_column.push_back(y);
 block_direct.push_back(direct);
}
int count(){//计算当前位置周围 墙的个数
 int cnt=0;
 if(x_num+1<=m){
  push(x_num+1,y_num,down);
  cnt++;
 } //down
 if(y_num+1<=n){
  push(x_num,y_num+1,right);
  cnt++;
 } //right
 if(x_num-1>=1){
  push(x_num-1,y_num,up);
  cnt++;
 } //up
 if(y_num-1>=1){
  push(x_num,y_num-1,left);
  cnt++;
 } //left
 return cnt;
}
int main(){
 start.x=1;//定义起始点
 start.y=1;
 end.x=m;
 end.y=n;
 init();
 srand((unsigned)time(NULL));//随机数种子
 count();
 a[1][1]=2;
 while(block_row.size()){//第一步压入两堵墙（起点右边和起点下面）进入循环
  int num=block_row.size();
  int randnum=rand()%num;//生成0-num-1之间的随机数，同时也是vector里的下标
  x_num=block_row[randnum];//下面这两行可以不写
  y_num=block_column[randnum];
  switch(block_direct[randnum]){//选择一个方向进行后续操作，起始点 邻块 目标块 三块区域在同一直线上 随后移动到目标块的位置
   case down:{
    x_num=block_row[randnum]+1;
    y_num=block_column[randnum];
    break;
   }
   case right:{
    x_num=block_row[randnum];
    y_num=block_column[randnum]+1;
    break; 
   }
   case left:{
    x_num=block_row[randnum];
    y_num=block_column[randnum]-1;
    break;
   }
   case up:{
    x_num=block_row[randnum]-1;
    y_num=block_column[randnum];
    break;
   }
  }
  if(a[x_num][y_num]==1){//目标块如果是墙
   a[block_row[randnum]][block_column[randnum]]=2;//打通墙
   a[x_num][y_num]=2;//打通目标块
   count();//再次计算当前位置周围的邻墙个数并保存进vector
  }   
  block_row.erase(block_row.begin()+randnum);//删除这堵墙(把用不了的墙删了，对于那些已经施工过了不必再施工了，同时也是确保我们能跳出循环)
  block_column.erase(block_column.begin()+randnum);
  block_direct.erase(block_direct.begin()+randnum);
 }
 for(int i=0;i<=m+1;i++){
  for(int j=0;j<=n+1;j++){
   if(a[i][j]==2){
    printf("* ");
   }
   else{
    printf("1 ");
   }
  }
  printf("\n");
 }
 return 0;
}