Handling Sum Queries After Update

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18

class Solution:
    def handleQuery(self, nums1: List[int], nums2: List[int], queries: List[List[int]]) -> List[int]:
        l = len(nums1)
        init = 1 << l
        nums1 = int("".join(map(str, nums1)), 2)
        cur= sum(nums2)
        ans =[]
        for f,s,t in queries:
            if f==1:
                a = (init>>s)-1
                b = (init>>(t+1))-1
                nums1 = nums1^a^b
            elif f==2:
                cnt =nums1.bit_count()
                cur+=s*cnt
            else:
                ans.append(cur)
        return ans