我需要创建一个程序,将2个数字分别添加到2个链接列表中。我让程序工作了,但我在最后得到一个额外的-1值,我不需要。我如何去掉它?下面是一些背景。
/*
Definition for singly-linked list.
public class ListNode {
public int val;
public ListNode next;
public ListNode(int x) { val = x; }
}
*/
public class Solution {
public ListNode AddTwoNumbers(ListNode l1, ListNode l2) {
int a, k = 0;
ListNode l3 = new ListNode(-1), curr = l3;
do
{
if(k == 1)
a = l1.val + l2.val + 1;
else
a = l1.val + l2.val;
k = 0;
if(a >= 10){
curr.val = a - 10;
k = 1;
}
else
curr.val = a;
l1 = l1.next; l2 = l2.next;
curr.next = new ListNode(-1);
curr = curr.next;
}while(l1 != null && l2 != null);
return l3;
}
}
如果我对你的问题理解正确的话,我想这是因为你在循环结束时创建了一个新节点。在这一行
curr.next = new ListNode(-1);
通过在周期结束时创建一个新的节点... do...while
周期,列表中的最后一个节点将始终包含这个 -1
. 你想要的是在创建新节点之前评估是否满足终止条件。
试试下面的代码,其中我把创建新节点的工作交给了post condition。
public class Solution {
public ListNode AddTwoNumbers(ListNode l1, ListNode l2) {
ListNode l3 = new ListNode(-1), curr = l3;
for(int a=0, k = 0; l1 != null && l2 != null ; curr.next = new ListNode(-1), curr = curr.next)
{
if(k == 1)
a = l1.val + l2.val + 1;
else
a = l1.val + l2.val;
k = 0;
if(a >= 10){
curr.val = a - 10;
k = 1;
}
else
curr.val = a;
l1 = l1.next; l2 = l2.next;
}
return l3;
}
}
你还需要考虑其他的边缘情况,比如一个数字比另一个数字的位数多。 l1 = 7 -> 5 和 l2 = 8 -> 6 -> 5
private static ListNode AddTwoNumbers(ListNode l1, ListNode l2)
{
if (l1 == null) return l2;
if (l2 == null) return l1;
ListNode l3 = new ListNode(-1), cur = l3;
int a, k = 0;
while (l1 != null || l2 != null)
{
if (k == 1)
a = (l1 == null ? 0 : l1.val) + (l2 == null ? 0 : l2.val) + 1;
else
a = (l1 == null ? 0 : l1.val) + (l2 == null ? 0 : l2.val);
k = 0;
if (a >= 10)
{
cur.val = a - 10;
k = 1;
}
else
cur.val = a;
l1 = l1?.next;
l2 = l2?.next;
if (l1 == null && l2 == null)
break;
cur.next = new ListNode(-1);
cur = cur.next;
}
if (k == 1)
cur.next = new ListNode(1);
return l3;
}