地牢房间迷宫走廊生成（二），Python实现洪水法、完美迷宫

---

前言

  前面通过随机房间、房门，对房门寻路生成走廊。由于使用A星算法，寻到的是最短路径，这样生成的走廊过直和简单。如果需要生成弯曲的走廊（这样的走廊能增加探险的乐趣和战斗拉扯），需要做些什么呢？
  如果我们要在原基础上修改，可以随机选取几个走廊的点，将其扩大复杂化，那么有没有其他方法呢？通过在一块地图上生成一个弯曲的迷宫可以实现这个需求。在这里，无论这个地图上有无房间。

  首先考虑到数据结构的问题，使用什么样的数据表示地图数据？包括房间、墙壁、房门、走廊、空白点。在这里，我使用一个三维数组表示整个图，前两维的点是坐标，第三维一共有6个数据，前4个依次表示为左上右下是否连通，第5个表示为该点是否存在物体或被遍历，第6个表示物体的类别id，数据表示为map[rows,cols,6]，当然第三维不一定全是数字，也可以使用二维的链表。

---

1 随机房间和房门

  在地图上随机寻找一定大小的房间，如果合适则放入房间。在这里需要限制循环的次数，避免无法放置房间时无限循环下去。随机房门时，在房间最外围随机寻找一个点当做房门（当然也可以使用另一个随机数来设定房门的数量）。为了简化代码，假定从房间的左上角出发，随机产生x和y方向的偏移量，使用预定义好的四个权重对x和y加权，将x和y映射到四个边上。此外，将数据可视化，使用matplotlib将地图画到图像上，在此需要一个数组表示图像。
如下所示

import random
import numpy as np
from matplotlib import pyplot as plt

def randomRoom(M):
    height, width, d = M.shape
    min_size = 5
    max_size = 10
    room_num = 20
    count = 0
    try_count = 0
    room = []
    while count < room_num:
        if try_count > 1000:
            break
        r = random.randint(min_size, max_size)
        c = random.randint(min_size, max_size)
        left = random.randint(3, width - 4)
        top = random.randint(3, height - 4)
        f = True
        for i in range(top-1,top+r+2):
            for j in range(left-1,left+c+2):
                if i >= height-3 or j >= width-3:
                    f = False
                    break
                if M[i,j,5] == 1:
                    f = False
                    break
        if f:
            room.append([left,top,c,r])
            for i in range(top, top + r + 1):
                for j in range(left, left + c + 1):
                    if i >= height-3 or j >= width-3:
                        continue
                    M[i, j, :] = 1
            count += 1
        try_count += 1
    return room

def randomDoor(map,room_list):
    for it in room_list:
        door_x,door_y = randomWall(it)
        for i in range(it[3]+1):
            map[it[1]+i, it[0], 5] = 3
            map[it[1]+i, it[0]+it[2], 5] = 3
        for i in range(it[2]+1):
            map[it[1], it[0]+i, 5] = 3
            map[it[1]+it[3], it[0]+i, 5] = 3
        map[door_x,door_y,5] = 2

def randomWall(room):
    direction = random.randint(0,3)
    dir = [[0,1,-1,0],[1,0,0,-1],[1,0,0,room[3]],[0,1,room[2],0]]
    x_off = random.randint(1,room[2]-2)
    y_off = random.randint(1,room[3]-2)
    x = room[0] + x_off*dir[direction][0] #+ dir[direction][2]
    y = room[1] + y_off*dir[direction][1] #+ dir[direction][3]
    return y,x

def drawMap(M, image):
    rows, cols, deep = M.shape
    block = [[2,8,0,2],[0,2,2,8],[2,8,8,10],[8,10,2,8]]
    color = [0,0,230,140]
    for row in range(0, rows