Generating a particular sequnce of numbers
이전 댓글 표시
Hi,
given a variable natural number d, I'm trying to generate a sequence of the form:
[1 2 1 3 2 1 4 3 2 1.......d d-1 d-2......3 2 1].
I don't want to use for loop for this process, does anyone know a better (faster) method. I tried the colon operator without any success.
Thank you.
Adi
채택된 답변
추가 답변 (6개)
Roger Stafford
2013년 7월 27일
Here's another method to try:
N = d*(d+1)/2;
A = zeros(1,N);
n = 1:d;
A((n.^2-n+2)/2) = n;
A = cumsum(A)-(1:N)+1;
댓글 수: 1
Adi gahlawat
2013년 7월 27일
편집: Adi gahlawat
2013년 7월 27일
Azzi Abdelmalek
2013년 7월 28일
편집: Azzi Abdelmalek
2013년 7월 28일
Edit
This is twice faster then Stafford's answer
A4=zeros(1,d*(d+1)/2); % Pre-allocate
c=0;
for k=1:d
A4(c+1:c+k)=k:-1:1;
c=c+k;
end
댓글 수: 1
Richard Brown
2013년 7월 29일
Even faster:
k = 1;
n = d*(d+1)/2;
out = zeros(n, 1);
for i = 1:d
for j = i:-1:1
out(k) = j;
k = k + 1;
end
end
댓글 수: 7
Jan
2013년 7월 29일
In my measurements this is remarkably slower than Azzi's loop approach. Which Matlab version are you using?
Richard Brown
2013년 7월 29일
2012a on 64 bit Linux. Can try R2013a tomorrow for interest's sake :-)
Azzi Abdelmalek
2013년 7월 29일
Test with d=3000 and for loop (500)
d=3000;
tic
for k=1:500
k = 1;
n = d*(d+1)/2;
out = zeros(n, 1);
for i = 1:d
for j = i:-1:1
out(k) = j;
k = k + 1;
end
end
end
toc
tic
for k=1:500
A4=zeros(1,d*(d+1)/2); % Pre-allocate
c=0;
for k=1:d
A4(c+1:c+k)=k:-1:1;
c=c+k;
end
end
toc
isequal(A4,out')
Elapsed time is 23.695618 seconds. Richard's answer
Elapsed time is 17.408498 seconds. Azzi's answer
Richard Brown
2013년 7월 29일
Can you try again with a different loop counter name? I'm not in front of a computer right now so can't check whether that interferes or not...
Azzi Abdelmalek
2013년 7월 29일
Almost, the same result
Elapsed time is 22.940850 seconds.
Elapsed time is 16.967270 seconds.
Under R2011b I get for d=1000 and 500 repetitions:
Elapsed time is 3.466296 seconds. Azzi's loop
Elapsed time is 3.765340 seconds. Richard's double loop
Elapsed time is 1.897343 seconds. C-Mex (see my answer)
Richard Brown
2013년 7월 29일
편집: Richard Brown
2013년 7월 29일
I checked again, and I agree with Azzi. My method was running faster because of another case I had in between his and mine. The JIT was doing some kind of unanticipated optimisation between cases.
I get similar orders of magnitude results to Azzi for R2012a if I remove that case, and if I run in R2013a (Linux), his method is twice as fast.
Shame, I like it when JIT brings performance of completely naive loops up to vectorised speed :)
Jan
2013년 7월 29일
An finally the C-Mex:
#include "mex.h"
void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray*prhs[]) {
mwSize d, i, j;
double *r;
d = (mwSize) mxGetScalar(prhs[0]);
plhs[0] = mxCreateDoubleMatrix(1, d * (d + 1) / 2, mxREAL);
r = mxGetPr(plhs[0]);
for (i = 1; i <= d; i++) {
for (j = i; j != 0; *r++ = j--) ;
}
}
And if your number d can be limited to 65535, the times shrink from 1.9 to 0.34 seconds:
#include "mex.h"
void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray*prhs[]) {
uint16_T d, i, j, *r;
d = (uint16_T) mxGetScalar(prhs[0]);
plhs[0] = mxCreateNumericMatrix(1, d * (d + 1) / 2, mxUINT16_CLASS, mxREAL);
r = (uint16_T *) mxGetData(plhs[0]);
for (i = 1; i <= d; i++) {
for (j = i; j != 0; *r++ = j--) ;
}
}
For UINT32 0.89 seconds are required.
댓글 수: 1
Richard Brown
2013년 7월 29일
Nice. I imagine d would be limited to less than 65535, that's a pretty huge vector otherwise
Richard Brown
2013년 7월 29일
편집: Richard Brown
2013년 7월 29일
Also comparable, but not (quite) faster
n = 1:(d*(d+1)/2);
a = ceil(0.5*(-1 + sqrt(1 + 8*n)));
out = a.*(a + 1)/2 - n + 1;
댓글 수: 3
Richard Brown
2013년 7월 29일
Potentially suffers from floating point errors, but I checked it up to d = 10000 :)
Jan
2013년 7월 29일
@Richard: How did you find this formula?
Richard Brown
2013년 7월 29일
If you look at the sequence, and add 0, 1, 2, 3, 4 ... you get
n: 1 2 3 4 5 6 7 8 9 10
1 3 3 6 6 6 10 10 10 10
Note that these are the triangular numbers, and that the triangular numbers 1, 3, 6, 10 appear in their corresponding positions, The a-th triangular number is given by
n = a (a + 1) / 2
So if you solve this quadratic for a where n is a triangular number, you get the index of the triangular number. If you do this for a value of n in between two triangular numbers, you can round this up, and invert the formula to get the nearest triangular number above (which is what the sequence is). Finally, you just subtract the sequence 0, 1, 2, ... to recover the original one.
Andrei Bobrov
2013년 7월 27일
편집: Andrei Bobrov
2013년 7월 30일
out = nonzeros(triu(toeplitz(1:d)));
or
out = bsxfun(@minus,1:d,(0:d-1)');
out = out(out>0);
or
z = 1:d;
z2 = cumsum(z);
z1 = z2 - z + 1;
for jj = d:-1:1
out(z1(jj):z2(jj)) = jj:-1:1;
end
or
out = ones(d*(d+1)/2,1);
ii = cumsum(d:-1:1) - (d:-1:1) + 1;
out(ii(2:end)) = 1-d : -1;
out = flipud(cumsum(out));
카테고리
도움말 센터 및 File Exchange에서 Performance and Memory에 대해 자세히 알아보기
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!