Problem:


Score : 500 points

Problem Statement

There is an infinitely long street that runs west to east, which we consider as a number line.

There are N roadworks scheduled on this street.
The i-th roadwork blocks the point at coordinate X_i from time S_i - 0.5 to time T_i - 0.5.

Q people are standing at coordinate 0. The i-th person will start the coordinate 0 at time D_i, continue to walk with speed 1 in the positive direction and stop walking when reaching a blocked point.

Find the distance each of the Q people will walk.

Constraints

All values in input are integers.

1 \leq N, Q \leq 2 \times 10^5

0 \leq S_i < T_i \leq 10^9

1 \leq X_i \leq 10^9

0 \leq D_1 < D_2 < ... < D_Q \leq 10^9

If i \neq j and X_i = X_j, the intervals [S_i, T_i) and [S_j, T_j) do not overlap.

Input

Input is given from Standard Input in the following format:

N Q
S_1 T_1 X_1
:
S_N T_N X_N
D_1
:
D_Q

Output

Print Q lines. The i-th line should contain the distance the i-th person will walk or -1 if that person walks forever.

Sample Input 1

4 6
1 3 2
7 13 10
18 20 13
3 4 2
0
1
2
3
5
8

Sample Output 1

2
2
10
-1
13
-1

The first person starts coordinate 0 at time 0 and stops walking at coordinate 2 when reaching a point blocked by the first roadwork at time 2.

The second person starts coordinate 0 at time 1 and reaches coordinate 2 at time 3. The first roadwork has ended, but the fourth roadwork has begun, so this person also stops walking at coordinate 2.

The fourth and sixth persons encounter no roadworks while walking, so they walk forever. The output for these cases is -1.



Sample input:
4 6
1 3 2
7 13 10
18 20 13
3 4 2
0
1
2
3
5
8


Sample output:
2
2
10
-1
13
-1

Submission status: Time Limit Exceeded

Solution code:
def main():
    import sys
    from operator import itemgetter
    input = sys.stdin.readline

    class SegTree():
        def __init__(self, A):
            n_ = len(A)
            self.n = 1
            while self.n < n_:
                self.n *= 2
            self.tree = [1] * (2 * self.n - 1)
            for k, a in enumerate(A):
                self.update(k, a)
        def update(self, k, a):
            k += self.n - 1
            self.tree[k] = a
            while k > 0:
                k = (k - 1) // 2
                self.tree[k] = min(self.tree[k * 2 + 1], self.tree[k * 2 + 2])
        def query(self, l, r):
            l += self.n - 1
            r += self.n - 2
            res = 1
            while r-l>1:
                if l&1 == 0:
                    res = min(res,self.tree[l])
                if r&1 == 1:
                    res = min(res,self.tree[r])
                    r -= 1
                l = l//2
                r = (r-1)//2
            if l == r:
                res = min(res,self.tree[l])
            else:
                res = min(min(res,self.tree[l]),self.tree[r])
            return res

    n, q = map(int, input().split())
    stx = []
    tmp = set([])

    for _ in range(n):
        s,t,x = map(int, input().split())
        stx.append((s,t,x))
        tmp.add(x-t)
        tmp.add(x-s)

    d = {}
    for i in range(q):
        di = -int(input())
        tmp.add(di)
        d[di] = i

    stx.sort(key = itemgetter(2))

    v2i = {}
    tmp = list(tmp)
    tmp.sort()

    for i, v in enumerate(tmp):
        v2i[v] = i

    l = [1] * (len(tmp) + 1)

    for di in d:
        l[v2i[di]] = di
            
    sgt = SegTree(l)
    ans = {}
        
    for s, t, x in stx:
        l = v2i[x-t] + 1
        r = v2i[x-s] + 1
        v = sgt.query(l, r)
        while v != 1:
            i_d = d[v]
            i_sg = v2i[v]
            ans[i_d] = x
            sgt.update(i_sg, 1)
            v = sgt.query(l, r)

    for i in range(q):
        if i in ans:
            print(ans[i])
        else:
            print(-1)
            
if __name__ == '__main__':
    main()

Submission status: Time Limit Exceeded

Solution code:
def main():
    import sys
    from operator import itemgetter
    input = sys.stdin.readline

    INF = 1

    class SegTree():
        def __init__(self, A):
            n_ = len(A)
            self.n = 1
            while self.n < n_:
                self.n *= 2
            self.tree = [INF] * (2 * self.n - 1)
            for k, a in enumerate(A):
                self.update(k, a)
        def update(self, k, a):
            k += self.n - 1
            self.tree[k] = a
            while k > 0:
                k = (k - 1) // 2
                self.tree[k] = min(self.tree[k * 2 + 1], self.tree[k * 2 + 2])
        def query(self, l, r):
            l += self.n - 1
            r += self.n - 2
            res = INF
            while r-l>1:
                if l&1 == 0:
                    res = min(res,self.tree[l])
                if r&1 == 1:
                    res = min(res,self.tree[r])
                    r -= 1
                l = l//2
                r = (r-1)//2
            if l == r:
                res = min(res,self.tree[l])
            else:
                res = min(min(res,self.tree[l]),self.tree[r])
            return res

    n, q = map(int, input().split())
    stx = []
    tmp = set([])

    for _ in range(n):
        s,t,x = map(int, input().split())
        stx.append((s,t,x))
        tmp.add(x-t)
        tmp.add(x-s)

    d = {}
    for i in range(q):
        di = -int(input())
        tmp.add(di)
        d[di] = i

    stx.sort(key = itemgetter(2))

    v2i = {}
    tmp = list(tmp)
    tmp.sort()

    for i, v in enumerate(tmp):
        v2i[v] = i

    l = [1] * (len(tmp) + 1)

    for di in d:
        l[v2i[di]] = di
            
    sgt = SegTree(l)
    ans = {}

    sgt_update = sgt.update
    sgt_query = sgt.query
        
    for s, t, x in stx:
        l = v2i[x-t]
        r = v2i[x-s]
        v = sgt_query(l+1, r+1)
        for _ in d:
            if v == 1:
                break
            i_d = d[v]
            i_sg = v2i[v]
            ans[i_d] = x
            sgt_update(i_sg, 1)
            v = sgt_query(l+1, r+1)

    for i in range(q):
        if i in ans:
            print(ans[i])
        else:
            print(-1)
            
if __name__ == '__main__':
    main()

Submission status: Time Limit Exceeded

Solution code:
def main():
    import sys
    from operator import itemgetter
    input = sys.stdin.readline

    class SegTree():
        def __init__(self, A):
            n_ = len(A)
            self.n = 1
            while self.n < n_:
                self.n *= 2
            self.tree = [1] * (2 * self.n - 1)
            for k, a in enumerate(A):
                self.update(k, a)
        def update(self, k, a):
            k += self.n - 1
            self.tree[k] = a
            while k > 0:
                k = (k - 1) // 2
                self.tree[k] = min(self.tree[k * 2 + 1], self.tree[k * 2 + 2])
        def query(self, l, r):
            l += self.n - 1
            r += self.n - 2
            res = 1
            while r-l>1:
                if l&1 == 0:
                    res = min(res,self.tree[l])
                if r&1 == 1:
                    res = min(res,self.tree[r])
                    r -= 1
                l = l//2
                r = (r-1)//2
            if l == r:
                res = min(res,self.tree[l])
            else:
                res = min(min(res,self.tree[l]),self.tree[r])
            return res

    n, q = map(int, input().split())
    stx = []
    tmp = set([])

    for _ in range(n):
        s,t,x = map(int, input().split())
        stx.append((s,t,x))
        tmp.add(x-t)
        tmp.add(x-s)

    d = {}
    for i in range(q):
        di = -int(input())
        tmp.add(di)
        d[di] = i

    stx.sort(key = itemgetter(2))

    v2i = {}
    tmp = list(tmp)
    tmp.sort()

    for i, v in enumerate(tmp):
        v2i[v] = i

    l = [1] * (len(tmp) + 1)

    for di in d:
        l[v2i[di]] = di
            
    sgt = SegTree(l)
    ans = {}

    sgt_update = sgt.update
    sgt_query = sgt.query
        
    for s, t, x in stx:
        l = v2i[x-t]
        r = v2i[x-s]
        v = sgt_query(l+1, r+1)
        for _ in d:
            if v == 1:
                break
            i_d = d[v]
            i_sg = v2i[v]
            ans[i_d] = x
            sgt_update(i_sg, 1)
            v = sgt_query(l+1, r+1)

    for i in range(q):
        if i in ans:
            print(ans[i])
        else:
            print(-1)
            
if __name__ == '__main__':
    main()

Submission status: Time Limit Exceeded

Solution code:
def main():
    import sys
    from operator import itemgetter
    input = sys.stdin.readline

    class SegTree():
        def __init__(self, A):
            n_ = len(A)
            self.n = 1
            while self.n < n_:
                self.n *= 2
            self.tree = [1 for _ in range(2 * self.n - 1)]
            for k, a in enumerate(A):
                self.update(k, a)
        def update(self, k, a):
            k += self.n - 1
            self.tree[k] = a
            while k > 0:
                k = (k - 1) // 2
                self.tree[k] = min(self.tree[k * 2 + 1], self.tree[k * 2 + 2])
        def query(self, l, r):
            l += self.n - 1
            r += self.n - 2
            res = 1
            while r-l>1:
                if l&1 == 0:
                    res = min(res,self.tree[l])
                if r&1 == 1:
                    res = min(res,self.tree[r])
                    r -= 1
                l = l//2
                r = (r-1)//2
            if l == r:
                res = min(res,self.tree[l])
            else:
                res = min(min(res,self.tree[l]),self.tree[r])
            return res

    n, q = map(int, input().split())
    stx = []
    tmp = set([])

    for _ in range(n):
        s,t,x = map(int, input().split())
        stx.append((s,t,x))
        tmp.add(x-t)
        tmp.add(x-s)

    d = {}
    for i in range(q):
        di = -int(input())
        tmp.add(di)
        d[di] = i

    stx.sort(key = itemgetter(2))

    v2i = {}
    tmp = list(tmp)
    tmp.sort()

    for i, v in enumerate(tmp):
        v2i[v] = i

    l = [1] * (len(tmp) + 1)

    for di in d:
        l[v2i[di]] = di
            
    sgt = SegTree(l)
    ans = {}

    sgt_update = sgt.update
    sgt_query = sgt.query
        
    for s, t, x in stx:
        l = v2i[x-t]
        r = v2i[x-s]
        v = sgt_query(l+1, r+1)
        for _ in d:
            if v == 1:
                break
            i_d = d[v]
            i_sg = v2i[v]
            ans[i_d] = x
            sgt_update(i_sg, 1)
            v = sgt_query(l+1, r+1)

    for i in range(q):
        if i in ans:
            print(ans[i])
        else:
            print(-1)
            
if __name__ == '__main__':
    main()

Submission status: Time Limit Exceeded

Solution code:
def main():
    import sys
    from operator import itemgetter
    input = sys.stdin.readline

    class SegTree():
        def __init__(self, A):
            n_ = len(A)
            self.n = 1
            while self.n < n_:
                self.n *= 2
            self.tree = [1 for _ in range(2 * self.n - 1)]
            for k, a in enumerate(A):
                self.update(k, a)
        def update(self, k, a):
            k += self.n - 1
            self.tree[k] = a
            while k > 0:
                k = (k - 1) // 2
                self.tree[k] = min(self.tree[k * 2 + 1], self.tree[k * 2 + 2])
        def query(self, l, r):
            l += self.n - 1
            r += self.n - 2
            res = 1
            while r-l>1:
                if l&1 == 0:
                    res = min(res,self.tree[l])
                if r&1 == 1:
                    res = min(res,self.tree[r])
                    r -= 1
                l = l//2
                r = (r-1)//2
            if l == r:
                res = min(res,self.tree[l])
            else:
                res = min(min(res,self.tree[l]),self.tree[r])
            return res

    n, q = map(int, input().split())
    stx = []
    tmp = set([])

    for _ in range(n):
        s,t,x = map(int, input().split())
        stx.append((s,t,x))
        tmp.add(x-t)
        tmp.add(x-s)

    d = {}
    for i in range(q):
        di = -int(input())
        tmp.add(di)
        d[di] = i

    stx.sort(key = itemgetter(2))

    v2i = {}
    tmp = list(tmp)
    tmp.sort()

    for i, v in enumerate(tmp):
        v2i[v] = i

    l = [1] * (len(tmp) + 1)

    for di in d:
        l[v2i[di]] = di
            
    sgt = SegTree(l)
    ans = {}

    sgt_update = sgt.update
    sgt_query = sgt.query
        
    for s, t, x in stx:
        l = v2i[x-t]
        r = v2i[x-s]
        v = sgt_query(l+1, r+1)
        while v != 1:
            i_d = d[v]
            i_sg = v2i[v]
            ans[i_d] = x
            sgt_update(i_sg, 1)
            v = sgt_query(l+1, r+1)

    for i in range(q):
        if i in ans:
            print(ans[i])
        else:
            print(-1)
            
if __name__ == '__main__':
    main()

Submission status: Time Limit Exceeded

Solution code:
def main():
    import sys
    from operator import itemgetter
    input = sys.stdin.readline

    class SegTree():
        def __init__(self, A):
            n_ = len(A)
            self.n = 1
            while self.n < n_:
                self.n *= 2
            self.tree = [1 for _ in range(2 * self.n - 1)]
            for k, a in enumerate(A):
                self.update(k, a)
        def update(self, k, a):
            k += self.n - 1
            self.tree[k] = a
            while k > 0:
                k = (k - 1) // 2
                self.tree[k] = min(self.tree[k * 2 + 1], self.tree[k * 2 + 2])
        def query(self, l, r):
            l += self.n - 1
            r += self.n - 2
            res = 1
            while r-l>1:
                if l&1 == 0:
                    res = min(res,self.tree[l])
                if r&1 == 1:
                    res = min(res,self.tree[r])
                    r -= 1
                l = l//2
                r = (r-1)//2
            if l == r:
                res = min(res,self.tree[l])
            else:
                res = min(min(res,self.tree[l]),self.tree[r])
            return res

    n, q = map(int, input().split())
    stx = []
    tmp = set([])

    for _ in range(n):
        s,t,x = map(int, input().split())
        stx.append((s,t,x))
        tmp.add(x-t)
        tmp.add(x-s)

    d = {}
    for i in range(q):
        di = -int(input())
        tmp.add(di)
        d[di] = i

    stx.sort(key = itemgetter(2))

    v2i = {}

    for i, v in enumerate(sorted(list(tmp))):
        v2i[v] = i

    l = [1] * (len(tmp) + 1)

    for di in d:
        l[v2i[di]] = di
            
    sgt = SegTree(l)
    ans = {}

    sgt_update = sgt.update
    sgt_query = sgt.query
        
    for s, t, x in stx:
        l = v2i[x-t]
        r = v2i[x-s]
        v = sgt_query(l+1, r+1)
        while v != 1:
            i_d = d[v]
            i_sg = v2i[v]
            ans[i_d] = x
            sgt_update(i_sg, 1)
            v = sgt_query(l+1, r+1)

    for i in range(q):
        if i in ans:
            print(ans[i])
        else:
            print(-1)
            
if __name__ == '__main__':
    main()

Submission status: Time Limit Exceeded

Solution code:
def main():
    import sys
    input = sys.stdin.readline

    class SegTree():
        def __init__(self, A):
            n_ = len(A)
            self.n = 1
            while self.n < n_:
                self.n *= 2
            self.tree = [1 for _ in range(2 * self.n - 1)]
            for k, a in enumerate(A):
                self.update(k, a)
        def update(self, k, a):
            k += self.n - 1
            self.tree[k] = a
            while k > 0:
                k = (k - 1) // 2
                self.tree[k] = min(self.tree[k * 2 + 1], self.tree[k * 2 + 2])
        def query(self, l, r):
            l += self.n - 1
            r += self.n - 2
            res = 1
            while r-l>1:
                if l&1 == 0:
                    res = min(res,self.tree[l])
                if r&1 == 1:
                    res = min(res,self.tree[r])
                    r -= 1
                l = l//2
                r = (r-1)//2
            if l == r:
                res = min(res,self.tree[l])
            else:
                res = min(min(res,self.tree[l]),self.tree[r])
            return res

    n, q = map(int, input().split())
    stx = []
    tmp = set([])

    for _ in range(n):
        s,t,x = map(int, input().split())
        stx.append((s,t,x))
        tmp.add(x-t)
        tmp.add(x-s)

    d = {}
    for i in range(q):
        di = -int(input())
        tmp.add(di)
        d[di] = i

    stx.sort(key = lambda x:x[2])

    v2i = {}

    for i, v in enumerate(sorted(list(tmp))):
        v2i[v] = i

    l = [1] * (len(tmp) + 1)

    for di in d:
        l[v2i[di]] = di
            
    sgt = SegTree(l)
    ans = {}

    sgt_update = sgt.update
    sgt_query = sgt.query
        
    for s, t, x in stx:
        l = v2i[x-t]
        r = v2i[x-s]
        v = sgt_query(l+1, r+1)
        while v != 1:
            i_d = d[v]
            i_sg = v2i[v]
            ans[i_d] = x
            sgt_update(i_sg, 1)
            v = sgt_query(l+1, r+1)

    for i in range(q):
        if i in ans:
            print(ans[i])
        else:
            print(-1)
            
if __name__ == '__main__':
    main()

Submission status: Time Limit Exceeded

Solution code:
def main():
    import sys
    input = sys.stdin.readline

    class SegTree():
        def __init__(self, A):
            n_ = len(A)
            self.n = 1
            while self.n < n_:
                self.n *= 2
            self.tree = [1 for _ in range(2 * self.n - 1)]
            for k, a in enumerate(A):
                self.update(k, a)
        def update(self, k, a):
            k += self.n - 1
            self.tree[k] = a
            while k > 0:
                k = (k - 1) // 2
                self.tree[k] = min(self.tree[k * 2 + 1], self.tree[k * 2 + 2])
        def query(self, l, r):
            l += self.n - 1
            r += self.n - 2
            res = 1
            while r-l>1:
                if l&1 == 0:
                    res = min(res,self.tree[l])
                if r&1 == 1:
                    res = min(res,self.tree[r])
                    r -= 1
                l = l//2
                r = (r-1)//2
            if l == r:
                res = min(res,self.tree[l])
            else:
                res = min(min(res,self.tree[l]),self.tree[r])
            return res

    n, q = map(int, input().split())
    stx = []
    tmp = set([])

    for _ in range(n):
        s,t,x = map(int, input().split())
        stx.append((s,t,x))
        tmp.add(x-t)
        tmp.add(x-s)

    d = {}
    for i in range(q):
        di = -int(input())
        tmp.add(di)
        d[di] = i

    stx.sort(key = lambda x:x[2])

    v2i = {}

    for i, v in enumerate(sorted(list(tmp))):
        v2i[v] = i

    l = [1] * (len(tmp) + 1)

    for di in d:
        l[v2i[di]] = di
            
    sgt = SegTree(l)
    ans = {}

    sgt_update = sgt.update
    sgt_query = sgt.query
        
    for s, t, x in stx:
        l = v2i[x-t]
        r = v2i[x-s]
        v = sgt_query(l+1, r+1)
        while v != 1:
            i_d = d[v]
            i_sg = v2i[v]
            ans[i_d] = x
            sgt_update(i_sg, 1)
            v = sgt_query(l+1, r+1)

    for i in range(q):
        if i in ans:
            print(ans[i])
        else:
            print(-1)
            

main()

Submission status: Time Limit Exceeded

Solution code:
def main():
    import sys
    input = sys.stdin.readline

    INF = 1

    class SegTree():
        def __init__(self, A):
            n_ = len(A)
            self.n = 1
            while self.n < n_:
                self.n *= 2
            self.tree = [INF for _ in range(2 * self.n - 1)]
            for k, a in enumerate(A):
                self.update(k, a)
        def update(self, k, a):
            k += self.n - 1
            self.tree[k] = a
            while k > 0:
                k = (k - 1) // 2
                self.tree[k] = min(self.tree[k * 2 + 1], self.tree[k * 2 + 2])
        def query(self, l, r):
            l += self.n - 1
            r += self.n - 2
            res = INF
            while r-l>1:
                if l&1 == 0:
                    res = min(res,self.tree[l])
                if r&1 == 1:
                    res = min(res,self.tree[r])
                    r -= 1
                l = l//2
                r = (r-1)//2
            if l == r:
                res = min(res,self.tree[l])
            else:
                res = min(min(res,self.tree[l]),self.tree[r])
            return res

    n, q = map(int, input().split())
    stx = []
    tmp = set([])

    for _ in range(n):
        s,t,x = map(int, input().split())
        stx.append((s,t,x))
        tmp.add(x-t)
        tmp.add(x-s)

    d = {}
    for i in range(q):
        di = -int(input())
        tmp.add(di)
        d[di] = i

    stx.sort(key = lambda x:x[2])

    v2i = {}

    for i, v in enumerate(sorted(list(tmp))):
        v2i[v] = i

    l = [1] * (len(tmp) + 1)

    for di in d:
        l[v2i[di]] = di
            
    sgt = SegTree(l)
    ans = {}

    sgt_update = sgt.update
    sgt_query = sgt.query
        
    for s, t, x in stx:
        l = v2i[x-t]
        r = v2i[x-s]
        v = sgt_query(l+1, r+1)
        while v != 1:
            i_d = d[v]
            i_sg = v2i[v]
            ans[i_d] = x
            sgt_update(i_sg, 1)
            v = sgt_query(l+1, r+1)

    for i in range(q):
        if i in ans:
            print(ans[i])
        else:
            print(-1)
            

main()

Submission status: Time Limit Exceeded

Solution code:
def main():
    import sys
    input = sys.stdin.readline

    INF = 1

    class SegTree():
        def __init__(self, A):
            n_ = len(A)
            self.n = 1
            while self.n < n_:
                self.n *= 2
            self.tree = [INF for _ in range(2 * self.n - 1)]
            for k, a in enumerate(A):
                self.update(k, a)
        def update(self, k, a):
            k += self.n - 1
            self.tree[k] = a
            while k > 0:
                k = (k - 1) // 2
                self.tree[k] = min(self.tree[k * 2 + 1], self.tree[k * 2 + 2])
        def query(self, l, r):
            l += self.n - 1
            r += self.n - 2
            res = INF
            while r-l>1:
                if l&1 == 0:
                    res = min(res,self.tree[l])
                if r&1 == 1:
                    res = min(res,self.tree[r])
                    r -= 1
                l = l//2
                r = (r-1)//2
            if l == r:
                res = min(res,self.tree[l])
            else:
                res = min(min(res,self.tree[l]),self.tree[r])
            return res

    n, q = map(int, input().split())
    stx = []
    tmp = set([])

    for _ in range(n):
        s,t,x = map(int, input().split())
        stx.append([s,t,x])
        tmp.add(x-t)
        tmp.add(x-s)

    d = {}
    for i in range(q):
        di = -int(input())
        tmp.add(di)
        d[di] = i

    stx.sort(key = lambda x:x[2])

    v2i = {}

    for i, v in enumerate(sorted(list(tmp))):
        v2i[v] = i

    l = []
    for _ in range(len(tmp) + 1):
        l.append(1)

    for di in d:
        l[v2i[di]] = di
            
    sgt = SegTree(l)
    ans = {}

    sgt_update = sgt.update
    sgt_query = sgt.query
        
    for s, t, x in stx:
        l = v2i[x-t]
        r = v2i[x-s]
        v = sgt_query(l+1, r+1)
        while v != 1:
            i_d = d[v]
            i_sg = v2i[v]
            ans[i_d] = x
            sgt_update(i_sg, 1)
            v = sgt_query(l+1, r+1)

    for i in range(q):
        if i in ans:
            print(ans[i])
        else:
            print(-1)
            

main()

Submission status: Time Limit Exceeded

Solution code:
def main():
    import sys
    from collections import defaultdict
    input = sys.stdin.readline

    INF = 1

    class SegTree():
        def __init__(self, A):
            n_ = len(A)
            self.n = 1
            while self.n < n_:
                self.n *= 2
            self.tree = [INF for _ in range(2 * self.n - 1)]
            for k, a in enumerate(A):
                self.update(k, a)
        def update(self, k, a):
            k += self.n - 1
            self.tree[k] = a
            while k > 0:
                k = (k - 1) // 2
                self.tree[k] = min(self.tree[k * 2 + 1], self.tree[k * 2 + 2])
        def query(self, l, r):
            l += self.n - 1
            r += self.n - 2
            res = INF
            while r-l>1:
                if l&1 == 0:
                    res = min(res,self.tree[l])
                if r&1 == 1:
                    res = min(res,self.tree[r])
                    r -= 1
                l = l//2
                r = (r-1)//2
            if l == r:
                res = min(res,self.tree[l])
            else:
                res = min(min(res,self.tree[l]),self.tree[r])
            return res

    n, q = map(int, input().split())
    stx = []
    tmp = set([])

    for _ in range(n):
        s,t,x = map(int, input().split())
        stx.append([s,t,x])
        tmp.add(x-t)
        tmp.add(x-s)

    d = {}
    for i in range(q):
        di = -int(input())
        tmp.add(di)
        d[di] = i

    stx.sort(key = lambda x:x[2])

    v2i = {}

    for i, v in enumerate(sorted(list(tmp))):
        v2i[v] = i

    l = [1] * (len(tmp) + 1)

    for di in d:
        l[v2i[di]] = di
            
    sgt = SegTree(l)
    ans = defaultdict(lambda: -1)

    sgt_update = sgt.update
    sgt_query = sgt.query
        
    for s, t, x in stx:
        l = v2i[x-t]
        r = v2i[x-s]
        v = sgt_query(l+1, r+1)
        while v != 1:
            i_d = d[v]
            i_sg = v2i[v]
            ans[i_d] = x
            sgt_update(i_sg, 1)
            v = sgt_query(l+1, r+1)

    for i in range(q):
        print(ans[i])
            

main()

Submission status: Time Limit Exceeded

Solution code:
def main():
    import sys
    input = sys.stdin.readline

    INF = 1

    class SegTree():
        def __init__(self, A):
            n_ = len(A)
            self.n = 1
            while self.n < n_:
                self.n *= 2
            self.tree = [INF for _ in range(2 * self.n - 1)]
            for k, a in enumerate(A):
                self.update(k, a)
        def update(self, k, a):
            k += self.n - 1
            self.tree[k] = a
            while k > 0:
                k = (k - 1) // 2
                self.tree[k] = min(self.tree[k * 2 + 1], self.tree[k * 2 + 2])
        def query(self, l, r):
            l += self.n - 1
            r += self.n - 2
            res = INF
            while r-l>1:
                if l&1 == 0:
                    res = min(res,self.tree[l])
                if r&1 == 1:
                    res = min(res,self.tree[r])
                    r -= 1
                l = l//2
                r = (r-1)//2
            if l == r:
                res = min(res,self.tree[l])
            else:
                res = min(min(res,self.tree[l]),self.tree[r])
            return res

    n, q = map(int, input().split())
    stx = []
    tmp = set([])

    for _ in range(n):
        s,t,x = map(int, input().split())
        stx.append([s,t,x])
        tmp.add(x-t)
        tmp.add(x-s)

    d = {}
    for i in range(q):
        di = -int(input())
        tmp.add(di)
        d[di] = i

    stx.sort(key = lambda x:x[2])

    v2i = {}

    for i, v in enumerate(sorted(list(tmp))):
        v2i[v] = i

    l = [1] * (len(tmp) + 1)

    for di in d:
        l[v2i[di]] = di
            
    sgt = SegTree(l)
    ans = {}

    sgt_update = sgt.update
    sgt_query = sgt.query
        
    for s, t, x in stx:
        l = v2i[x-t]
        r = v2i[x-s]
        v = sgt_query(l+1, r+1)
        while v != 1:
            i_d = d[v]
            i_sg = v2i[v]
            ans[i_d] = x
            sgt_update(i_sg, 1)
            v = sgt_query(l+1, r+1)

    for i in range(q):
        if i in ans:
            print(ans[i])
        else:
            print(-1)
            

main()

Submission status: Time Limit Exceeded

Solution code:
def main():
    import sys
    input = sys.stdin.readline

    INF = 1

    class SegTree():
        def __init__(self, A):
            n_ = len(A)
            self.n = 1
            while self.n < n_:
                self.n *= 2
            self.tree = [INF for _ in range(2 * self.n - 1)]
            for k, a in enumerate(A):
                self.update(k, a)
        def update(self, k, a):
            k += self.n - 1
            self.tree[k] = a
            while k > 0:
                k = (k - 1) // 2
                self.tree[k] = min(self.tree[k * 2 + 1], self.tree[k * 2 + 2])
        def query(self, l, r):
            l += self.n - 1
            r += self.n - 2
            res = INF
            while r-l>1:
                if l&1 == 0:
                    res = min(res,self.tree[l])
                if r&1 == 1:
                    res = min(res,self.tree[r])
                    r -= 1
                l = l//2
                r = (r-1)//2
            if l == r:
                res = min(res,self.tree[l])
            else:
                res = min(min(res,self.tree[l]),self.tree[r])
            return res

    n, q = map(int, input().split())
    stx = []
    tmp = set([])
    tmp_add = tmp.add
    stx_append = stx.append
    
    for _ in range(n):
        s,t,x = map(int, input().split())
        stx_append([s,t,x])
        tmp_add(x-t)
        tmp_add(x-s)

    d = {}
    for i in range(q):
        di = -int(input())
        tmp_add(di)
        d[di] = i

    stx.sort(key = lambda x:x[2])

    v2i = {}

    for i, v in enumerate(sorted(list(tmp))):
        v2i[v] = i

    l = [1] * (len(tmp) + 1)

    for di in d:
        l[v2i[di]] = di

            
    sgt = SegTree(l)
    ans = [-1] * q

    sgt_update = sgt.update
    sgt_query = sgt.query
        
    for s, t, x in stx:
        l = v2i[x-t]
        r = v2i[x-s]
        v = sgt_query(l+1, r+1)
        while v != 1:
            i_d = d[v]
            i_sg = v2i[v]
            ans[i_d] = x
            sgt_update(i_sg, 1)
            v = sgt_query(l+1, r+1)

    for ansi in ans:
        print(ansi)
            

main()