Go 和 PHP 基于两组数计算相加的结果

Go 和 PHP 基于两组数计算相加的结果

author: he xiaodong date: 2020-04-18

原文链接:go leetcode,php 代码个人原创

两数相加(Add-Two-Numbers)

这是 LeetCode 的第二题,题目挺常规的,题干如下:

给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。 如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。 您可以假设除了数字 0 之外,这两个数都不会以 0 开头。 示例:     输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)     输出:7 -> 0 -> 8     原因:342 + 465 = 807 来源:力扣

解题思路

这个题目的解法类似我们小时候算两个多位数的加法:从低位开始相加,大于等于10时则取值的个位数,并向前进1

只不过现在多位数用链表来表示了并且最后的值用链表返回了而已。

根据上面这个思路需要注意的是进位( carry ),相加的时要加上 carry 的值。

Go 实现

func addTwoNumbersOptimize(l1 *ListNode, l2 *ListNode) *ListNode {
    var (
        carry        int
        currentValue int
        centerValue  int
        point        = &ListNode{}
        head         = point
    )
    for l1 != nil || l2 != nil || carry != 0 {
        lValue := 0
        rValue := 0
        if l1 != nil {
            lValue = l1.Val
            l1 = l1.Next
        }
        if l2 != nil {
            rValue = l2.Val
            l2 = l2.Next
        }
        centerValue = lValue + rValue + carry
        currentValue = centerValue % 10
        carry = centerValue / 10
        point.Next = &ListNode{Val: currentValue}
        point = point.Next

    }
    return head.Next
}

PHP 实现

function addTwoNumbers($first, $second)
{
    $carry = 0;
    $str = '';

    // 循环次数以较长的数组为准
    $firstCount = count($first);
    $secondCount = count($second);
    $count = $firstCount > $secondCount ? $firstCount : $secondCount;

    for ($i = 0; $i < $count; $i++) {
        $firstValue = $first[0] ?? 0;
        $secondValue = $second[0] ?? 0;
        array_shift($first);
        array_shift($second);

        // 计算,额外处理最后一次的情况,在空字符串前追加数据
        $centerValue = $firstValue + $secondValue + $carry;
        $currentValue = $i === $count - 1 ? (int)$centerValue : $centerValue % 10;
        $carry = $centerValue / 10;
        $str = $currentValue . $str;
    }

    return $str;
}

echo addTwoNumbers([2, 4, 3], [5, 6, 4]); // output: 807
echo addTwoNumbers([2, 4, 3], [5, 6, 7]); // output: 1107