52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
from __future__ import annotations
|
||
from utils.algo_context import AlgoContext, NULL_CTX
|
||
from utils.algo_int import Int
|
||
|
||
|
||
def irange(start_or_stop, stop=None, step: int = 1, ctx: AlgoContext = None):
|
||
"""
|
||
Drop-in Ersatz für range(), der Int-Objekte zurückgibt.
|
||
|
||
Wird wie Pythons range() aufgerufen:
|
||
irange(stop)
|
||
irange(start, stop)
|
||
irange(start, stop, step)
|
||
|
||
Indexarithmetik und Zählung
|
||
---------------------------
|
||
Ohne ctx-Argument (Standard) erhalten die erzeugten Indices einen NULL_CTX –
|
||
Arithmetik auf Loop-Indices (z.B. ``j - 1``) wird dann **nicht** gezählt.
|
||
Das entspricht dem üblichen Lehrbuchwunsch: nur Operationen auf Array-Inhalten
|
||
sollen in die Komplexitätsanalyse einfließen.
|
||
|
||
Mit ctx-Argument werden auch Indexoperationen gezählt:
|
||
for j in irange(n, ctx=ctx): ...
|
||
|
||
Beispiel
|
||
--------
|
||
ctx = AlgoContext()
|
||
z = Array.random(10, 0, 99, ctx)
|
||
for i in irange(len(z) - 1): # i ist Int, Arithmetik nicht gezählt
|
||
if z[i] > z[i + 1]: # Vergleich gezählt (z-Zellen haben ctx)
|
||
z.swap(i, i + 1)
|
||
"""
|
||
_ctx = ctx if ctx is not None else NULL_CTX
|
||
|
||
if stop is None:
|
||
start, stop_ = 0, int(start_or_stop)
|
||
else:
|
||
start, stop_ = int(start_or_stop), int(stop)
|
||
step = int(step)
|
||
|
||
assert step != 0, "irange: step darf nicht 0 sein"
|
||
|
||
num = start
|
||
if step > 0:
|
||
while num < stop_:
|
||
yield Int(num, _ctx)
|
||
num += step
|
||
else:
|
||
while num > stop_:
|
||
yield Int(num, _ctx)
|
||
num += step
|