基因序列分组优化

# 导入模块：sys用于读取输入，Counter用于统计元素出现次数 import sys from collections import Counter def helper(nums): """ 处理基因分组的核心函数 参数nums：输入的基因ID列表（整数类型） 返回：符合要求的分组结果（字符串格式），无解则返回"null" """ n = len(nums) # 获取输入列表的长度（样本总数） count = Counter(nums) # 统计每个基因ID的出现次数（如{1:2, 3:1}） A = [] # 实验组：需要满足“组内无重复、总和最小、数量为n//2” B = [] # 对照组：需要满足“组内无重复、数量为n//2” single_ids = [] # 存储出现次数为1的ID（后续需分配到A或B） # 遍历每个ID及其出现次数，做初步分组/无解判断 for key, val in count.items(): # 若某个ID出现超过2次：无法分组（组内不能有重复） if val > 2: return "null" # 若ID出现2次：给A、B组各分配1个（保证组内无重复） if val == 2: A.append(key) B.append(key) # 若ID出现1次：暂存到single_ids，后续分配 if val == 1: single_ids.append(key) # 对“出现1次的ID”排序：方便选最小的给A组（保证A组总和最小） single_ids.sort() # 计算A组还需要的ID数量：A组总容量是n//2，已从“出现2次的ID”拿了len(A)个 need = n // 2 - len(A) # 从排序后的single_ids中，取前need个最小的ID加入A组 for _ in range(need): A.append(single_ids.pop(0)) # pop(0)取列表首个元素（最小的） # 剩下的single_ids全部加入B组 B.extend(single_ids) # 两组按升序排序（满足题目输出要求） A.sort() B.sort() # 按题目要求格式返回：A组和B组用空格分隔，两组之间换行 return " ".join(map(str, A)) + "\n" + " ".join(map(str, B)) # 读取输入（处理多组输入，每行是一组基因ID） for line in sys.stdin: line = line.strip() # 去除换行/空格 if not line: continue input_list = line.split() # 按空格分割成字符串列表 input_nums = list(map(int, input_list)) # 转成整数列表（必须，否则数值排序错误） print(helper(input_nums)) # 调用分组函数并打印结果