How to multiply values to specific elements in a matrix?

I have a matrix

                 A=[ 1 0 0 0;
                     0 1 0 0;
                     1 0 0 0;
                     0 0 0 1 ]

I want to multiply an array B= [1,5,7,8] to wherever 1's are in matrix A. Resultant matrix is:

A = [1 0 0 0;
     0 5 0 0 ;
     7 0 0 0;
     0 0 0 8]

How can I do it in vectorized form? Please note that number of ones in input matrix A will always equal the length of array B

Thanks!

Answer 1

[the cyclist]

In MATLAB 2016b (and presumably later yet-to-be-released versions), you can just write

A.*B'

In earlier versions, you need to use bsxfun:

bsxfun(@times,A,B')

Answer 2

[Roger Stafford]

Assuming the number of ones in A is equal to the length of B,

   At = A.’;
   At(At==1) = B;
   A = At.’;