无法理解表达

问题描述

我正在做一个 leetcode 问题，在查看解决方案时，我对解决方案中的某行感到非常傻眼。 leetcode 问题尤其是 https://leetcode.com/problems/binary-subarrays-with-sum/

解决办法：

class Solution:
    def numSubarraysWithSum(self,A: List[int],S: int) -> int:

        res = 0
        sm = 0
        sums = collections.defaultdict(int)

        for a in A:
            sm += a
            res += sums[sm - S] + (sm == S)
            sums[sm] += 1
        return res

我理解除了这条线之外发生的一切：

   res += sums[sm - S] + (sm == S)

我以前从未在加法运算中见过条件。

解决方法

条件表达式计算为布尔值，在 Python 中只是 int 的子类型。 False 为 0，True 为 1，因此如果 sm == S 等于 sm，添加 S 与添加 1 或否则添加 0 相同。