题解 | #MP3光标位置#

MP3光标位置

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

import sys
num=int(input())
op=input()
up=op.count('U')
down=op.count('D')

move=down-up
pos=1
#if move>0:
#    pos=pos+move%num

#if move<0:
#    pos=num+move%num-pos

def move(str1,pos,num,list1):
    if str1=="U":
        pos-=1
        if num>4:
            if pos==0:
                pos=num
                list1=[num-3,num-2,num-1,num]
            elif pos>0 and pos<list1[0]:
                list1=[pos,pos+1,pos+2,pos+3]
        elif num<=4:
            if pos==0:
                pos=num
            list1=[]
            for i in range(1,num+1):
                list1.append(i)
    elif str1=="D":
        pos+=1
        if num>4:
            if pos==num+1:
                pos=1
                list1=[1,2,3,4]
            elif pos<=num and pos>list1[-1]:
                list1=[pos-3,pos-2,pos-1,pos]
        elif num<4:
            if pos==num+1:
                pos=1
            list1=[]
            for i in range(1,num+1):
                list1.append(i)
    return(pos,list1)

list1=[1,2,3,4]
for i in op:
    pos,list1=move(i,pos,num,list1)

print(*list1,sep=' ')
print(pos)

全部评论

相关推荐

点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务