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() |