Digamos que eu tenho uma matrizk = [1 2 0 0 5 4 0]

Eu posso calcular uma máscara da seguinte formam = k > 0 = [1 1 0 0 1 1 0]

Usando apenas a máscara me as seguintes operações

Shift esquerda / direita E / ou Adicionar / Subtrair / Multiplicar

Eu posso compactar k nos seguintes[1 2 5 4]

Aqui está como eu faço atualmente (pseudocódigo MATLAB):

function out = compact( in )
    d = in
    for i = 1:size(in, 2) %do (# of items in in) passes
        m = d > 0
        %shift left, pad w/ 0 on right
        ml = [m(2:end) 0] % shift
        dl = [d(2:end) 0] % shift

        %if the data originally has a gap, fill it in w/ the 
        %left shifted one
        use = (m == 0) & (ml == 1) %2 comparison  

        d = use .* dl + ~use .* d

        %zero out elements that have been moved to the left
        use_r = [0 use(1:end-1)]
        d = d .* ~use_r
    end

    out = d(1 : size(find(in > 0), 2)) %truncate the end
end

Intuiçã

Cada iteração, deslocamos a máscara para a esquerda e comparamos a máscara. Definimos um índice com os dados deslocados para a esquerda se descobrirmos que, após esse deslocamento, um índice originalmente nulo (máscara [i] = 0) agora é válido (máscara [i] = 1).

Questã

O algoritmo acima tem O (N * (3 turnos + 2 comparações + AND + adiciona + 3 multiplica)). Existe uma maneira de melhorar sua eficiência?