LeetCode 刷题日记 160. 相交链表

这道题给了两个链表，这两个链表有概率出现相交的部分，如果有的话，说明这是一个相交链表。写一段程序判断该链表是否为相交链表，如果是的话，返回链表中的相交节点，如果不是，返回NULL。代码如下：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        unordered_set<ListNode*> set;
        ListNode* ha = headA;
        ListNode* hb = headB;
        while(ha){
            set.insert(ha);//将链表A中的每一个数塞入set
            ha = ha -> next;
        }
        while(hb){
            if(set.count(hb)==1)return hb;//若set中有链表B的值，则返回该值
            hb = hb -> next;
        }
        return NULL;
    }
};

这道题里，使用了unordered_set<ListNode*> set;来创建一个集合，通过set.insert()函数将链表塞入集合，再用set.count()来判断集合中是否有链表B的值。