Used chatgpt AI for optimization 1

Author: Raven-Murphy

rounds/3_dna/solution.py

@@ -5,13 +5,63 @@
 own faster implementation.
 """
 
-from .baseline import find_matches as _baseline
+from __future__ import annotations
+
+def _find_positions(sequence: bytes, pattern: bytes) -> list[int]:
+    positions = []
+    start = 0
+    find = sequence.find
+
+    while True:
+        pos = find(pattern, start)
+
+        if pos == -1:
+            break
+
+        positions.append(pos)
+        start = pos + 1
+
+    return positions
 
 
 def find_matches(fasta_path: str, pattern: bytes) -> list[tuple[str, list[int]]]:
     """Find every FASTA record whose sequence contains ``pattern``.
 
     Returns ``[(record_id, [positions...]), ...]`` in file order.
     """
-    # TODO: remove this delegation and write your own implementation here.
-    return _baseline(fasta_path, pattern)
+    matches: list[tuple[str, list[int]]] = []
+
+    with open(fasta_path, "rb") as f:
+        record_id = None
+        seq_parts = []
+
+        for line in f:
+
+            if line.startswith(b">"):
+
+                # process previous record
+                if record_id is not None:
+                    sequence = b"".join(seq_parts)
+
+                    positions = _find_positions(sequence, pattern)
+
+                    if positions:
+                        matches.append((record_id, positions))
+
+                # begin new FASTA record
+                record_id = line[1:].strip().decode("ascii")
+                seq_parts = []
+
+            else:
+                seq_parts.append(line.strip())
+
+        # process final record
+        if record_id is not None:
+            sequence = b"".join(seq_parts)
+
+            positions = _find_positions(sequence, pattern)
+
+            if positions:
+                matches.append((record_id, positions))
+
+    return matches