python 算法题（1）

两个班级的小朋友在操场上，需要根据输入帮两个班级的小朋友找回自己的班级。例如输入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)