Untitled

numSteps = 100;

C = [1, 1; 1, -1] / sqrt(2);

psi = zeros(2, 2*numSteps+1);
psi(:, numSteps+1) = [1, 1i] / sqrt(2);

for step = 1:numSteps
    psi = C * psi;
    psi = [circshift(psi(1,:), [0, 1]); circshift(psi(2,:), [0, -1])];
end

prob_distribution = sum(abs(psi).^2, 1);
plot([-numSteps:numSteps], prob_distribution, 'LineWidth', 2);
xlabel('Position');
ylabel('Sannolikhet');
grid on;