链表拼接 (20分)
本题要求实现一个合并两个有序链表的简单函数。链表结点定义如下:
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
代码如下(嘻嘻嘻,正能量):
#include<stdio.h>
struct ListNode *mergelists(struct ListNode *list1, struct ListNode *list2)
{ int t,k=0;
struct ListNode *p,*q,*head,*m,*n;
m=list1;n=list2;
while(m!=NULL||n!=NULL)
{ k++;
if(m->data<n->data)
{
p=(struct ListNode*)malloc(sizeof(struct ListNode));
p->data=m->data;
m=m->next;
if(k==1) head=p;
else q->next=p;
q=p;
if(m==NULL){t=0;break;}
}
else
{
p=(struct ListNode*)malloc(sizeof(struct ListNode));
p->data=n->data;
n=n->next;
if(k==1) head=p;
else q->next=p;
q=p;
if(n==NULL){t=1;break;}
}
}
if(t==0)
while(n!=NULL)
{
p=(struct ListNode*)malloc(sizeof(struct ListNode));
p->data=n->data;
q->next=p;
q=p;
n=n->next;
}
if(t==1)
while(m!=NULL)
{
p=(struct ListNode*)malloc(sizeof(struct ListNode));
p->data=m->data;
q->next=p;
q=p;
m=m->next;
}
q->next=NULL;
return(head);
}