2012/01/27 (Fri) Matlab: Codyで頭の体操！

Matlabプログラミングの問題が出されて、答えを書くとその長さに応じて得点が入る。短いほど高得点らしい。
http://www.mathworks.com/matlabcentral/cody/


今日はこれを考えてみる（ソース）。

INSTRUCTIONS

This is a multi-dimensional variant of the normal multiplication table used to teach elementary students multiplication. In this variant, we are going to produce a table that multiplies the divisors 1 to n (input) agains itself in d dimensions.

Note: Inputting d = 0 should return the number 1 and d = 1 should return a column vector with the elements 1 to n.

Example:

Input:

n = 3;
d = 3;

Output:

tt(:,:,1) = [ 1  2  3
              2  4  6
              3  6  9  ];
tt(:,:,2) = [ 2  4  6
              4  8  12
              6  12 18 ];
tt(:,:,3) = [ 3  6  9
              6  12 18
              9  18 27 ];

これで一応dが2以上の場合に対応できるようだが、forループ使っちゃってるし、d=1,0の場合を別に書かなきゃいけないからあんまり美しくない。ループなしで書く手もありそうだが・・・。


n = 3;

d = 3;

 

% --------------------------------------------------------------------

order = 1:d;

A = repmat((1:n)', [1, n * ones(1, d - 1)]); 

OUT = repmat(1, n * ones(1, d));

for dd = 1:d

    oo = (1 == dd) * order + (1 < dd) * ( (order == 1) * dd + (order == dd) * 1 + (order ~= 1 & order ~= dd) .* order);

    OUT = OUT .* permute(A, oo);

end

 

OUT

% ---------------------------------------------------------------------