题解 | #MP3光标位置#
MP3光标位置
http://www.nowcoder.com/practice/eaf5b886bd6645dd9cfb5406f3753e15
短代码 三个指针 省列表
c_map = {
"U": -1,
"D": 1,
}
def run(n, seq):
ind = 0
a,b = ind, ind+4
for s in seq:
ind = ind + c_map.get(s, 0)
if ind < 0:
ind += n
a,b = max(n-4, 0), n
elif ind >= n:
ind += -n
a,b = 0, min(4, n)
if ind < a:
a,b = a-1, b-1
elif ind >= b:
a,b = a+1, b+1
win = list(range(a+1, b+1))
return win, ind+1
n = int(input())
seq = input().strip()
l, r = run(n, seq)
print(" ".join(list(map(str, l))))
print(r)