1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
# assorted sequence helpers
from heapq import heapify, heappop, heappush
class PrioObj(object):
def __init__(self, val, le):
self.val = val
self.le = le
def __str__(self):
return str(self.val)
def __le__(self, other):
return self.le(self.val, other.val)
def imerge(le, *iters):
nxtheap = []
_le = lambda a, b: le(a[0], b[0])
for i in iters:
try:
it = iter(i)
nxtheap.append(PrioObj((it.next(), it), _le))
except StopIteration:
pass
heapify(nxtheap)
while nxtheap:
wrapper = heappop(nxtheap)
x, it = wrapper.val
yield x
try:
wrapper.val = (it.next(), it)
heappush(nxtheap, wrapper)
except StopIteration:
pass
def uniq(seq):
it = iter(seq)
last = it.next()
yield last
for x in it:
if x != last:
last = x
yield x
|
