校招真题练习021 迷路的牛牛(网易)

迷路的牛牛

题目描述
牛牛去犇犇老师家补课，出门的时候面向北方，但是现在他迷路了。虽然他手里有一张地图，但是他需要知道自己面向哪个方向，请你帮帮他。

输入描述:
每个输入包含一个测试用例。
每个测试用例的第一行包含一个正整数，表示转方向的次数N(N<=1000)。
接下来的一行包含一个长度为N的字符串，由L和R组成，L表示向左转，R表示向右转。

输出描述:
输出牛牛最后面向的方向，N表示北，S表示南，E表示东，W表示西。

n = int(input().strip())
D = input().strip()
direction = 'N'
for i in range(n):
    cur = D[i]
    if direction == 'N':
        if cur == 'L':
            direction = 'W'
        else:
            direction = 'E'
    elif direction == 'S':
        if cur == 'L':
            direction = 'E'
        else:
            direction = 'W'
    elif direction == 'W':
        if cur == 'L':
            direction = 'S'
        else:
            direction = 'N'
    else:
        if cur == 'L':
            direction = 'N'
        else:
            direction = 'S'
print(direction)