题解 | #MP3光标位置#

MP3光标位置

https://www.nowcoder.com/practice/eaf5b886bd6645dd9cfb5406f3753e15

while True:
    try:
        n = int(input())
        op = input()
        music = [i for i in range(1, n + 1)]
        cur = 0
        if len(music) <= 4:
            L = music
            for ch in op:
                if ch == "U":
                    cur -= 1
                    if cur == -len(music) - 1:
                        cur = -1
                elif ch == "D":
                    cur += 1
                    if cur == len(music):
                        cur = 0
        else:
            L = music[:4]
            first_L = L
            for ch in op:
                if ch == "U":
                    if L[cur] == 1:
                        L = music[-4:]
                        cur = 3
                    elif cur == 0:
                        i = music.index(L[0])
                        L = music[i - 1 : i + 3]
                    else:
                        cur = cur - 1
                if ch == "D":
                    if L[cur] == len(music):
                        cur = 0
                        L = first_L
                    elif cur == 3:
                        i = music.index(L[0])
                        L = music[i + 1 : i + 5]
                    else:
                        cur = cur + 1
        for i in L:
            print(i, end=" ")
        print()
        print(L[cur])
    except:
        break

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务