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