You are given a logical matrix BW of any dimension, and a dimension dim. You need to find the locations of the last non-zero element of BW in that given dimension. If no non-zeros exist at an element location, return a 0 at that location.

For example, given:

BW = [0 0 0 0; 1 1 1 1; 0 1 1 0]

When dim = 1, you should return:

ans = [2 3 3 2], because these are the row numbers of the last non-zeroes in each column.

When dim = 2, you should return:

ans = [0; 4; 3], because these are the column numbers of the last non-zeroes in each row.

Don't forget that the input isn't restricted to 2D matrices.