华为机试【10、贪吃蛇】

10、标题:贪吃蛇
【贪吃蛇】贪吃蛇是一个经典游戏，蛇的身体由若干方格连接而成，身体随蛇头移动。蛇头触碰到食物时，蛇的长度会增加一格。蛇头和身体的任一方格或者游戏版图边界碰撞时，游戏结束。

下面让我们来完成贪吃蛇游戏的模拟：
给定一个NM的数组ar，代表NM个方格组成的版图，贪吃蛇每次移动一个方格。若ar[i][j]=='H'，表示该方可为贪吃蛇的起始位置;若ar[i][j]=='F'，表示该方格为食物，若ar[i][j]=='E'，表示该方格为空格。
贪吃蛇初始长度为1，初始移动方向为向左。输入为给定一系列贪吃蛇的移动操作，返回操作后蛇的长度，如果在操作执行完之前已经游戏结束，返回游戏结束时贪吃蛇的长度。
贪吃蛇移动、吃食物和碰撞处理的细节见下面图示:

图一

图二

图三

图四

图五

图六

图 1:截取了贪吃蛇移动的一个中间状态，H表示蛇头，F表示食物，数字为蛇身体各节的 编号，蛇为向左移动，此时蛇头和食物已经相邻。

图 2:蛇头向左移动一格，蛇头和食物重叠，注意此时食物的格子成为了新的蛇头，第 1节 身体移动到蛇头位置，第 2节身体移动到第 1节身体位置，以此类推，最后添加第 4节升 起到原来第 3节身体的位置。

图 3:蛇头继续向左移动一格，身体的各节按上述规则移动，此时蛇头已经和边界相邻，但 还未碰撞。

图 4:蛇头继续向左移动一格，此时蛇头已经超过边界，发生碰撞，游戏结束。

图 5和图 6给出一个蛇头和身体碰撞的例子，蛇为向上移动。图 5时蛇头和第 7节身体相 邻，但还未碰撞;图 6蛇头向上移动一格，此时蛇头和第 8节身体都移动到了原来第 7节 身体的位置，发生碰撞，游戏结束。

输入描述: 输入第 1行为空格分隔的字母，代表贪吃蛇的移动操作。字母取值为 U、D、L、R、G，其中U、D、L、R分别表示贪吃蛇往上、下、左、右转向，转向时贪吃蛇不移动，G表示贪吃蛇按 当前的方向移动一格。用例保证输入的操作正确。
第 2行为空格分隔的两个数，指定为 N和 M，为数组的行和列数。余下 N行每行是空格分 隔的 M个字母。字母取值为 H、F和 E，H表示贪吃蛇的起始位置，F表示食物，E表示该 方格为空。用例保证有且只有一个 H，而 F和 E会有多个。
输出描述:
输出一个数字为蛇的长度。
示 例:
输入
DGG
33
FFF
FFH
EFE
输出
1

def snake():
    # a1 = ['D', 'G', 'L', 'G', 'G', 'U', 'G', 'G', 'R', 'G', 'G', 'D', 'G', 'L', 'G']
    # n, m = 3, 3
    # a2 = [['F', 'F', 'F'], ['F', 'F', 'H'], ['E', 'F', 'E']]
    a1 = input().split(" ")
    n, m = [int(x) for x in input().split(" ")]
    a2 = []
    for i in range(n):
        a2.append(input().split(" "))
    # 找到起始位置
    start = (0, 0)
    for i in range(n):
        for j in range(m):
            if a2[i][j] == 'H':
                a2[i][j] = 'E'  # 头开始移动之后变为空格
                start = (i, j)
    body = [start]
    direction = 'L'
    for i in a1:
        if i == 'U':
            direction = 'U'
        elif i == 'D':
            direction = 'D'
        elif i == 'L':
            direction = 'L'
        elif i == 'R':
            direction = 'R'
        elif i == 'G':
            if direction == 'U':
                next = (body[0][0] - 1, body[0][1])
            elif direction == 'D':
                next = (body[0][0] + 1, body[0][1])
            elif direction == 'L':
                next = (body[0][0], body[0][1] - 1)
            else:
                next = (body[0][0], body[0][1] + 1)
            if next[0] < 0 or next[1] < 0 or next[0] > n - 1 or next[1] > m - 1 or next in body[:-1]:
                return len(body)
            if a2[next[0]][next[1]] == 'E':
                body = [next] + body[:-1]
            elif a2[next[0]][next[1]] == 'F':
                body = [next] + body[:]
                a2[next[0]][next[1]] = 'E'  # 吃完之后，变为空格
    return len(body)


print(snake())