1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13

class Solution {
public:
    int finalPositionOfSnake(int n, vector<string>& commands) {
        int startX = 0, startY = 0;
        for (auto& command : commands) {
            if (command == "RIGHT" || command == "LEFT")
                startY = startY + (command == "LEFT" ? -1 : 1);
            else if (command == "UP" || command == "DOWN")
                startX = startX + (command == "UP" ? -1 : 1);
        }
        return ((startX)*n - 1) + startY + 1;
    }
};