原文链接: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