@(n,k)sum(k.^(0:n^2-1))

ans(3,6)