两个班级的小朋友在操场上,需要根据输入帮两个班级的小朋友找回自己的班级。例如输入1/N 2/Y,说明这两个小朋友是在同一个班级,如果输入为1N 2N,说明不是同一个班级。
要求:
找到班级后,按升序排列
例如:
输入:3/Y 4/N 1/Y 20/Y 7/N 2/N
返回:
3 7
1 2 4 20
解题思路:
1、将输入的数据存入数组
2、对数组中的数据进行判断,创建两个班级数组,为Y时直接追加,为N时切换数组存放
str = input().split()
a = [str[0].split("/", 1)[0]]
b = []
temp = 0
for x in range(1,len(str)):
if ((str[x][-1])) == 'N':
temp = (temp+1)%2 #使0变1,变0
if temp == 0:
a.append(str[x].split("/",1)[0])
else:
b.append(str[x].split("/",1)[0])
a.sort()
b.sort()
print(a)
print(b)



