import sys
n = int(sys.stdin.readline().strip())
lines = sys.stdin.read().splitlines()
ans_list = []
count = 0
for i in range(n):
if count == 20:
break
word, wbeen = lines[i].split()[:2] # Read first two words efficiently
if word == "TAK":
if count <= 10 or (count <= 19 and int(wbeen) <= 1):
ans_list.append(str(i + 1))
count += 1
print(" ".join(ans_list))
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | import sys n = int(sys.stdin.readline().strip()) lines = sys.stdin.read().splitlines() ans_list = [] count = 0 for i in range(n): if count == 20: break word, wbeen = lines[i].split()[:2] # Read first two words efficiently if word == "TAK": if count <= 10 or (count <= 19 and int(wbeen) <= 1): ans_list.append(str(i + 1)) count += 1 print(" ".join(ans_list)) |
English