namespace Kata.Verification {
    open Microsoft.Quantum.Katas;

    // Two qubit parity Measurement
    operation StatePrep_ParityMeasurement(qs : Qubit[], state : Int, alpha : Double) : Unit is Adj {
        // prep cos(alpha) * |0..0⟩ + sin(alpha) * |1..1⟩
        Ry(2.0 * alpha, qs[0]);
        CNOT(qs[0], qs[1]);

        if state == 1 {
            // flip the state of the first qubit
            X(qs[0]);
        }
    }

    @EntryPoint()
    operation CheckSolution() : Bool {
        let isCorrect = DistinguishStates_MultiQubit(
            2,
            2,
            StatePrep_ParityMeasurement,
            Kata.ParityMeasurement,
            true,
            ["α|00⟩ + β|11⟩", "α|01⟩ + β|10⟩"]);

        if (isCorrect) {
            Message("Correct!");
        } else {
            Message("Incorrect.");
        }

        isCorrect
    }
}
