实战数据结构(3)_两个单链表间的合并操作

/************************************************************************/
/* 要求：
链表a非递减,链表b非递减
现在要求：
1.实现两个链表的合并,并且元素也是非递减
2.不能占用多余节点,即只能用已有的节点,不能开辟额外的节点
exp:
list a:2 4 4 5 7 8 
list b:3 4 6 9 
则合并后：
list a: 2 3 4 4 4 5 6 7 8 9 
                              */
/************************************************************************/
/************************************************************************/
/* 问题：
  1.还是不能一次写对代码，还是要考编译器来查找错误，指针使用很容易内存泄露
  2.还是在最后一个节点时候，scur=NULL了。就不能引用了scur->data内存泄露

                                                                  */
/************************************************************************/


#include <stdio.h>
#include <stdlib.h>

typedef struct node
{
	int data;
	struct node *next;
}listnode;


void CreateList(listnode *head,int nodenum,int *data);
void printflist(listnode *head);
void DeleElement(listnode *lista,listnode *listb,int *leave,int dataa);
void mergelist(listnode *head1,listnode *head2,int length_a,int length_b);

void main()
{
	int dataa[]={2,4,4,5,7,8};
	int datab[]={3,4,6,9,11,25,69,78};
	listnode *head1=(listnode *)malloc(sizeof(listnode));
	listnode *head2=(listnode *)malloc(sizeof(listnode));
	if(head1==NULL|| NULL==head2)
	{
		printf("头结点malloc分配失败\n");
		exit(-1);
	}
	 int length_a=sizeof(dataa)/sizeof(dataa[0]);
	 int length_b=sizeof(datab)/sizeof(datab[0]);
	CreateList(head1,length_a,dataa);
	printf("链表A的元素有%d个：\n",length_a);
	printflist(head1);
	CreateList(head2,length_b,datab);
	printf("链表B的元素有%d个：\n",length_b);
	printflist(head2);
	mergelist(head1,head2,length_a,length_b);
}

void CreateList(listnode *head,int nodenum,int *data)
{
	
    listnode *h=head,*pre=h,*newnode;   
    for(int i=0;i<nodenum;i++)  
    {     
        if(NULL==(newnode=(listnode *)malloc(sizeof(listnode))))//开辟一个新节点  
        {  
            printf("malloc申请失败");  
            return ;  
        }  
		newnode->data=data[i];
        pre->next=newnode;  
        pre=newnode;  
    }  
    pre->next=NULL;  
    return ;  

}

void printflist(listnode *head)
{
	listnode *p=head->next;
	while(p!=NULL)
	{	
		printf("%4d",p->data);
		p=p->next;
	}
	printf("\n");
	return ;	
}

void DeleElement(listnode *lista,listnode *listb,int *leave,int dataa)
{
	//链表a删除b在a中出现的元素
	listnode *cur_a=lista->next,*pre_a=lista,*cur_b=listb->next;
	int count=0;
	while(cur_a!=NULL) //控制整个流程
	{
		while(cur_b!=NULL && cur_a->data!=cur_b->data )		//将链表b中的元素在a中遍历
				cur_b=cur_b->next;
		if(NULL!=cur_b)		//找到了 将a中此节点 删除
		{
			pre_a->next=cur_a->next;
			free(cur_a);
			cur_a=pre_a->next; //这里漏掉了    当前元素也要后继一下
			count++;
		}
		else	//当前a指向的元素在b中都没有,cur_a向后移动一位
		{	
			pre_a=cur_a;
			cur_a=cur_a->next;
		}
			cur_b=listb->next; //不管查找了没有 b都需要从新开始匹配
	}
	*leave=dataa-count; //返回剩下a中元素个数
	return ;
}

void mergelist(listnode *head1,listnode *head2,int length_a,int length_b)
{
	printf("合并两个链表后元素个数为%d且不递减顺序为:\n",length_a+length_b);
	listnode *long_head=(length_a>=length_b)? head1:head2;
	listnode *short_head=(length_a<length_b)? head1:head2;//先比较两个链表的长度，短的用来控制主流程 时间复杂度低
	listnode *lpre=long_head,*scur=short_head->next; //lpre指向long的头结点,scur指向short第一个节点
	listnode *lcur=lpre->next; //lcur指向long的第一个节点
	listnode *temp;
	while(lcur!=NULL && scur!=NULL) //short控制主流程 当short　NULL时结束 并且long不为最后一个节点
	{
		while(scur->data > lcur->data) //当short>long时,long 后继一位
		{	//当遇到最后一个节点时候，lcur==NULL了，不能再lcur->data了。这里 lcur!=NULL要放到逻辑前
			lpre=lcur;
			lcur=lcur->next;
		}
		if(scur->data <= lcur->data) //当short遇到不大于的数了 此时lcur就是要插入的点
		{	
			temp=scur->next;	//记录short下一个节点
			lpre->next=scur;
			scur->next=lcur;    //将scur插入到long中
			lpre=lpre->next;	//这里lpre后继一位 保持在lcur前一个节点 lcur不变
			scur=temp;		//scur 后继一位
		}
	}
	if(NULL==lcur)	//short 要插入的节点为最后一个节点 直接插入到最后
	{	
		lpre->next=scur; //直接插入
		scur=scur->next;	//到了最后一个节点了，short也应该是最后一个节点了 要scur=NULL
	}
	free(short_head);		//注意要释放短的头节点
	printflist(long_head);
}