23. 贪吃蛇

213 阅读3分钟

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

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

图 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

思路

链表就搞定了

#include <iostream>
#include <string>
#include <set>
#include <list>
#include <vector>
#include <sstream>
using namespace std;

vector<string> split(string& s) {
    stringstream ss(s);
    string item;
    vector<string>res;
    while(getline(ss ,item, ' ')) {
        res.push_back(item);
    }
    return res;
}

int main() {
    set<pair<int, int>> posSet;
    list<pair<int,int>> posList;


    string tmp;
    getline(cin, tmp);
    vector<string> commands = split(tmp);
    int n, m;
    cin >> n >> m;
    vector<vector<char>> pool(m, vector<char>(n, '-'));
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            cin >> pool[i][j];
            if (pool[i][j] == 'H') {
                posList.emplace_back(i, j);
                posSet.emplace(i, j);
            }
        }
    }

    for (int i = 0; i < commands.size(); i++) {
        int k,v;
        if (commands[i] == "D") {
             k = posList.front().first + 1;
             v = posList.front().second;
        }
        if (commands[i] == "U") {
             k = posList.front().first - 1;
             v = posList.front().second;
        }
        if (commands[i] == "L") {
             k = posList.front().first;
             v = posList.front().second - 1;
        }
        if (commands[i] == "D") {
             k = posList.front().first;
             v = posList.front().second + 1;
        }

        if (k < 0 || k >= m || v < 0 || v >= n || posSet.count({k, v}) != 0) {
            break;
        }
        if  (pool[k][v] == 'F') posList.emplace_front(k, v);
        if  (pool[k][v] == 'E') {
            auto tmp = posList.back();
            posSet.erase(tmp);
            posList.emplace_front(k, v);
            posList.pop_back();
        }
    }
    cout << posList.size() << endl;
    return 0;
}