Commit fef26fb

committed

solves min average waiting time in python

1 parent 38ce8a8 commit fef26fbCopy full SHA for fef26fb

File tree

+28

-0

lines changed

+28

-0

lines changed

Lines changed: 28 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,28 @@`
	`1`	`+from heapq import heappush, heappop`
	`2`	`+`
	`3`	`+tasks = []`
	`4`	`+`
	`5`	`+N = int(input())`
	`6`	`+`
	`7`	`+for _ in range(N):`
	`8`	`+ arrival, cook_time = map(int, input().split())`
	`9`	`+ tasks.append((arrival, cook_time))`
	`10`	`+`
	`11`	`+tasks.sort(reverse=True)`
	`12`	`+`
	`13`	`+pq = []`
	`14`	`+time_waiting = 0`
	`15`	`+current_time = 0`
	`16`	`+`
	`17`	`+while tasks or pq:`
	`18`	`+ while tasks and tasks[-1][0] <= current_time:`
	`19`	`+ heappush(pq, tasks.pop()[::-1])`
	`20`	`+ if pq:`
	`21`	`+ current_task = heappop(pq)`
	`22`	`+ current_time += current_task[0]`
	`23`	`+ time_waiting += current_time - current_task[1]`
	`24`	`+ else:`
	`25`	`+ heappush(pq, tasks.pop()[::-1])`
	`26`	`+ current_time = pq[0][1]`
	`27`	`+`
	`28`	`+print(time_waiting // N)`

Comments

(0)