3248. Snake in Matrix using Rust

Problem

There is a snake in an n x n matrix grid and can move in four possible directions. Each cell in the grid is identified by the position: grid[i][j] = (i * n) + j.

The snake starts at cell 0 and follows a sequence of commands.

You are given an integer n representing the size of the grid and an array of strings commands where each command[i] is either "UP", "RIGHT", "DOWN", and "LEFT". It’s guaranteed that the snake will remain within the grid boundaries throughout its movement.

Return the position of the final cell where the snake ends up after executing commands.

Example 1:

Input: n = 2, commands = [“RIGHT”,”DOWN”]

Output: 3

Example 2:

Input: n = 3, commands = [“DOWN”,”RIGHT”,”UP”]

Output: 1

Constraints:

• 2 <= n <= 10
• 1 <= commands.length <= 100
• commands consists only of "UP", "RIGHT", "DOWN", and "LEFT".
• The input is generated such the snake will not move outside of the boundaries.

Solution

First attempt : Time complexity O(n)

impl Solution {
    pub fn final_position_of_snake(n: i32, commands: Vec<String>) -> i32 {
        let mut output: i32 = 0;
        let mut initialPosition:Vec<i32> = vec![0,0];
        for i in 0..commands.len(){
           // println!("command is {} {:?}",i,commands[i]);
            if(commands[i] == "RIGHT"){
                initialPosition[1] = initialPosition[1] + 1; 
            }
            else if(commands[i] == "DOWN"){
                initialPosition[0] = initialPosition[0] + 1; 
            }
            else if(commands[i] == "LEFT"){
                initialPosition[1] = initialPosition[1] - 1; 
            }
            else if(commands[i] == "UP"){
                initialPosition[0] = initialPosition[0] - 1; 
            }
            else{
                println!("It will not come to this");
            }
        }
        //println!(" this is n {} and comamnds are {:?} final positions after operations are {:?}", n, commands, initialPosition);
        output = initialPosition[0]*n + initialPosition[1];
        return(output)
    }
}