Problem:


Score : 700 points

Problem Statement

We have a rooted binary tree with N vertices, where the vertices are numbered 1 to N.
Vertex 1 is the root, and the parent of Vertex i (i \geq 2) is Vertex \left[ \frac{i}{2} \right].

Each vertex has one item in it. The item in Vertex i has a value of V_i and a weight of W_i.
Now, process the following query Q times:

Given are a vertex v of the tree and a positive integer L.
  Let us choose some (possibly none) of the items in v and the ancestors of v so that their total weight is at most L.
  Find the maximum possible total value of the chosen items.

Here, Vertex u is said to be an ancestor of Vertex v when u is an indirect parent of v, that is, there exists a sequence of vertices w_1,w_2,\ldots,w_k (k\geq 2) where w_1=v, w_k=u, and w_{i+1} is the parent of w_i for each i.

Constraints

All values in input are integers.

1 \leq N < 2^{18}

1 \leq Q \leq 10^5

1 \leq V_i \leq 10^5

1 \leq W_i \leq 10^5

For the values v and L given in each query, 1 \leq v \leq N and 1 \leq L \leq 10^5.

Input

Let v_i and L_i be the values v and L given in the i-th query.
Then, Input is given from Standard Input in the following format:

N
V_1 W_1
:
V_N W_N
Q
v_1 L_1
:
v_Q L_Q

Output

For each integer i from 1 through Q,
the i-th line should contain the response to the i-th query.

Sample Input 1

3
1 2
2 3
3 4
3
1 1
2 5
3 5

Sample Output 1

0
3
3

In the first query, we are given only one choice: the item with (V, W)=(1,2). Since L = 1, we cannot actually choose it, so our response should be 0.

In the second query, we are given two choices: the items with (V, W)=(1,2) and (V, W)=(2,3). Since L = 5, we can choose both of them, so our response should be 3.

Sample Input 2

15
123 119
129 120
132 112
126 109
118 103
115 109
102 100
130 120
105 105
132 115
104 102
107 107
127 116
121 104
121 115
8
8 234
9 244
10 226
11 227
12 240
13 237
14 206
15 227

Sample Output 2

256
255
250
247
255
259
223
253



Sample input:
3
1 2
2 3
3 4
3
1 1
2 5
3 5


Sample output:
0
3
3

Submission status: Runtime Error

Solution code:
#!/usr/bin/env python3

def main():
    import sys
    def I(): return int(sys.stdin.readline())
    def LI(): return list(map(int,sys.stdin.readline().split()))

    n = I()
    V = []
    W = []
    m = min(n, 1 << 10)
    maxL = 10**5
    dp = [[0]*(maxL + 1) for _ in range(m + 1)]
    
    for i in range(1, m+1):
        v, w = LI()
        for j in range(maxL+1):
            if j < w:
                dp[i][j] = dp[i>>1][j]
            else:
                dp[i][j] = max(dp[i>>1][j], dp[i>>1][j - w] + v)
        V.append(v)
        W.append(w)
    
    q = I()
    for i in range(q):
        v, L = LI()
        if v <= m:
            print(dp[v][L])
            continue
        cnt = 0
        tmp = v
        v_ = []
        w_ = []
        while tmp > m:
            v_.append(V[tmp-1])
            w_.append(W[tmp-1])
            tmp >>= 1
            cnt += 1
        ret = dp[tmp][L]
        v_tmp = [0] * (1 << cnt)
        w_tmp = [0] * (1 << cnt)
        for b in range(1, 1 << cnt):
            lsb = b & -b
            l = lsb.bit_length()
            w_tmp[b] = w_tmp[b - lsb] + w_[l-1]
            if w_tmp[b] <= L:
                v_tmp[b] = v_tmp[b - lsb] + v_[l-1]
                ret = max(ret, v_tmp[b] + dp[tmp][L - w_tmp[b]])
        print(ret)
    return

if __name__ == "__main__":
    main()

Submission status: Runtime Error

Solution code:
#!/usr/bin/env python3
import sys

#solve
def solve():

    input = sys.stdin.buffer.readline
    LI = lambda: map(int, input().split())
    II = lambda: int(input())
    n = II()
    vw = [[]] * n
    M = 10 ** 5 + 1
    M0 = [0] * M
    XX = 1 << 9
    d = [[]] * XX
    d[0] = M0

    for i in range(1, min(XX, n + 1)):
        v, w = LI()
        vw[i - 1] = (v, w)
        di = d[i // 2][:]
        for j in range(M - 1, w - 1, -1):
            if di[j] < di[j - w] + v:
                di[j] = di[j - w] + v
        d[i] = di

    for i in range(XX, n + 1):
        v, w = LI()
        vw[i - 1] = (v, w)

    Vv = [i.bit_length() for i in range(n + 1)]
    n = [pow(2, i) for i in range(18)]
    lentl = [i - 9 for i in range(19)]
    maskset = [[]] * n[9]

    for i in range(1, 10):
        for mask in range(n[i - 1], n[i]):
            a = mask & -mask
            b = Vv[a] - 1
            maska = mask - a
            maskset[mask - 1] = (mask, b, maska)

    for _ in range(II()):
        V, L = LI()

        if V < XX:
            print(d[V][L])
            continue

        n2 = lentl[Vv[V]]
        vw2 = [None] * n2

        for i in range(n2):
            vw2[i] = (vw[V - 1][0], vw[V - 1][1])
            V //= 2

        lw = [0] * n[n2]
        lv = [0] * n[n2]
        dv = d[V]
        ans = dv[L]

        for i in range(n[n2] - 1):
            mask, b, maska = maskset[i]
            v, w = vw2[b]
            ww = lw[maska] + w
            Lww = L - ww
            lw[mask] = ww
            if Lww < 0: continue
            vv = lv[maska] + v
            lv[mask] = vv
            tmp = dv[Lww] + vv
            if tmp > ans:
                ans = tmp
        print(ans)
    return

if __name__ == "__main__":
    solve()

Submission status: Runtime Error

Solution code:
#!/usr/bin/env python3
import sys

#solve
def solve():

    input = sys.stdin.buffer.readline
    LI = lambda: map(int, input().split())
    II = lambda: int(input())
    n = II()
    vw = [[]] * n
    M = 10 ** 5 + 1
    M0 = [0] * M
    XX = 1 << 9
    d = [[]] * XX
    d[0] = M0

    for i in range(1, min(XX, n + 1)):
        v, w = LI()
        vw[i - 1] = (v, w)
        di = d[i // 2][:]
        for j in range(M - 1, w - 1, -1):
            if di[j] < di[j - w] + v:
                di[j] = di[j - w] + v
        d[i] = di

    for i in range(XX, n + 1):
        v, w = LI()
        vw[i - 1] = (v, w)

    vv = [i.bit_length() for i in range(n + 1)]
    n = [pow(2, i) for i in range(18)]
    lentl = [i - 9 for i in range(19)]
    maskset = [[]] * n[9]

    for i in range(1, 10):
        for mask in range(n[i - 1], n[i]):
            a = mask & -mask
            b = a.bit_length() - 1
            maska = mask - a
            maskset[mask - 1] = (mask, b, maska)

    for _ in range(II()):
        V, L = LI()

        if V < XX:
            print(d[V][L])
            continue

        n2 = lentl[vv[v]]
        vw2 = [None] * n2

        for i in range(n2):
            vw2[i] = (vw[V - 1][0], vw[V - 1][1])
            V //= 2

        lw = [0] * n[n2]
        lv = [0] * n[n2]
        dv = d[V]
        ans = dv[L]

        for i in range(n[n2] - 1):
            mask, b, maska = maskset[i]
            v, w = vw2[b]
            ww = lw[maska] + w
            Lww = L - ww
            lw[mask] = ww
            if Lww < 0: continue
            vv = lv[maska] + v
            lv[mask] = vv
            tmp = dv[Lww] + vv
            if tmp > ans:
                ans = tmp
        print(ans)
    return

if __name__ == "__main__":
    solve()

Submission status: Runtime Error

Solution code:
import sys
input = sys.stdin.buffer.readline

def main():
    N = int(input())
    VW = [list(map(int, input().split())) for _ in range(N)]
    Q = int(input())
    Query = [list(map(int, input().split())) for _ in range(Q)]

    dp = [[] for _ in range(1<<11)]
    dp[0] = [(0, 0)]
    for k in range(min(1<<11, N)):
        nv, nw = VW[k]
        k += 1
        ndp = []
        for v, w in dp[k//2]:
            ndp.append((v+nv, w+nw))
        ind = 0
        for v, w in dp[k//2]:
            while ind < len(ndp) and ndp[ind][1] < w:
                dp[k].append(ndp[ind])
                ind += 1
            if ind < len(ndp) and ndp[ind][1] == w:
                if ndp[ind][0] > v:
                    dp[k].append(ndp[ind])
                else:
                    dp[k].append((v, w))
                ind += 1
            else:
                dp[k].append((v, w))
        while ind < len(ndp):
            dp[k].append(ndp[ind])
            ind += 1

    for s, L in Query:
        dp2 = [(0, 0)]
        # make new dp
        while s >= (1<<11):
            nv, nw = VW[s-1]
            ndp = []
            for v, w in dp2:
                if w + nw <= L:
                    ndp.append((v+nv, w+nw))
            ind = 0
            new = []
            for v, w in dp2:
                while ind < len(ndp) and ndp[ind][1] < w:
                    new.append(ndp[ind])
                    ind += 1
                if ind < len(ndp) and ndp[ind][1] == w:
                    if ndp[ind][0] > v:
                        new.append(ndp[ind])
                    else:
                        new.append((v, w))
                    ind += 1
                else:
                    new.append((v, w))
            while ind < len(ndp):
                new.append(ndp[ind])
                ind += 1
            dp2 = new
            s = s//2
        
        # merge
        ind2 = len(dp2)-1
        ans = 0
        for v, w in dp[s]:
            while ind2 > 0 and dp2[ind2][1] + w > L:
                ind2 -= 1
            if ind2 == 0 and dp2[ind2][1] + w > L: break
            score = dp2[ind2][0] + v
            if score > ans:
                ans = score
        print(ans)

if __name__ == "__main__":
    main()

Submission status: Runtime Error

Solution code:
#!/usr/bin/env python3
import sys

#solve
def solve():
    input = sys.stdin.buffer.readline
    LI = lambda: map(int, input().split())
    II = lambda: int(input())
    n = II()
    vw = [[]] * n
    nbit = 1 << n.bit_length()
    M = 10 ** 5 + 1
    M0 = [0] * M
    XX = 1 << n.bit_length() // 2
    d = [[]] * XX
    d[0] = M0
    for i in range(1, XX):
        v, w = LI()
        vw[i - 1] = (v, w)
        di = d[i // 2][:]
        for j in range(M - 1, w - 1, -1):
            if di[j] < di[j - w] + v:
                di[j] = di[j - w] + v
        d[i] = di
    for i in range(XX, n + 1):
        v, w = LI()
        vw[i - 1] = (v, w)
    n = [pow(2, i) for i in range(18)]
    lentl = [i - i // 2 for i in range(18)]
    for _ in range(II()):
        V, L = LI()
        n2 = lentl[V.bit_length()]
        if V < XX:
            print(d[V][L])
            continue
        vw2 = [None] * n2
        for i in range(n2):
            vw2[i] = (vw[V - 1][0], vw[V - 1][1])
            V //= 2
        lw = [0] * n[n2]
        lv = [0] * n[n2]
        dv = d[V]
        ans = dv[L]
        for mask in range(1, n[n2]):
            a = mask & -mask
            b = a.bit_length() - 1
            v, w = vw2[b]
            maska = mask - a
            ww = lw[maska] + w
            Lww = L - ww
            lw[mask] = ww
            if Lww < 0: continue
            vv = lv[maska] + v
            lv[mask] = vv
            tmp = dv[Lww] + vv
            if tmp > ans:
                ans = tmp
        print(ans)
    return

if __name__ == "__main__":
    solve()

Submission status: Runtime Error

Solution code:
import math
from collections import deque
import time
import numpy as np

def filter(w1,w2,v):
    if w2<=w1:
        return v
    else:
        return 0

def main():
    N=int(input())
    V=[]
    W=[]

    M = pow(10, 5)
    D = int(math.log(N, 2)) + 1  # bound of tree
    if D >= 2:
        D = D // 2

    dp=np.zeros((N,M+1))

    BoundIndex = pow(2, D) - 2

    for i in range(N):
        v,w=map(int,input().split())
        if i==0:
            for w_ in range(M+1):
                dp[i,w_]=filter(w_,w,v)

        elif i<pow(2,D)-1:
            for w_ in range(M+1):
                if (w_>=w):
                    dp[i,w_]=max(dp[(i - 1) // 2,w_ - w] + v, dp[(i - 1) // 2,w_])
                else:
                    dp[i,w_]=dp[(i-1)//2,w_]

        V.append(v)
        W.append(w)

    V=np.array(V)
    W=np.array(W)
    Q=int(input())
    v=[]
    L=[]
    R=[]
    for i in range(Q):
        v_q,l = map(int,input().split())
        if v_q - 1 > BoundIndex:
            d = int(math.log(v_q, 2)) + 1
            k = d - D
            ALL = pow(2, k)
            rest_v = [v_q - 1]
            v_ = v_q - 1
            for _ in range(k - 1):
                v_ = (v_ - 1) // 2
                rest_v.append(v_)

            parent_node = (v_ - 1) // 2
            res = 0
            for a in range(ALL):
                restW = 0
                restV = 0
                # bit search
                for flag_index in range(k):
                    if a & 1 == 1:
                        ind = rest_v[flag_index]
                        restW += W[ind]
                        restV += V[ind]
                    a = a >> 1
                if l - restW >= 0:
                    res = max(res, restV + dp[parent_node,int(l - restW)])
            R.append(res)
        else:
            res=dp[v_q - 1,l]
            R.append(res)

    for i in range(Q):
        print(int(R[i]))



if __name__ == "__main__":
    main()

Submission status: Runtime Error

Solution code:
import sys
from math import log2
def input(): return sys.stdin.readline().strip()


def main():
    N = int(input())
    item = [(0, 0)]
    for _ in range(N):
        v, w = map(int, input().split())
        item.append((v, w))
    queri = []
    L = 0 # L = max(l_i)
    Q = int(input())
    for _ in range(Q):
        v, l = map(int, input().split())
        queri.append((v, l))
        L = max(L, l)

    """
    各クエリごとにナップサックをやったら間に合わないが、
    本問ではこれを木の上半分のみに適用し、下半分は全探索をおこなう。
    すると計算量はO(2^9 * L + 2^9 * Q)に落ちて間に合う！！！
    「半分全探索」の一種？
    """

    # 木の高さをnとして求めておく（N <= 2^n - 1）
    n = 0
    while 2 ** n <= N: n += 1
    halfN = 1 << (n // 2 + 1)

    # 木の上半分にいつものナップサックを行い事前計算。
    # dp[v][w] = (頂点1から頂点vまで辿った時、重さがw以下となるようにアイテムを選んだ時の価値の最大値)
    dp = [[0] * (L + 1) for _ in range(halfN)]
    for v in range(1, halfN):
        value, weight = item[v]
        for w in range(L + 1):
            if w >= weight:
                dp[v][w] = max(dp[v // 2][w], dp[v // 2][w - weight] + value)
            else:
                dp[v][w] = dp[v // 2][w]

    # クエリに回答
    for v, l in queri:
        if v < halfN:
            print(dp[v][l])
            continue

        # 木の下半分にいるvの先祖を全列挙。
        # 同時に後々bitDPする都合上itemも先祖列上に取り直す。
        num = 0
        #ancestor = [] # 先祖の頂点番号使わなかったわ
        item_temp = []
        current = v
        while current >= halfN:
            num += 1
            #ancestor.append(current)
            item_temp.append(item[current])
            current >>= 1

        # それらの中から選び方を全列挙し、各選びかたに対して最大価値を導出する。
        # ここでビット全探索する中で立っているビットを検索すると計算量が*10倍になるので、
        # 全探索した各結果（価値を重さ）は配列に記録してbitDPを行う。
        ans = dp[current][l]
        value_list = [0] * (1 << num)
        weight_list = [0] * (1 << num)
        for state in range(1, 1 << num):
            lowest_bit = state & (-state) # これで最下位ビットのみ立っている数が得られる
            x = lowest_bit.bit_length() - 1 # lowest_bitのビット位置を0-indexedで求める
            w_temp = weight_list[state - lowest_bit] + item_temp[x][1]
            weight_list[state] = w_temp # 変数を分離させた。キャッシュに乗りやすくなって高速化なるか！？
            if l >= w_temp:
                v_temp = value_list[state - lowest_bit] + item_temp[x][0]
                value_list[state] = v_temp
                ans = max(ans, v_temp + dp[current][l - w_temp])
        print(ans)


if __name__ == "__main__":
    main()

Submission status: Runtime Error

Solution code:
N=int(input())
VW=[[0,0]]+[list(map(int,input().split())) for _ in range(N)]
Q=int(input())
QU=[list(map(int,input().split())) for _ in range(Q)]
W=10**5

x=N
y=1
while x!=1:
  x=x//2
  y+=1
y=y//2 

from collections import deque
dp=[[0]*(W+1) for _ in range(2**y)]
q=deque([1])
#深さ優先探索的に調べる
while q:
  a=q.pop()
  for i in range(W+1):
    if i>=VW[a][1]:
      dp[a][i]=max([dp[a//2][i],dp[a//2][i-VW[a][1]]+VW[a][0]])
    else:
      dp[a][i]=dp[a//2][i]
  if a<2**(y-1):
    q.append(2*a)
    q.append(2*a+1)  

for i in range(Q):
  if QU[i][0]<2**y:
    print(dp[QU[i][0]][QU[i][1]])
  else:
    b=QU[i][0]
    ls=[b]
    while b>=2**y:
      b=b//2
      if b>=2**y:
        ls.insert(0,b)
    #bit全探索    
    A=[]  
    for k in range(2**(len(ls))):
      v=0
      w=0
      for j in range(len(ls)):
        if (k>>j)&1==1:
          v+=VW[ls[j]][0]
          w+=VW[ls[j]][1]
      A.append([v,w])
    ans=0  
    for j in range(len(A)):
      if A[j][1]<=QU[i][1]:
        ans=max([ans,dp[b][QU[i][1]-A[j][1]]+A[j][0]])
    print(ans)

Submission status: Runtime Error

Solution code:
import math
from collections import deque
import time
import numpy as np

def filter(w1,w2,v):
    if w2<=w1:
        return v
    else:
        return 0

def main():
    N=int(input())
    V=[]
    W=[]

    M = pow(10, 5)
    D = int(math.log(N, 2)) + 1  # bound of tree
    if D >= 2:
        D = D // 2

    dp=np.zeros((N,M+1))

    BoundIndex = pow(2, D) - 2

    for i in range(N):
        v,w=map(int,input().split())
        if i==0:
            for w_ in range(M+1):
                dp[i,w_]=filter(w_,w,v)

        elif i<pow(2,D)-1:
            for w_ in range(M+1):
                if (w_>=w):
                    dp[i,w_]=max(dp[(i - 1) // 2,w_ - w] + v, dp[(i - 1) // 2,w_])
                else:
                    dp[i,w_]=dp[(i-1)//2,w_]

        V.append(v)
        W.append(w)

    V=np.array(V)
    W=np.array(W)
    Q=int(input())
    v=[]
    L=[]
    R=[]
    for i in range(Q):
        v_q,l = map(int,input().split())
        if v_q - 1 > BoundIndex:
            d = int(math.log(v_q, 2)) + 1
            k = d - D
            ALL = pow(2, k)
            rest_v = [v_q - 1]
            v_ = v_q - 1
            for _ in range(k - 1):
                v_ = (v_ - 1) // 2
                rest_v.append(v_)

            parent_node = (v_ - 1) // 2
            res = 0
            for a in range(ALL):
                restW = 0
                restV = 0
                # bit search
                for flag_index in range(k):
                    if a & 1 == 1:
                        ind = rest_v[flag_index]
                        restW += W[ind]
                        restV += V[ind]
                    a = a >> 1
                if l - restW >= 0:
                    res = max(res, restV + dp[parent_node,l - restW])
            R.append(res)
        else:
            res=dp[v_q - 1,l]
            R.append(res)

    for i in range(Q):
        print(int(R[i]))



if __name__ == "__main__":
    main()

Submission status: Runtime Error

Solution code:
#!/usr/bin/env python3

def main():
    import sys
    def I(): return int(sys.stdin.readline())
    def LI(): return list(map(int,sys.stdin.readline().split()))

    n = I()
    V = []
    W = []
    for _ in range(n):
        v, w = LI()
        V.append(v)
        W.append(w)
    
    q = I()
    vl = []
    maxL = 0
    for _ in range(q):
        v, l = LI()
        maxL = max(maxL, l)
        vl.append([v,l])
    
    m = min(n, 1 << 10)
    dp = [[0]*(maxL + 1) for _ in range(m + 1)]
    
    for i in range(1, m+1):
        v, w = V[i-1], W[i-1]
        for j in range(maxL+1):
            if j < w:
                dp[i][j] = dp[i>>1][j]
            else:
                dp[i][j] = max(dp[i>>1][j], dp[i>>1][j - w] + v)
    
    for i in range(q):
        v, L = vl[i]
        if v <= m:
            print(dp[v][L])
            continue
        cnt = 0
        tmp = v
        v_ = []
        w_ = []
        while tmp > m:
            v_.append(V[tmp-1])
            w_.append(W[tmp-1])
            tmp >>= 1
            cnt += 1
        ret = 0
        v_tmp = [0] * (1 << cnt)
        w_tmp = [0] * (1 << cnt)
        for b in range(1 << cnt):
            lsb = b & -b
            l = lsb.bit_length()
            w_tmp[b] = w_tmp[b - lsb] + w_[l-1]
            if w_tmp[b] <= L:
                v_tmp[b] = v_tmp[b - lsb] + v_[l-1]
                ret = max(ret, v_tmp[b] + dp[v][L - w_tmp[b]])
        print(ret)
    return

if __name__ == "__main__":
    main()

Submission status: Runtime Error

Solution code:
#!/usr/bin/env python3
from collections import defaultdict
import sys
input = sys.stdin.buffer.readline
LI = lambda:map(int, input().split())
II = lambda:int(input())

#solve
def solve():
    n = II()
    vw = []
    nbit = 1 << n.bit_length()
    M = 10 ** 5 + 1
    M0 = [0] * M
    d = [M0]
    dp = [[]]
    XX = 1 << n.bit_length() // 2
    for _ in range(1, XX):
        v, w = LI()
        vw.append((v, w))
        di = d[i // 2][:]
        for j in range(M - 1, w - 1, -1):
            if di[j] < di[j - w] + v:
                di[j] = di[j - w] + v
        d.append(di)
    for _ in range(XX, n + 1):
        v, w = LI()
        vw.append((v, w))
    n = [pow(2, i) for i in range(18)]
    for _ in range(II()):
        V, L = LI()
        lent = V.bit_length()
        if V < XX:
            print(d[V][L])
            continue
        n1 = lent // 2
        n2 = lent - n1
        vw2 = []
        for _ in range(n2):
            vw2.append((vw[V - 1][0], vw[V - 1][1]))
            V //= 2
        lw = [0] * n[n2]
        lv = [0] * n[n2]
        dv = d[V]
        ans = dv[L]
        for mask in range(1, n[n2]):
            a = mask & -mask
            b = a.bit_length() - 1
            v, w = vw2[b]
            maska = mask - a
            ww = lw[maska] + w
            Lww = L - ww
            lw[mask] = ww
            if Lww < 0: continue
            vv = lv[maska] + v
            lv[mask] = vv
            if dv[Lww] + vv > ans:
                ans = dv[Lww] + vv
        print(ans)
    return


solve()

Submission status: Runtime Error

Solution code:
N=int(input())
VW=[[0,0]]+[list(map(int,input().split())) for _ in range(N)]
Q=int(input())
QU=[list(map(int,input().split())) for _ in range(Q)]
W=10**5

x=N
y=1
while x!=1:
  x=x//2
  y+=1
  
from collections import deque
dp=[[0]*(W+1) for _ in range(2**(y//2))]
q=deque([1])
#深さ優先探索的に調べる
while q:
  a=q.pop()
  for i in range(W+1):
    if i>=VW[a][1]:
      dp[a][i]=max([dp[a//2][i],dp[a//2][i-VW[a][1]]+VW[a][0]])
    else:
      dp[a][i]=dp[a//2][i]
  if a<2**(y//2-1):
    q.append(2*a)
    q.append(2*a+1)  

for i in range(Q):
  if QU[i][0]<2**(y//2):
    print(dp[QU[i][0]][QU[i][1]])
  else:
    b=QU[i][0]
    ls=[b]
    while b>=2**(y//2):
      b=b//2
      if b>=2**(y//2):
        ls.insert(0,b)
    #bit全探索    
    A=[]  
    for k in range(2**(len(ls))):
      v=0
      w=0
      for j in range(len(ls)):
        if (k>>j)&1==1:
          v+=VW[ls[j]][0]
          w+=VW[ls[j]][1]
      A.append([v,w])
    B=[]  
    for j in range(len(A)):
      if A[j][1]<=QU[i][1]:
        B.append(dp[b][QU[i][1]-A[j][1]]+A[j][0])
    print(max(B))

Submission status: Runtime Error

Solution code:
N=int(input())
VW=[[0,0]]+[list(map(int,input().split())) for _ in range(N)]
Q=int(input())
QU=[list(map(int,input().split())) for _ in range(Q)]
W=10**5

x=N
y=1
while x!=1:
  x=x//2
  y+=1
  
from collections import deque
dp=[[0]*(W+1) for _ in range(2**(y//2))]
q=deque([1])
#深さ優先探索的に調べる
while q:
  a=q.pop()
  for i in range(W+1):
    if i>=VW[a][1]:
      dp[a][i]=max([dp[a//2][i],dp[a//2][i-VW[a][1]]+VW[a][0]])
    else:
      dp[a][i]=dp[a//2][i]
  if a<2**(y//2-1):
    q.append(2*a)
    q.append(2*a+1)

for i in range(Q):
  if QU[i][0]<2**(y//2):
    print(dp[QU[i][0]][QU[i][1]])
  else:
    b=QU[i][0]
    ls=[b]
    while b>=2**(y//2):
      b=b//2
      if b>=2**(y//2):
        ls.insert(0,b)
    #bit全探索    
    A=[]  
    for k in range(2**(len(ls))):
      v=0
      w=0
      for j in range(len(ls)):
        if (k>>j)&1==1:
          v+=VW[ls[j]][0]
          w+=VW[ls[j]][1]
      A.append([v,w])
    B=[]  
    for j in range(len(A)):
      B.append(dp[b][QU[i][1]-A[j][1]]+A[j][0])
    print(max(B))

Submission status: Runtime Error

Solution code:
import sys
input=sys.stdin.readline

def calc():
  n=int(input())
  arr=[list(map(int,input().split())) for _ in range(n)]
  dp=[[0]*(10**5+1) for _ in range(2**10+1)]
  for i in range(1,min(n,2**10)+1):
    v,w=arr[i-1]
    for j in range(10**5,w-1,-1):
      dp[i][j]=max(dp[i//2][j],dp[i//2][j-w]+v)
    for j in range(w-1,-1,-1):
      dp[i][j]=dp[i//2][j]
  q=int(input())
  for _ in range(q):
    v,l=map(int,input().split())
    if v<=2**10:
      print(dp[v][l])
    else:
      vs=[]
      cnt=0
      while v>2**10:
        vs.append(v)
        v//=2
        cnt+=1
      ans=0
      for i in range(2**cnt):
        weight=0
        value=0
        for j in range(cnt):
          if i&(1<<j):
            weight+=arr[vs[j]][1]
            value+=arr[vs[j]][0]
        if weight>l:
          continue
        ans=max(ans,dp[v][l-weight]+value)
      print(ans)
calc()

Submission status: Runtime Error

Solution code:
#!python3

LI = lambda: list(map(int, input().split()))

# input
N = int(input())
VW = [LI() for _ in range(N)]
Q = int(input())
VL = [LI() for _ in range(Q)]

MAX = 10 ** 5 + 1
k = N.bit_length()
n = 1 << ((k >> 1) + 1)
dp = [[0] * MAX for _ in range(n)]


def solve(i, v, l):
    if l < 0:
        return 0
    if i < n:
        return dp[i][l] + v
    x = solve(i >> 1, v + VW[i - 1][0], l - VW[i - 1][1])
    y = solve(i >> 1, v, l)
    return max(x, y)


def main():
    dp[0][0] = 0
    for i in range(1, n):
        p = i // 2
        v, w = VW[i - 1]
        dp[i][0] = 0
        for j in range(MAX):
            dp[i][j] = max(dp[p][j], dp[i][j - 1])
            if j >= w:
                dp[i][j] = max(dp[i][j], dp[p][j - w] + v)
    
    for v, l in VL:
        ans = solve(v, 0, l)
        print(ans)


if __name__ == "__main__":
    main()

Submission status: Runtime Error

Solution code:
import sys
input=sys.stdin.readline

def calc():
  n=int(input())
  arr=[list(map(int,input().split())) for _ in range(n)]
  dp=[[0]*(10**5+1) for _ in range(2**9+1)]
  for i in range(1,min(n,2**9)+1):
    v,w=arr[i-1]
    for j in range(10**5,w-1,-1):
      dp[i][j]=max(dp[i//2][j],dp[i//2][j-w]+v)
    for j in range(w-1,-1,-1):
      dp[i][j]=dp[i//2][j]
  q=int(input())
  for _ in range(q):
    v,l=map(int,input().split())
    if v<=2**9:
      print(dp[v][l])
    else:
      vs=[]
      cnt=0
      while v>2**9:
        vs.append(v)
        v//=2
        cnt+=1
      ans=0
      for i in range(2**cnt):
        weight=0
        value=0
        for j in range(cnt):
          if i&(1<<j):
            weight+=arr[vs[j]][1]
            value+=arr[vs[j]][0]
        if weight>l:
          continue
        ans=max(ans,dp[v][l-weight]+value)
      print(ans)
calc()

Submission status: Runtime Error

Solution code:
#!/usr/bin/env python3
from collections import defaultdict
import sys
input = sys.stdin.buffer.readline
LI = lambda:map(int, input().split())
II = lambda:int(input())

#solve
def solve():
    n = II()
    vw = []
    nbit = 1 << n.bit_length()
    M = 10 ** 5 + 1
    M0 = [0] * M
    d = [M0]
    for i in range(n):
        v, w = LI()
        vw.append((v, w))
    dp = [[]]
    for i in range(1, 1 << n.bit_length() // 2):
        di = M0[:]
        n2 = i.bit_length()
        lw = [0] * (1 << n2)
        lv = [0] * (1 << n2)
        dpi = dp[i // 2][:]
        dpi.append((vw[i - 1][0], vw[i - 1][1]))
        for mask in range(1, 1 << n2):
            a = mask & -mask
            b = a.bit_length() - 1
            v, w = dpi[b]
            ww = lw[mask - a] + w
            lw[mask] = ww
            if ww >= M: continue
            vv = lv[mask - a] + v
            lv[mask] = vv
            if di[ww] < vv:
                di[ww] = vv
        for j in range(M - 1):
            if di[j + 1] < di[j]:
                di[j + 1] = di[j]
        d.append(di)
    for _ in range(II()):
        V, L = LI()
        lent = V.bit_length()
        n1 = lent // 2
        n2 = lent - n1
        vw2 = []
        for _ in range(n2):
            vw2.append((vw[V - 1][0], vw[V - 1][1]))
            V //= 2
        lw = [0] * (1 << n2)
        lv = [0] * (1 << n2)
        dv = d[V]
        ans = dv[L]
        for mask in range(1, 1 << n2):
            a = mask & -mask
            b = a.bit_length() - 1
            v, w = vw2[b]
            ww = lw[mask - a] + w
            lw[mask] = ww
            if ww > L: continue
            vv = lv[mask - a] + v
            lv[mask] = vv
            if dv[L - ww] + vv > ans:
                ans = dv[L - ww] + vv
        print(ans)
    return


solve()

Submission status: Runtime Error

Solution code:
import sys
input = sys.stdin.readline
import numpy as np
from numba import njit

@njit
def solve_dp(l_max, k):
    dp = [[0] * (l_max + 1) for _ in  range(2 ** k)]
    for i in range(1, 2 ** k):
        for j in range(l_max + 1):
            if j >= weight[i]:
                dp[i][j] = max(dp[i//2][j], 
                               dp[i//2][j-weight[i]] + value[i])
            else:
                dp[i][j] = dp[i//2][j]
    return dp

@njit
def main(n, k, value, weight, query_v, query_l):
    # 木全体の半分の深さまでdpテーブルを作っておく
    l_max = max(query_l)
    dp = solve_dp(l_max, k)

    # 残りの半分についてはクエリ毎に処理
    for v, l in zip(query_v, query_l):
        if v < 2 ** k:
            print(dp[v][l])
            continue

        indices = []
        tmp = v
        while tmp >= 2 ** k:
            indices.append(tmp)
            tmp //= 2
        li = len(indices)
        
        ans = 0
        for item in range(1<<li):
            item = [(item >> i) & 1 for i in range(li)]
            v_sum, w_sum = 0, 0
            for i in range(li):
                if item[i]:
                    idx = indices[i]
                    v_sum += value[idx]
                    w_sum += weight[idx]
            if w_sum > l:
                continue
            ans = max(ans, v_sum + dp[tmp][l-w_sum])
        print(ans)


n = int(input())
k = (n-1).bit_length() // 2
value = [-1]
weight = [-1]
for _ in range(n):
    v, w = map(int, input().split())
    value.append(v)
    weight.append(w)

q = int(input())
query_v, query_l = [], []
for _ in range(q):
    v, l = map(int, input().split())
    query_v.append(v)
    query_l.append(l)

value, weight = np.array(value), np.array(weight)
query_v, query_l = np.array(query_v), np.array(query_l)

main(n, k, value, weight, query_v, query_l)