本题要求实现一个合并两个有序链表的简单函数。链表结点定义如下:
struct ListNode {
int data;
struct ListNode *next;
};
函数接口定义
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
我的做法:
先比较list1和list2的data大小,谁小谁当head(p:记录初始链表排列顺序),另一个记another(q:记录初始链表排列顺序)。
写出链表为空的特殊情况。
写一个循环让another链表里的结点不断加入head里,注意:p和another都变成空的时候结束!
若head先空,让another跟在它后面。
我的思路:
通过代码:
struct ListNode *mergelists(struct ListNode *list1, struct ListNode *list2){
struct ListNode *head,*another,*p,*q,*s;
int flag=1,flag2=1;
if(list1==NULL)
return list2;
else if(list2==NULL)
return list1;
if(list1->data>list2->data){
head=list2;
p=list2->next;
s=list2;
another=list1;
q=list1->next;
}
else{
head=list1;
p=list1->next;
s=list1;
another=list2;
q=list2->next;
}
if(s==list1)
flag=0;
while(p!=NULL&&another!=NULL){
if(p->data>another->data){
another->next=p;//!注意顺序不要搞反!
head->next=another;//!
head=head->next;
another=q;
if(q!=NULL)//不要让q溢出!不然会段错误
q=q->next;
}else{
head=p;
p=p->next;
}
if(p==NULL){
flag2=0;//记录谁先为空
break;
}else if(another==NULL){
flag2=1;
break;
}
}
if(flag2==0){
head->next=another;//即让另外一个跟在其后
}
if(flag==1){//分辨开头结点
return list2;
}else{
return list1;
}
}