interview/airbnb_page_display.py at master · ryf123/interview · GitHub

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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
from collections import OrderedDict
def page_display(rows,size):
	d = OrderedDict()
	for row in rows:
		host_id = row.split(",")[0]
		if host_id not in d:
			d[host_id] = [row]
		else:
			d[host_id].append(row)
	pages = []
	while len(d)>0:
		page = []
		count = 0
		to_be_deleted = []
		for key in d:
			if count == size:
				break
			page.append(d[key][0])
			d[key].pop(0)
			if len(d[key]) == 0:
				to_be_deleted.append(key)
			count+=1
		for key in to_be_deleted:
			del d[key]
		pages.append(page)
	for i,page in enumerate(pages):
		print "Page: {}; total: {}".format(str(i),str(len(page)))
		print page


data = [
"1,28,300.1,SanFrancisco",
"4,5,209.1,SanFrancisco",
"20,7,208.1,SanFrancisco",
"23,8,207.1,SanFrancisco",
"16,10,206.1,Oakland",
"1,16,205.1,SanFrancisco",
"6,29,204.1,SanFrancisco",
"7,20,203.1,SanFrancisco",
"8,21,202.1,SanFrancisco",
"2,18,201.1,SanFrancisco",
"2,30,200.1,SanFrancisco",
"15,27,109.1,Oakland",
"10,13,108.1,Oakland",
"11,26,107.1,Oakland",
"12,9,106.1,Oakland",
"13,1,105.1,Oakland",
"22,17,104.1,Oakland",
"1,2,103.1,Oakland",
"28,24,102.1,Oakland",
"18,14,11.1,SanJose",
"6,25,10.1,Oakland",
"19,15,9.1,SanJose",
"3,19,8.1,SanJose",
"3,11,7.1,Oakland",
"27,12,6.1,Oakland",
"1,3,5.1,Oakland",
"25,4,4.1,SanJose",
"5,6,3.1,SanJose",
"29,22,2.1,SanJose",
"30,23,1.1,SanJose"
]
page_display(data,12)