快手 2019 算法笔试（视频图片输出问题）

图片由网上提供仅供参考：

 

import sys

N = int(sys.stdin.readline().strip())
M = int(sys.stdin.readline().strip())

arr = []
for i in range(M):
    arr.append(sys.stdin.readline().strip())

result = []

if M == 0:
    print(0)
elif M < N:
    print(M)
    for item in arr:
        print(item)
else:
    i = 0
    tmp = 0

    queue = []
    while i < M:
        if arr[i][0] == "V":
            result.append(arr[i])
            tmp += 1
            if tmp >= (N-1) and len(queue) != 0:
                tmp = 0
                result.append(queue[0])
                queue.pop(0)
        else:
            if tmp >= (N-1):
                result.append(arr[i])
                tmp = 0
            else:
                queue.append(arr[i])
        i += 1
    print(len(result))
    for item in result:
        print(item)