POJ 4115:鸣人和佐助

题目

总时间限制: 1000ms 内存限制: 65536kB
描述
佐助被大蛇丸诱骗走了，鸣人在多少时间内能追上他呢？

已知一张地图（以二维矩阵的形式表示）以及佐助和鸣人的位置。地图上的每个位置都可以走到，只不过有些位置上有大蛇丸的手下，需要先打败大蛇丸的手下才能到这些位置。鸣人有一定数量的查克拉，每一个单位的查克拉可以打败一个大蛇丸的手下。假设鸣人可以往上下左右四个方向移动，每移动一个距离需要花费1个单位时间，打败大蛇丸的手下不需要时间。如果鸣人查克拉消耗完了，则只可以走到没有大蛇丸手下的位置，不可以再移动到有大蛇丸手下的位置。佐助在此期间不移动，大蛇丸的手下也不移动。请问，鸣人要追上佐助最少需要花费多少时间？
输入
输入的第一行包含三个整数：M，N，T。代表M行N列的地图和鸣人初始的查克拉数量T。0 < M,N < 200，0 ≤ T < 10
后面是M行N列的地图，其中@代表鸣人，+代表佐助。*代表通路，#代表大蛇丸的手下。
输出
输出包含一个整数R，代表鸣人追上佐助最少需要花费的时间。如果鸣人无法追上佐助，则输出-1。
样例输入

样例输出
样例输出1
6

样例输出2
4

思路

很有意思的题目。
BFS，用状态记录下每一步的查克拉数量。

代码

while True:
    try:
        M, N, T = input().split()
        M = int(M); N = int(N); T = int(T)
        grid= []
        r, c = 0, 0
        for i in range(M):
            grid.append(list(input()))
            if '@' in grid[i]:
                r, c = i, grid[i].index('@')
        queue = []
        visited = [[[0 for i in range(20)] for j in range(N)] for k in range(M)]
        queue.append([r, c, T, 0])
        directions = [[-1, 0], [1, 0], [0, 1], [0, -1]]
        visited[r][c][T] = 1
        found = False
        while len(queue):
            if found:
                break
            tmp = queue[0]
            queue.pop(0)
            if grid[tmp[0]][tmp[1]] == '+':
                found = True
                print(tmp[3])
                break
            for direction in directions:
                new_r = tmp[0] + direction[0]
                new_c = tmp[1] + direction[1]
                if new_r >= 0 and new_r < M and new_c >= 0 and new_c < N:
                    if grid[new_r][new_c] == '*' and not visited[new_r][new_c][tmp[2]]:
                        visited[new_r][new_c][tmp[2]] = 1
                        queue.append([new_r, new_c, tmp[2], tmp[3] + 1])
                    elif grid[new_r][new_c] == '#' and tmp[2] > 0 and not visited[new_r][new_c][tmp[2] - 1]:
                        visited[new_r][new_c][tmp[2] - 1] = 1
                        queue.append([new_r, new_c, tmp[2] - 1, tmp[3] + 1])
                    elif grid[new_r][new_c] == '+':
                        found = True
                        print(tmp[3] + 1)
                        break
        if not found:
            print(-1)
    except:
        break