import sys buff = sys.stdin.readlines() l = 0 fin = "" for i, line in enumerate(buff[1:]): s, x = line.split() if s == "TAK": if l <= 10: l += 1 fin += f"{i + 1} " continue if int(x) < 2: l += 1 if l >= 20: fin += f"{i + 1}\n" break else: fin += f"{i + 1} " sys.stdout.write(fin)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | import sys buff = sys.stdin.readlines() l = 0 fin = "" for i, line in enumerate(buff[1:]): s, x = line.split() if s == "TAK": if l <= 10: l += 1 fin += f"{i + 1} " continue if int(x) < 2: l += 1 if l >= 20: fin += f"{i + 1}\n" break else: fin += f"{i + 1} " sys.stdout.write(fin) |