本题要求实现一个合并两个有序链表的简单函数。
函数接口定义:
struct ListNode *mergelists(struct ListNode *list1, struct ListNode *list2);
其中list1
和list2
是用户传入的两个按data
升序链接的链表的头指针;函数mergelists
将两个链表合并成一个按data
升序链接的链表,并返回结果链表的头指针。
裁判测试程序样例:
#include <stdio.h>
#include <stdlib.h>
struct ListNode {
int data;
struct ListNode *next;
};
struct ListNode *createlist(); /*裁判实现,细节不表*/
struct ListNode *mergelists(struct ListNode *list1, struct ListNode *list2);
void printlist( struct ListNode *head )
{
struct ListNode *p = head;
while (p) {
printf("%d ", p->data);
p = p->next;
}
printf("\n");
}
int main()
{
struct ListNode *list1, *list2;
list1 = createlist();
list2 = createlist();
list1 = mergelists(list1, list2);
printlist(list1);
return 0;
}
/* 你的代码将被嵌在这里 */
输入样例:
1 3 5 7 -1
2 4 6 -1
输出样例:
1 2 3 4 5 6 7
通过代码:
struct ListNode *mergelists(struct ListNode *list1, struct ListNode *list2){
struct ListNode *head,*pnew,*px,*temp;
if(list1==NULL) return list2;
if(list2==NULL) return list1; //检查链表是否为空
//判断list1、list2中第一个节点那个更小
if(list1->data >= list2->data){
head = list2;
pnew = list2; //用pnew指向第一个节点更小的list
px = list1;
}
else{
head=list1;
pnew=list1;
px=list2;
}
while(1){ //交替拼接
while(pnew -> next != NULL && pnew->next->data <= px->data){
pnew = pnew->next; //仍是pnew所指的链表的下一个节点更小,让pnew后移
}
//已遍历完pnew所指的全部节点,将px所指链表接到pnew所指链表后
if(pnew -> next == NULL){
pnew -> next = px;
return head;
}
/// pnew->next->data > px->next ,交换指针
else{
temp = pnew->next;
pnew -> next = px; //将px链接到pnew后面
pnew = px; //pnew后移
px = temp; //px指向原pnew所指节点的下一个节点
}
}
}