题目描述
输入两个单调递增的链表,输出两个链表合成后的链表,当然我们需要合成后的链表满足单调不减规则。
开始总是不能通过测试,输入{1,3,5} {2,4,6} ,我的输出{1,3,4,5,6} 我的2去哪里了,自以为代码没写错,又提交了好几遍(事实证明只会降低通过比例。。。)
原来是求初始的pHead2时,直接复制上一步(==改成!=),用的pHead2 = (pHead != pHead1) ? (pHead2->next) : pHead2; 然鹅,就算之前pHead等于pHead1,经过上一步,pHead1已经往后走了一步,现在肯定不等了...... 所以pHead2总会往后走一步,2被忽略。
class Solution {
public:
ListNode* Merge(ListNode* pHead1, ListNode* pHead2)
{
if(pHead1 == NULL || pHead2 == NULL) return (pHead1 == NULL) ? pHead2 : pHead1;
ListNode* pHead;
pHead = (pHead1->val <= pHead2->val) ? pHead1 : pHead2;
pHead1 = (pHead == pHead1) ? (pHead1->next) : pHead1;
pHead2 = (pHead == pHead2) ? (pHead2->next) : pHead2;
ListNode* node = pHead;
while(pHead1 != NULL && pHead2 != NULL)
{
if(pHead1->val <= pHead2->val)
{
node->next = pHead1;
node = pHead1;
pHead1 = pHead1->next;
}
else
{
node->next = pHead2;
node = pHead2;
pHead2 = pHead2->next;
}
}
if(pHead2 == NULL)
node->next = pHead1;
else
node->next = pHead2;
return pHead;
}
};
优化
这样有些麻烦,还容易出错,干脆new一个新节点temp,node初始化为该节点。然后pHead1,pHead2从头开始比较大小,最后直接返回temp->next,就是最后生成的链表的头节点了。嘿嘿嘿
but,还有递归的解法啊,总是想不到唉
拎出最小的节点,剩余的节点排序,拎出的节点的next指针指向函数对剩余节点做处理后的返回值。