分离算法

src/heurams/kernel/puzzles/mcq.py

@@ -0,0 +1,69 @@
+from base import BasePuzzle
+import random
+
+class MCQPuzzle(BasePuzzle):
+    """选择题谜题生成器
+    
+    Args:
+        mapping: 正确选项映射 {问题: 答案}
+        jammer: 干扰项列表
+        max_riddles_num: 最大生成谜题数 (默认2个)
+        prefix: 问题前缀
+    """
+
+    def __init__(
+        self,
+        mapping: dict,
+        jammer: list,
+        max_riddles_num: int = 2,
+        prefix: str = ""
+    ):
+        self.prefix = prefix
+        self.mapping = mapping
+        self.jammer = list(set(jammer + list(mapping.values())))
+        while len(self.jammer) < 4:
+            self.jammer.append(" ")
+        self.max_riddles_num = max(1, min(max_riddles_num, 5))
+        self.wording = "选择题 - 尚未刷新谜题"
+        self.answer = ["选择题 - 尚未刷新谜题"]
+        self.options = []
+
+    def refresh(self):
+        """刷新谜题，根据题目数量生成适当数量的谜题"""
+        if not self.mapping:
+            self.wording = "无可用题目"
+            self.answer = ["无答案"]
+            self.options = []
+            return
+        
+        num_questions = min(self.max_riddles_num, len(self.mapping))
+        questions = random.sample(list(self.mapping.items()), num_questions)
+        puzzles = []
+        answers = []
+        all_options = []
+        
+        for question, correct_answer in questions:
+            options = [correct_answer]
+            available_jammers = [
+                j for j in self.jammer if j != correct_answer
+            ]
+            if len(available_jammers) >= 3:
+                selected_jammers = random.sample(available_jammers, 3)
+            else:
+                selected_jammers = random.choices(available_jammers, k=3)
+            options.extend(selected_jammers)
+            random.shuffle(options)
+            puzzles.append(question)
+            answers.append(correct_answer)
+            all_options.append(options)
+        
+        question_texts = []
+        for i, puzzle in enumerate(puzzles):
+            question_texts.append(f"{self.prefix}:\n  {i+1}. {puzzle}")
+        
+        self.wording = question_texts
+        self.answer = answers
+        self.options = all_options
+
+    def __str__(self):
+        return f"{self.wording}\n正确答案: {', '.join(self.answer)}"