leetcode: Intersection of Two Linked Lists

很久没做leetcode,更新了些新题做一下。

这题要求我们找到两个链表的公共部分,要求时间代价为O(n),空间代价为O(1)。

其实做法有很多种,比如先扫描一个链表,将其每个节点hash保存,再遍历第二个链表,判断每个节点是否在第一个链表中有出现。但这样空间复杂度为O(n),不符合要求。

我的做法如下:

显然,若两个链表存在公共节点,那么自公共节点之后两个链表的节点都是一样的;链表长度可能不一样,若我们将链表都限制在同样长度,那么从两端同时向next跳,当两端节点一样时即找到了公共节点,这个过程是O(n)的。限制链表长度可以实现便利两个链表获取各自的长度,将较长的那个向next跳 |lenA-lenB| 次以保证两个链表长度一致。这个过程也是O(n)的。

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) {
 *         val = x;
 *         next = null;
 *     }
 * }
 */
public class Solution {
    public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
        int lenA=0,lenB=0;
        ListNode stA = headA,stB = headB;
        while(stA!=null)
        {
            lenA++;
            stA = stA.next;
        }
        while(stB!=null)
        {
            lenB++;
            stB = stB.next;
        }
        if(lenA>lenB)
        {
            for(int i=1;i<=lenA-lenB;i++)
            {
                headA = headA.next;
            }
        }
        else if(lenB>lenA)
        {
            for(int i=1;i<=lenB-lenA;i++)
            {
                headB = headB.next;
            }
        }
        while(headA!=null&&headB!=null)
        {
            if(headA==headB)
            {
                return headA;
            }
            headA = headA.next;
            headB = headB.next;
        }
        return null;
    }
}


看了下网上的解答,发现了更好的做法。虽然我的解法也符合要求,但实际上要遍历两遍链表。我们假设链表的公共部分为C,第一个链表自己的部分为A,第二个链表自己的部分为B,那么第一个链表为A+C,第二个链表为B+C。

由于两个链表长度不一,我们不妨扩展一下将其串起来(链表一后接一个链表二,链表二后接一个链表一)

此时链表一为A+C+B+C,链表二为B+C+A+C;

我们发现此时,链表一和链表二前三组都为A+C+B(不重复),那么同时开始遍历完这么多个节点之后,链表一和链表二将同时访问C,即为所求的公共节点起始;返回即可

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值