两个链表有一个交点,如何在时间复杂度 O(n) 和 空间复杂度 O(1) 的条件下实现?_字节跳动面试题
阅读原文时间:2023年07月08日阅读:2

输入两个链表,找出它们的第一个公共结点

我们可以首先遍历两个链表得到它们的长度,就能知道哪个链表比较长,

我们可以首先遍历两个链表得到它们的长度,就能知道哪个链表比较长,以及长的链表比短的链表多几个结点。在第二次遍历的时候,在较长的链表上先走若干步,接着再同时在两个链表上遍历,找到的第一个相同的结点就是它们的第一个公共结点。

经过分析我们发现两个有公共结点而部分重合的链表,拓扑形状看起来像一个Y,而不可能像X,如下图所示,两个链表在值为6的结点处交汇:

比如在上图的两个链表中,我们可以先遍历一次得到它们的长度分别为5和4,也就是较长的链表与较短的链表相比多一个结点。第二次先在长的链表上走1步,到达结点2。接下来分别从结点2和结点4出发同时遍历两个结点,直到找到它们第一个相同的结点6,这就是我们想要的结果。

1 /*
2 struct ListNode{
3 int val;
4 struct ListNode *next;
5 ListNode(int x):val(x),next(NULL){}
6 };
7 */
8 List *FindFirstCommonNode(ListNode *pHead1, ListNode *pHead2) {
9 int findListLength(ListNode *pHead) {
10 if (pHead == NULL)return 0;
11 int sum = 1;
12 pHead = pHead->next;
13 while (pHead != NULL) {
14 sum++;
15 pHead = pHead->next;
16 }
17 return sum;
18 }
19 ListNode *walkStep(ListNode *pHead, int step) {
20 while (step--) {
21 pHead = pHead->next;
22 }
23 return pHead;
24 }
25 int FindListLength(ListNode *pHead1, ListNode *pHead2) {
26 int len1 = findListLenth(pHead1);
27 int len2 = findListLenth(pHead2);
28 if (len1 > len2) {
29 pHead1=walkStep(pHead1,len1-len2);
30 }else{
31 pHead2=walkStep(pHead2,len2-len1);
32 }
33 while(pHead1){
34 if(pHead1==pHead2)return pHead1;
35 pHead1=pHead1->next;
36 pHead2=pHead2->next;
37 }
38 return NULL;
39 }
40 }