結果

問題 No.759 悪くない忘年会にしような！
ユーザー matsu7874
提出日時 2018-12-06 02:50:33
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,056 ms / 2,000 ms
コード長 1,675 bytes
コンパイル時間 770 ms
コンパイル使用メモリ 81,708 KB
実行使用メモリ 107,116 KB
最終ジャッジ日時 2024-09-13 12:53:34
合計ジャッジ時間 23,742 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 64
権限があれば一括ダウンロードができます

ソースコード

diff # 

class SegmentTree():

    def __init__(self, size, initial=-1):
        node_size = 1
        while node_size < size:
            node_size <<= 1
        self.st = [initial] * ((node_size << 1) - 1)
        self.size = node_size

    def update(self, target, new_value):
        target += self.size - 1
        self.st[target] = new_value
        while target > 0:
            parent = (target - 1) >> 1
            if self.st[parent] < self.st[target]:
                self.st[parent] = self.st[target]
                target = parent
            else:
                break

    def query(self, a, b):
        # [a,b)の最大値を求める。
        return self._query(a, b, 0, 0, self.size)

    def _query(self, a, b, k, l, r):
        if r <= a or b <= l:
            return -1
        if a <= l and r <= b:
            return self.st[k]
        vl = self._query(a, b, (k << 1) + 1, l, (l + r) >> 1)
        vr = self._query(a, b, (k << 1) + 2, (l + r) >> 1, r)
        return max(vl, vr)


def main():
    n = int(input())

    points = [tuple(map(int, input().split())) for i in range(n)]

    v_max = 10**4
    partition_by_r = [[] for i in range(v_max + 1)]

    pt_max = SegmentTree(v_max + 1, -1)
    res = []

    for i, point in enumerate(points):
        p, t, r = point
        partition_by_r[r].append((p, t, i + 1))
    for ptis in partition_by_r[::-1]:
        ptis.sort(reverse=True)
        for pti in ptis:
            p, t, i = pti
            if t > pt_max.query(p, v_max + 1):
                res.append(i)
                pt_max.update(p, t)

    res.sort()
    for point_id in res:
        print(point_id)


if __name__ == '__main__':
    main()
0