The Oracle of the Exact Distance | PennyLane Challenges

To attempt this challenge, please switch to a larger screen size.

Challenge statement

This challenge was part of the Canadian Quantum Cup 2023 coding competition.

A common concept associated with search algorithms is that of an oracle—an operator that changes the sign of the inputs that we are looking for. For example, let us label the possible inputs via the computational basis of the three-qubit states: Suppose that the sought-after element is . Then the action of the oracle is as follows:

and

However, oracles do not have to accept only one input, sometimes they accept more and will change sign to those that meet a certain property. We'll encounter an example of this in this challenge! Our goal will be to build an oracle capable of changing the sign of two three-qubit states only when these two states are separated by a distance , as shown in the figure below.

Although it may seem like a simple task, you may need some auxiliary qubits, so don't worry, you will be able to use them 😉

We will leave room at the top of the challenge template so that you can create as many helper functions as you want. However, the use of qml.QubitUnitary is not allowed. You must build this oracle without access to the circuit matrix, as would be done in a real quantum computer.

Challenge code

The only function that must be completed is the quantum function oracle_distance, which takes as input the distance d (int), You can create as many helper functions as you want at the top of the challenge template. You can use up to 5 auxiliary qubits. Remember that the output state of the auxiliary qubits must be reset to after use. Otherwise, this could cause problems when implementing the operator in different algorithms.

Input

As input to this problem, you are given:

d (int): the distance with which we will check that the oracle is working properly.

Output

As an output, we will use your oracle to print a list containing the values and . The elements of such list are indexed by the ordered pairs . If the index satisfies , then the corresponding element of the list is . Otherwise, it is equal to .

Test cases

We will test that your oracle generates the correct matrix by using all the possible values of the distance d, As an example, here is what you should get for and

test_input: 2
expected_output: [1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, -1, 1, 1, 1, -1, 1, 1, 1, 1, -1, 1, 1, 1, -1, 1, 1, 1, 1, -1, 1, 1, 1, -1, 1, 1, 1, 1, -1, 1, 1, 1, -1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1]

test_input: 7
expected_output: [1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1]

import json

import pennylane as qml

import pennylane.numpy as np

wires_m = [0, 1, 2] # qubits needed to encode m

wires_n = [3, 4, 5] # qubits needed to encode n

wires_aux = [6, 7, 8, 9, 10] # auxiliary qubits you can use

# Put your code here #

# Create all the helper functions you need here

def oracle_distance(d):

"""

Args:

d (int): the distance with which we will check that the oracle is working properly.

This function does not return anything, it is a quantum function that applies

necessary gates that implement the requested oracle.

"""

# Put your code here

# These functions are responsible for testing the solution.

wires_m = [0, 1, 2]

wires_n = [3, 4, 5]

wires_aux = [6, 7, 8, 9, 10]

dev = qml.device("default.qubit", wires=11)

@qml.qnode(dev)

def circuit(m, n, d):

qml.BasisEmbedding(m, wires=wires_m)

qml.BasisEmbedding(n, wires=wires_n)

oracle_distance(d)

return qml.state()

def run(test_case_input: str) -> str:

outputs = []

d = int(json.loads(test_case_input))

for n in range(8):

for m in range(8):

outputs.append(sum(circuit(n, m, d)).real)

output_list = [round(float(elem)) for elem in outputs]

return str(output_list)

def check(solution_output: str, expected_output: str) -> None:

i = 0

solution_output = json.loads(solution_output)

expected_output = json.loads(expected_output)

assert np.allclose(solution_output, expected_output)

names = list(qml.specs(circuit)(np.random.randint(7), np.random.randint(7), np.random.randint(7))['resources'].gate_types.keys())

assert names.count("QubitUnitary") == 0, "Can't use custom-built gates!"

# These are the public test cases

test_cases = [

('0', '[-1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1]'),

('1', '[1, -1, 1, 1, 1, 1, 1, 1, -1, 1, -1, 1, 1, 1, 1, 1, 1, -1, 1, -1, 1, 1, 1, 1, 1, 1, -1, 1, -1, 1, 1, 1, 1, 1, 1, -1, 1, -1, 1, 1, 1, 1, 1, 1, -1, 1, -1, 1, 1, 1, 1, 1, 1, -1, 1, -1, 1, 1, 1, 1, 1, 1, -1, 1]'),

('2', '[1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, -1, 1, 1, 1, -1, 1, 1, 1, 1, -1, 1, 1, 1, -1, 1, 1, 1, 1, -1, 1, 1, 1, -1, 1, 1, 1, 1, -1, 1, 1, 1, -1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1]'),

('3', '[1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, -1, 1, 1, 1, 1, 1, -1, 1, 1, -1, 1, 1, 1, 1, 1, -1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1]'),

('4', '[1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1]'),

('5', '[1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1]'),

('6', '[1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1]'),

('7', '[1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 1, 1]')

]

# This will run the public test cases locally

for i, (input_, expected_output) in enumerate(test_cases):

print(f"Running test case {i} with input '{input_}'...")

try:

output = run(input_)

except Exception as exc:

print(f"Runtime Error. {exc}")

else:

if message := check(output, expected_output):

print(f"Wrong Answer. Have: '{output}'. Want: '{expected_output}'.")

else:

print("Correct!")

or to submit your code