Update grover_search_algorithm.py

Author: AnshMNSoni

quantum/grover_search_algorithm.py

@@ -1,34 +1,65 @@
-from qiskit import QuantumCircuit, Aer, execute
-from qiskit.visualization import plot_histogram
-import math
+"""
+Grover's Search Algorithm implementation using Qiskit.
 
-# Number of qubits
-n = 2
+Grover's algorithm is a quantum algorithm for searching an unsorted database
+with quadratic speedup over classical algorithms.
 
-# Create Quantum Circuit
-qc = QuantumCircuit(n, n)
+Wikipedia:
+https://en.wikipedia.org/wiki/Grover%27s_algorithm
+"""
 
-# Step 1: Initialize in superposition
-qc.h(range(n))
+from typing import Dict
+from qiskit import QuantumCircuit, transpile
+from qiskit_aer import AerSimulator
 
-# -------- ORACLE (marks |11>) --------
-qc.cz(0, 1)
 
-# -------- DIFFUSER --------
-qc.h(range(n))
-qc.x(range(n))
-qc.h(1)
-qc.cx(0, 1)
-qc.h(1)
-qc.x(range(n))
-qc.h(range(n))
+def grover_search(shots: int = 1024) -> Dict[str, int]:
+    """
+    Runs Grover's search algorithm for 2 qubits and returns measurement results.
 
-# Measure
-qc.measure(range(n), range(n))
+    The oracle marks the |11> state.
 
-# Run on simulator
-backend = Aer.get_backend("qasm_simulator")
-result = execute(qc, backend, shots=1024).result()
-counts = result.get_counts()
+    Args:
+        shots (int): Number of simulation shots.
 
-print("Measurement Result:", counts)
+    Returns:
+        Dict[str, int]: Measurement counts.
+
+    Example:
+    >>> result = grover_search(100)
+    >>> isinstance(result, dict)
+    True
+    """
+
+    n = 2
+    qc = QuantumCircuit(n, n)
+
+    # Initialize superposition
+    qc.h(range(n))
+
+    # Oracle marking |11>
+    qc.cz(0, 1)
+
+    # Diffuser
+    qc.h(range(n))
+    qc.x(range(n))
+    qc.h(1)
+    qc.cx(0, 1)
+    qc.h(1)
+    qc.x(range(n))
+    qc.h(range(n))
+
+    # Measurement
+    qc.measure(range(n), range(n))
+
+    # Run on simulator
+    backend = AerSimulator()
+    compiled = transpile(qc, backend)
+    result = backend.run(compiled, shots=shots).result()
+    counts = result.get_counts()
+
+    return counts
+
+
+if __name__ == "__main__":
+    print(grover_search())