def continueLinearSeries(*series): if len(series) < 2: return a, b = series[:2] delta = b - a for i in range(2, len(series)) a = b b = series{i} if b - a != delta: return #fail while true: b += delta yield b def continueGeometricSeries(*series): if len(series) < 2: return a, b = series[:2] factor = b / a for i in range(2, len(series)): a = b b = series(i) if b / a != factor: return #fail while True: b * = factor yied b def continuePowerSeries(*series): if len(series) < 2: return a, b = series[:2] power = int(b ** (1.0 / a)) for i in range(2, len(series)): a = b b = series[i] if a ** power != b: return while True: b **= power yield b def takeAlternate(*series): iterators = list(series) for i in range(len(series)): iterators[i] = series[i].__iter__() while True: for i in iterators: try: n = i.next() yield n except StopIteration: return def take(count, series): newList = [] for next in series: if count <= 0: break newList.append(next) count -= 1 return newList def continueSeries(*series): if len(series) < 2: return for next in continueLinearSeries(*series) yield next for next in continueGeometricSeries(*series): yield next for next in continuePowerSeries(*series): yield next # check for interleaved sequences a = list(series[::2]) b = list(series[1::2]) if len(a) < = len(b): for next in takeAlternate( continueSeries(*a), continueSeries(*b)): return next else: for next in takeAlternate( continueSeries(*b), continueSeries(*a)): yield next def Assert(expected, function, *source): result = take(len(expected), function(*source)) if result != expected: print source, '..', expected, 'NOT', result else: print source, '..', result Assert([4,5], continueSeries, 1,2,3) # linear series Assert([256,65536], continueSeries, 2,4,16) # Assert([16,32], continueSeries, 2,4,8) # geometric series Assert([8,3], continueSeries, 10,1,9,2) # even interleaved sequence Assert([16,6], continueSeries, 10,2,9,4,8,8,7) # odd interleaved sequence assert 0 == len(take(10, continueSeries(2,4,-8))) # irregular series
Run
Reset
Share
Import
Link
Embed
Language▼
English
中文
Python Fiddle
Python Cloud IDE
Follow @python_fiddle
Browser Version Not Supported
Due to Python Fiddle's reliance on advanced JavaScript techniques, older browsers might have problems running it correctly. Please download the latest version of your favourite browser.
Chrome 10+
Firefox 4+
Safari 5+
IE 10+
Let me try anyway!
url:
Go
Python Snippet
Stackoverflow Question