leetcode刷题-相交链表

编写一个程序,找到两个单链表相交的起始节点。

leetcode刷题-相交链表_第1张图片

 思路:考虑两个简单无环的单向链表,若相交,从某个节点之后节点相同,一定呈Y字形相交。

方法一、借用栈结构,使用两个栈,分别加入A, B链表的节点;然后对栈遍历,若栈顶元素相等,则一定相交,记录此时的节点,元素出栈,一直到栈顶元素不相等,此时之前记录的节点就是相交节点。

代码:

/**
 * 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) {
        ListNode *pa = headA, *pb=headB;
        stack  sa;  //链表 A 入栈
        while (pa)
        {
            sa.push(pa);
            pa = pa->next;
        }
        
        stack  sb;    //链表 B 入栈
        while (pb)
        {
            sb.push(pb);
            pb = pb->next;
        }

        ListNode * intersect = NULL, ta, tb;
        while (!sa.empty()&&!sb.empty()) 比较栈顶元素找交点
        {
            if (sa.top()==sb.top()) //栈顶元素相等
            {
                intersect = sa.top();
                sa.pop();
                sb.pop();
            }
            else
                break;
        }
        return   intersect;
    }
};

方法二、若对空间复杂度做出限制,我们就不能使用栈了。

 

题目:https://leetcode-cn.com/problems/intersection-of-two-linked-lists/

你可能感兴趣的:(c++,#链表)