《数据结构》2.5-将链表A分解成B和C

/*
设计一个算法，将带头结点的单链表A分解成两个结构相同的单链表B和C
使得B中的元素是A中大于等于0的元素，C中的元素是A中小于0的元素。
要求存储空间仍使用A的存储空间。 
*/

#include<stdio.h>
typedef struct LNode{
	int data;
	struct LNode *next;
}LNode,*LinkList;

/*
初始化单链表 
*/
int InitList(LinkList &L){
	L=new LNode;
	L->next=NULL;
	return 1;
}

//创建链表 
void CreateList(LinkList &L,int n){
	L=new LNode;
	L->next=NULL;
	for(int i=0;i<n;i++){
		printf("请输入第%d个元素的值:",i+1);
		struct LNode *p;
		p=new LNode;
		scanf("%d",&p->data);
		p->next=L->next;
		L->next=p;
	}
}

//遍历 
void TraveList(LinkList L){
	struct LNode *p;
	p=L->next;
	while(p){
		printf("%d ",p->data);
		p=p->next;
	}
	printf("\n");
}

void CutList(LinkList &A,LinkList &B,LinkList &C){
	struct LNode *pa,*pb,*pc;
	pa=A->next;
	//B=C=A;
	B=new LNode;
	C=new LNode;
	B->next=NULL;
	C->next=NULL;
	pb=B;
	pc=C;
	while(pa){
		if(pa->data>=0){
			pb->next=pa;
			pa=pa->next;
			pb=pb->next;
			pb->next=NULL;//这一句不能少 
		}else if(pa->data<0){
			pc->next=pa;
			pa=pa->next;
			pc=pc->next;
			pc->next=NULL;//不能少 
		}
	} 
}

int main(){
	LinkList A,B,C;
	
	if(InitList(A)){
		printf("链表A初始化成功!\n");
	}else{
		printf("链表A初始化失败!\n");
	}
	
	if(InitList(B)){
		printf("链表B初始化成功!\n");
	}else{
		printf("链表B初始化失败!\n");
	}
	
	if(InitList(C)){
		printf("链表C初始化成功!\n");
	}else{
		printf("链表C初始化失败!\n");
	}
	
	printf("请输入链表A的长度:");
	int n1;
	scanf("%d",&n1);
	CreateList(A,n1);
	printf("链表A的结构如下:\n");
	TraveList(A);
	
	CutList(A,B,C);
	
	printf("B链表的结构如下:\n");
	TraveList(B);
	
	printf("C链表的结构如下:\n");
	TraveList(C);
	
	return 0; 
}