题目描述:
幼儿园两个班的小朋友在排队时混在了一起,每位小朋友都知道自己是否与前面一位小朋友是否同班,请你帮忙把同班的小朋友找出来。
小朋友的编号为整数,与前一位小朋友同班用Y表示,不同班用N表示。
输入描述:
输入为空格分开的小朋友编号和是否同班标志。
比如:6/N 2/Y 3/N 4/Y,表示共4位小朋友,2和6同班,3和2不同班,4和3同班。
其中,小朋友总数不超过999,每个小朋友编号大于0,小于等于999。
不考虑输入格式错误问题。
输出描述:
输出为两行,每一行记录一个班小朋友的编号,编号用空格分开。且:
编号需要按照大小升序排列,分班记录中第一个编号小的排在第一行;
若只有一个班的小朋友,第二行为空行;
若输入不符合要求,则直接输出字符串ERROR。
if __name__ == '__main__':
str1 = input()
list1 = str1.split(" ")
a = []
b = []
y_x = None
try:
for num, i in enumerate(list1):
num_id = i.split("/")[0]
num_ju1 = i.split("/")[1]
if num == 0:
a.append(num_id)
y_x = "a"
else:
if num_ju1 == "N":
if y_x == "a":
b.append(num_id)
y_x = "b"
else:
a.append(num_id)
y_x = "a"
else:
if y_x == "a":
a.append(num_id)
else:
b.append(num_id)
a1 = sorted(a)
b1 = sorted(b)
print(' '.join(a1))
print(' '.join(b1))
except:
print("ERROR")
结果展示: