626A - Robot Sequence - CodeForces Solution

commandsChars = {
    "U": "+y",
    "D": "-y",
    "R": "+x",
    "L": "-x"
}


countOfCommands = int(input())
commandsString = input()

commands = []

class Robot():
    
    def __init__(self) -> None:
        self.x = 0
        self.y = 0
    
    def Move(self, coord: str):
        
        if coord == "+x":
            self.x += 1
        
        elif coord == "-x":
            self.x -= 1
        
        elif coord == "+y":
            self.y += 1
            
        else:
            self.y -= 1
    
    def isInZeroZero(self):
        if self.x == 0 and self.y == 0:
            return True
        else:
            return False   

    def Restart(self):
        self.x = 0
        self.y = 0     

class Vault():
    
    def __init__(self) -> None:
        self.paths = []
        
    def Add(self, path: list):
        
                    
                                                                
        self.paths.append(path)

robot = Robot()
pathsVault = Vault()

for command in commandsString:
    commands.append(commandsChars[command])

for firstElementIndex in range(0, countOfCommands):
    for secondElement in range(firstElementIndex, countOfCommands):
        secondElementIndex = countOfCommands - secondElement - 1
        
        nowCommands = commands[firstElementIndex:secondElement + 1]

        
        if len(nowCommands) > 1:
            
            for element in nowCommands:
                robot.Move(element)
            
            if robot.isInZeroZero() == True:
                
                pathsVault.Add(element)
            
            robot.Restart()

for firstElementIndex in range(0, countOfCommands):
    for secondElementIndex in range(firstElementIndex, countOfCommands):
        
        nowCommands = command[firstElementIndex:secondElementIndex]
        
        if len(nowCommands) > 1:
            
            for element in nowCommands:
                robot.Move(element)
            
            if robot.isInZeroZero() == True:
                
                pathsVault.Add(element)
            
            robot.Restart()

print(len(pathsVault.paths))