Matlab:
function f = pyths(n)
[x y z] = meshgrid(1:n, 1:n, 1:n);
sol = x.^2 + y.^2 == z.^2;
f = [x(sol) y(sol) z(sol)];

こういう例題って効率良く解こうとすると
途端に汚くなるんだよな