在LeetCode上刷了一题比较基础的算法题,一开始也能解出来,不过在解题过程中用了比较多的if判断,看起来代码比较差,经过思考和改进把原来的算法优化了。
题目:
给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4) 输出:7 -> 0 -> 8 原因:342 + 465 = 807
思路:
因为两个链表的大小不一定相同,所以这样不利于循环的时候对同一位置的数进行计算,所以采取补0的措施:
原链表一:1 -> 5 -> 3 -> 2 = 2351
原链表二:5 -> 6 -> 9 = 965
补0后:
原链表一:1 -> 5 -> 3 -> 2 -> null = 2351
原链表二:5 -> 6 -> 9 -> 0 -> null = 0965
这样补0后数大小没有改变,在循环的时候反而能更好处理,而不用判断两个链表同一节点上其中一个是否为空。
然后使用循环对链表进行遍历,计算同一节点的两个数字相加,把相加的数赋值给新链表的同一节点。如果有进位,相加的数值保留个位,下一次遍历的时候,两个节点相加还要加一。
上面的补0后的链表,可以看出,最终循环结束是当链表一的next和链表二的next为null,反过来说就是当链表一的next不等于链表二的next时要一直循环。
下面贴上代码:
/** * Definition for singly-linked list. * public class ListNode { * int val; * ListNode next; * ListNode(int x) { val = x; } * } */ class Solution { public ListNode addTwoNumbers(ListNode l1, ListNode l2) { // node是头结点,node.next才是真正的结点开始 ListNode node = new ListNode(0); // 0结点,其中任意一个结点长度不一致使用0结点补充 ListNode zero = new ListNode(0); // 指针 ListNode point = node; // node1 + node2 + level的数值相加总和 int sum = 0; // 进位标志 int level = 0; do{ point.next = new ListNode(0); point = point.next; sum = l1.val + l2.val + level; level = sum / 10; point.val = sum % 10; l1 = l1.next == null ? zero : l1.next; l2 = l2.next == null ? zero : l2.next; } while(l1 != l2); if(level == 1) { point.next = new ListNode(1); } return node.next; } }
这里重点说一下
(1)level = sum / 10 :对10整除,那么就得到了进位数,如果sum小于10,则会得到0。
(2)point.val = sum % 10 :对10去余,得到个位数的值。