LeetCode-Python-1073. 负二进制数相加

最新推荐文章于 2023-05-18 23:00:38 发布

原创最新推荐文章于 2023-05-18 23:00:38 发布 · 723 阅读

1 ·

CC 4.0 BY-SA版权

Leetcode 同时被 2 个专栏收录

821 篇文章

订阅专栏

Python

664 篇文章

订阅专栏

本文介绍了一种处理-2进制数相加的算法，通过将输入的两个数组形式的-2进制数转换为整数，进行加法运算后，再将结果转换回-2进制数组形式输出。示例展示了如何使用该算法计算两个特定-2进制数的和。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给出基数为 -2 的两个数 arr1 和 arr2，返回两数相加的结果。

数字以 数组形式 给出：数组由若干 0 和 1 组成，按最高有效位到最低有效位的顺序排列。例如，arr = [1,1,0,1] 表示数字 (-2)^3 + (-2)^2 + (-2)^0 = -3。数组形式 的数字也同样不含前导零：以 arr 为例，这意味着要么 arr == [0]，要么 arr[0] == 1。

返回相同表示形式的 arr1 和 arr2 相加的结果。两数的表示形式为：不含前导零、由若干 0 和 1 组成的数组。

示例：

输入：arr1 = [1,1,1,1,1], arr2 = [1,0,1]
输出：[1,0,0,0,0]
解释：arr1 表示 11，arr2 表示 5，输出表示 16 。

提示：

1 <= arr1.length <= 1000
1 <= arr2.length <= 1000
arr1 和 arr2 都不含前导零
arr1[i] 为 0 或 1
arr2[i] 为 0 或 1

思路：

先转INT再加再转-2进制数组输出。

class Solution(object):
    def addNegabinary(self, arr1, arr2):
        """
        :type arr1: List[int]
        :type arr2: List[int]
        :rtype: List[int]
        """
        def convertInt(string):
            tmp = 1
            string = string[::-1]
            res = 0
            for i, x in enumerate(string):
                res += tmp * int(x)
                tmp *= -2
                
            return res
        n1 = convertInt(arr1)
        n2 = convertInt(arr2)
        # print n1, n2, convertInt(self.baseNeg2(n1 + n2))
        return self.baseNeg2(n1 + n2)
    
    
    
    def baseNeg2(self, N):
        """
        :type N: int
        :rtype: str
        """
        res = []
        # n = N
        while N:
            N, b = divmod(N, 2)
            N = -N
            res.append(str(b))
        return "".join(res[::-1]) or "0"
        # return "0" if not n else "".join(res[::-1])