牛客_华为_ HJ64 MP3光标位置

buer103

于 2023-05-04 18:34:36 发布

阅读量123

点赞数

分类专栏： LeetCode刷题文章标签： python

本文链接：https://blog.csdn.net/JStana/article/details/130492786

版权

LeetCode刷题专栏收录该内容

29 篇文章 0 订阅

订阅专栏

HJ64 MP3光标位置

在这里插入图片描述

我服了这个B题！！！
折磨我半小时


## 维护一个滑动窗口i j
## 光标初始的位置为第1首歌


## 歌曲数量
N = int(input())
ops = input()
arr = list(range(1,N+1))
i = 0
j = 3
if N<=4:
    print(' '.join(map(str,arr)))
    for op in ops:
        if op=='U':
            if i==0:i=N-1
            else:i-=1
        if op=='D':
            if i==N-1:i=0
            else:i+=1
    print(arr[i])
## N>4
else:
    for op in ops:
        if op=='U':
            if i==0:
                i=N-1
                j=N-1
            elif j-i==3:
                i-=1
                j-=1
            else:
                i-=1
        if op=='D':
            if i==N-1:
                i=0
                j=3
            elif i==j:
                i+=1
                j=j+1
            else:
                i+=1

    print(' '.join(map(str,arr[j-3:j+1])))
    print(arr[i])