import sys
import numpy as np
def main():
data = sys.stdin.read().strip().splitlines()
if not data:
return
n = int(data[0])
s_arr = np.empty(n, dtype='<U3')
x_arr = np.empty(n, dtype=int)
for i, line in enumerate(data[1:]):
parts = line.split()
s_arr[i] = parts[0]
x_arr[i] = int(parts[1])
eligible_mask = (s_arr == "TAK")
eligible_indices = np.nonzero(eligible_mask)[0]
group1 = eligible_indices[:10]
group2_candidates = eligible_indices[10:]
valid_group2_mask = x_arr[group2_candidates] < 2
group2 = group2_candidates[valid_group2_mask][:10]
finalists = np.concatenate((group1, group2))
finalists_sorted = np.sort(finalists) + 1
print(" ".join(map(str, finalists_sorted.tolist())))
if __name__ == '__main__':
main()
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | import sys import numpy as np def main(): data = sys.stdin.read().strip().splitlines() if not data: return n = int(data[0]) s_arr = np.empty(n, dtype='<U3') x_arr = np.empty(n, dtype=int) for i, line in enumerate(data[1:]): parts = line.split() s_arr[i] = parts[0] x_arr[i] = int(parts[1]) eligible_mask = (s_arr == "TAK") eligible_indices = np.nonzero(eligible_mask)[0] group1 = eligible_indices[:10] group2_candidates = eligible_indices[10:] valid_group2_mask = x_arr[group2_candidates] < 2 group2 = group2_candidates[valid_group2_mask][:10] finalists = np.concatenate((group1, group2)) finalists_sorted = np.sort(finalists) + 1 print(" ".join(map(str, finalists_sorted.tolist()))) if __name__ == '__main__': main() |
English