855. 考场就座

在考场里，一排有 N 个座位，分别编号为 0, 1, 2, ..., N-1 。

当学生进入考场后，他必须坐在能够使他与离他最近的人之间的距离达到最大化的座位上。如果有多个这样的座位，他会坐在编号最小的座位上。(另外，如果考场里没有人，那么学生就坐在 0 号座位上。)

返回 ExamRoom(int N) 类，它有两个公开的函数：其中，函数 ExamRoom.seat() 会返回一个 int （整型数据），代表学生坐的位置；函数 ExamRoom.leave(int p) 代表坐在座位 p 上的学生现在离开了考场。请确保每次调用 ExamRoom.leave(p) 时都有学生坐在座位 p 上。

 

示例：

输入：["ExamRoom","seat","seat","seat","seat","leave","seat"], [[10],[],[],[],[],[4],[]]
输出：[null,0,9,4,2,null,5]
解释：
ExamRoom(10) -> null
seat() -> 0，没有人在考场里，那么学生坐在 0 号座位上。
seat() -> 9，学生最后坐在 9 号座位上。
seat() -> 4，学生最后坐在 4 号座位上。
seat() -> 2，学生最后坐在 2 号座位上。
leave(4) -> null
seat() -> 5，学生最后坐在 5 号座位上。

 

提示：

1. 1 <= N <= 10^9
2. 在所有的测试样例中 ExamRoom.seat() 和 ExamRoom.leave() 最多被调用 10^4 次。
3. 调用 ExamRoom.leave(p) 时需要确保当前有学生坐在座位 p 上。

O(N)就可以AC，就是各种Corner case很蛋疼

class ExamRoom:

    def __init__(self, N):
        """
        :type N: int
        """
        self.intervals = [[0,N-1]]
        self.N = N

    def seat(self):
        """
        :rtype: int
        """
        if len(self.intervals)==1 and self.intervals[0][0]==0:
            self.intervals[0][0]=1
            return 0
        
        maxInterval,maxIdx, maxPos = -100, -1, -1
        for i,(s,t) in enumerate(self.intervals):
            if s>t: continue
            if s==0: d,pos=t,0
            elif t==self.N-1: d,pos=self.N-1-s,self.N-1
            else: d,pos=(s+t)//2-s,(s+t)//2
            
            if d>maxInterval or (d==maxInterval and pos<maxPos):
                maxInterval=d
                maxIdx = i
                maxPos = pos
        
        s,t = self.intervals.pop(maxIdx)
        self.intervals.append((s,maxPos-1))
        self.intervals.append((maxPos+1,t))
        return maxPos

    def leave(self, p):
        """
        :type p: int
        :rtype: void
        """
        if p==0:
            for i,(s,t) in enumerate(self.intervals):
                if s==1:
                    self.intervals.pop(i)
                    self.intervals.append([0,t])
                    return
        elif p==self.N-1:
            for i,(s,t) in enumerate(self.intervals):
                if t==self.N-2:
                    self.intervals.pop(i)
                    self.intervals.append(([s,self.N-1]))
                    return
                
        ss=tt=-1
        for i,(s,t) in enumerate(self.intervals):
            if t==p-1: ss=i
            if s==p+1: tt=i
        s1,_ = self.intervals.pop(ss)
        _,t2 = self.intervals.pop(tt if tt<ss else tt-1)
        self.intervals.append([s1,t2])
                


# Your ExamRoom object will be instantiated and called as such:
obj = ExamRoom(8)
print(obj.seat())
print(obj.seat())
print(obj.seat())
obj.leave(0)
obj.leave(7)
print(obj.seat())
print(obj.seat())
print(obj.seat())
print(obj.seat())
print(obj.seat())
print(obj.seat())
print(obj.seat())



#obj = ExamRoom(7)
#print(obj.seat())
#print(obj.seat())
#print(obj.seat())
#print(obj.seat())
#print(obj.seat())
#print(obj.seat())
#print(obj.seat())
#obj.leave(3)
#print(obj.seat())
#obj.leave(2)


#obj = ExamRoom(10)
#print(obj.seat())
#print(obj.seat())
#print(obj.seat())
#obj.leave(0)
#obj.leave(4)
#print(obj.seat())
#print(obj.seat())
#print(obj.seat())
#print(obj.seat())
#print(obj.seat())
#print(obj.seat())
#print(obj.seat())
#print(obj.seat())
#print(obj.seat())


#obj = ExamRoom(10)
#print(obj.seat())
#print(obj.seat())
#print(obj.seat())
#print(obj.seat())
#obj.leave(4)
#print(obj.seat())


#obj = ExamRoom(1)
#print(obj.seat())
#obj.leave(0)
#print(obj.seat())

#obj = ExamRoom(3)
#print(obj.seat())
#print(obj.seat())
#print(obj.seat())
#obj.leave(1)