求序列的交集（链表）

求序列的交集（链表）

问题描述 :

使用带头结点的单链表编程：
有两个序列，分别表示两个集合。
求它们的交集并输出。

输入说明 :

第一行输入序列A的信息：
第一个整数n(0<=n<=100)，表示共有n个元素，其后有n个整数，表示n个元素的数据
第一行输入序列B的信息：
第一个整数n(0<=n<=100)，表示共有n个元素，其后有n个整数，表示n个元素的数据

输出说明 :

输出交集的元素序列，输出格式见范例。
如果交集为空，则输出“head–>tail”
交集里的元素顺序，依照其在序列A中的顺序。
比如：
序列：
A：5 3 2 7
B：1 3 5 8
则交集为5和3，因为在序列A中，5在3的前面，所以在交集里5也在3的前面。

输入范例 :
4 5 3 2 7
4 1 3 5 8

输出范例 :
head–>5–>3–>tail

#include <iostream>
using namespace std;
struct ListNode
{
	int  num;
	struct ListNode  *next;
};

ListNode *createByArray(int sz[],int n){
    ListNode *head;
    head=new ListNode;
    head->next=NULL;
    ListNode *r;
    r=head;
    ListNode *p;
    for(int i=0;i<n;i++){
        p=new ListNode;
        p->num=sz[i];
        r->next=p;
        r=p;
    }
    r->next=NULL;
    return head;
}

void  displayLink(ListNode *head)//直接在A中输出
{
	ListNode *p;
    p=head->next;
	cout<<"head-->";
	while(p!= NULL)
	{
		cout<<p->num<<"-->";
		p=p->next;
	}
	cout<<"tail\n";
}

void fun(ListNode *la, ListNode *lb)
{
    ListNode *p1,*p2;
    ListNode *pre;
    p1=la->next; p2=lb->next;
    pre=la;
    while(p1!=NULL)
    {
        p2=lb->next;
        while(p2!=NULL)
        {
            if(p1->num==p2->num)
                break;
            p2=p2->next;
        }
        if(p2!=NULL)
        {
            pre=pre->next;
            p1=p1->next;
        }
        else{
            pre->next=p1->next;
            delete p1;
            p1=pre->next;
        }
    }
}

int main() {
    int n1,n2;
    cin>>n1;
    int a[n1];
    for(int i=0;i<n1;i++){
        cin>>a[i];
    }
    cin>>n2;
    int b[n2];
    for(int i=0;i<n2;i++){
        cin>>b[i];
    }

    ListNode *A,*B;
    A=createByArray(a,n1);
    B=createByArray(b,n2);
    fun(A,B);
    displayLink(A);
	return 0;
}