求解集合A与B的差集

已知集合A和B的元素分别用不含头结点的单链表存储,函数difference( )用于求解集合A与B的差集,并将结果保存在集合A的单链表中。例如,若集合A={5,10,20,15,25,30},集合B={5,15,35,25},完成计算后A={10,20,30}。

链接:https://www.nowcoder.com/questionTerminal/f06d3cd6112a47dfba3ec3bfb3fb2131
来源:牛客网

#include<stdio.h>

struct node{
    int elem;
    struct node* next;
};

/*
    本题的解法思路较简单:
    因为要求集合A和集合B的差集(A-B),结果保存在集合A中.
    所以我们取出集合A中每一个元素,然后在集合B中寻找(代码22行所实现) 找到即删除此节点 否则保留 
    此时while循环跳出只有两种情况,pb为NULL或者 pa->elem==pb->elem
    当pb不为null时,即找出集合A和集合B的公有元素,此时要删除这个元素 
        pre(实现删除节点之后的链接) 
        当pre为NULL时,说明是首次找到A和B的公有元素,此时 *LA指向pa->next 所以*LA仍然是头结点
        当pre不为NULL时, pre指向pa->next,顺利实现删除节点的链接 
*/


void difference(node** LA,node* LB){
    node*pa,*pb,*pre,*q;
    pre=NULL;
    pa=*LA;/*LA是指向指针的指针,pa指向集合的元素*/
    while(pa){
        pb=LB;/*pb指向集合B的元素*/
        while(pb && pa->elem!=pb->elem) /*在链表LB中寻找与pa所指元素相等的节点*/
            pb=pb->next;
        if(pb){ /*pa所指元素与pb所指元素相等*/
            if(!pre){
                *LA=pa->next;
            }else{
                pre->next=pa->next;
            }
            q=pa;/*求差集 所以要删除pa节点*/
            pa=pa->next;
            free(q);
        }else{
            pre=pa;
            pa=pa->next;
        }
    }
}

你可能感兴趣的:(数据结构)