Come ho detto in precedenza, utilizzando Compile
verrà assegnato un codice più veloce. Utilizzando un algoritmo dal fxtbook, il seguente codice genera una prossima partizione ordinamento lessicografico:
PermutationIterator[f_, n_Integer?Positive, nextFunc_] :=
Module[{this = Range[n]},
While[this =!= {-1}, f[this]; this = nextFunc[n, this]];]
Il codice seguente presuppone che gestiscono la versione 8:
ClearAll[cfNextPartition];
cfNextPartition[target : "MVM" | "C"] :=
cfNextPartition[target] =
Compile[{{n, _Integer}, {this, _Integer, 1}},
Module[{i = n, j = n, ni, next = this, r, s},
While[Part[next, --i] > Part[next, i + 1],
If[i == 1, i = 0; Break[]]];
If[i == 0, {-1}, ni = Part[next, i];
While[ni > Part[next, j], --j];
next[[i]] = Part[next, j]; next[[j]] = ni;
r = n; s = i + 1;
While[r > s, ni = Part[next, r]; next[[r]] = Part[next, s];
next[[s]] = ni; --r; ++s];
next
]], RuntimeOptions -> "Speed", CompilationTarget -> target
];
Poi
In[75]:= Reap[PermutationIterator[Sow, 4, cfNextPartition["C"]]][[2,
1]] === Permutations[Range[4]]
Out[75]= True
Questo è chiaramente migliore nelle prestazioni rispetto alla funzione originale gen
.
In[83]:= gen[dummy, 9] // Timing
Out[83]= {26.067, Null}
In[84]:= PermutationIterator[dummy, 9, cfNextPartition["C"]] // Timing
Out[84]= {1.03, Null}
mezzo della macchina virtuale di Mathematica non è molto più lento:
In[85]:= PermutationIterator[dummy, 9,
cfNextPartition["MVM"]] // Timing
Out[85]= {1.154, Null}
Naturalmente questo non è neanche lontanamente implementazione del codice C, ma fornisce un notevole velocità-up su codice puro di livello superiore.
FWIW, il modo in cui Python fa iteratori/generatori è piuttosto insolito. Finché hai una sorta di astrazione sullo stato (chiusure, classi), puoi implementarle in qualsiasi lingua. – jrockway
Ah, bello. Forse aggiungilo come risposta alla tua stessa domanda (è considerato piuttosto kosher farlo). O c'è ancora una domanda senza risposta qui? – dreeves
Bene, è necessario passare in modo esplicito le proprie funzioni, mentre il Python fornisce una sorta di calcolo per te e lo inserisce nel framework. Quindi, non è proprio perfetto. Ma abbastanza buono, anzi, lo uso ora. – nes1983