python 两个班小朋友排队时混在了一起，每个小朋友都知道自己跟前一个小朋友是不是同班，把同班的小朋友找出来

python 两个班小朋友排队时混在了一起，每个小朋友都知道自己跟前一个小朋友是不是同班，把同班的小朋友找出来

小朋友的编号为正数，与前面一个小朋友同班用Y表示，不同班用N表示

输入描述：输入空格分开的小朋友编号和是否同班的标志

如：6/N 2/Y 3/N 4/Y 表示一共4位小朋友

2和6是同班 3和2 不通班，4和三同班

小朋友总数<999

不考虑输入格式错误。 

#!/usr/bin/python
#-*- coding:utf-8 -*-

#instr = "6/N 3/Y 2/N 4/Y 5/Y 1/Y"
instr = raw_input("请输入小朋友的信息n")

inlist = instr.split(" ")
banji = 1
jiaban = []
yiban =[]
for i in inlist:
	student = i.split("/")[0]
	if banji == 1:
		if i.split("/")[1] == "Y":
			jiaban.append(student)
		else:
			yiban.append(student)
			banji = 2
	elif banji == 2:
		if i.split("/")[1] == "Y":
			yiban.append(student)
		else:
			jiaban.append(student)
			banji = 1
jiaban.sort()
yiban.sort()

print ' ' .join(jiaban)
print ' ' .join(yiban)

如果对你有帮助可以加个关注