Leetcode 160. 相交鏈表

1、問題分析

題目鏈接：https://leetcode-cn.com/problems/intersection-of-two-linked-lists/
  不服就幹，直接暴力破解，廢話不BB。代碼我已經進行了詳細的註釋，理解應該沒有問題，讀者可以作爲參考，如果看不懂(可以多看幾遍)，歡迎留言哦！我看到會解答一下。

2、問題解決

  筆者以C++方式解決。

#include "iostream"

using namespace std;

#include "algorithm"
#include "vector"
#include "queue"
#include "set"
#include "map"
#include "string"
#include "stack"

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */

struct ListNode {
    int val;
    ListNode *next;

    ListNode(int x) : val(x), next(NULL) {}
};

class Solution {
private:
    // 存儲 headA 鏈表的所有節點值
    vector<ListNode *> dp;
    // 存儲 headB 鏈表的所有節點值
    vector<ListNode *> dp2;
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        // 如果有一個鏈表爲空，則肯定沒有相交節點
        if (headA == NULL || headB == NULL) {
            return NULL;
        }

        // 將鏈表 headA 的所有節點存儲到 dp 數組中
        while (headA != NULL) {
            dp.push_back(headA);
            headA = headA->next;
        }

        // 將鏈表 headB 的所有節點存儲到 dp2 數組中
        while (headB != NULL) {
            dp2.push_back(headB);
            headB = headB->next;
        }

        // 從數組的末尾開始比較
        int i = dp.size() - 1;
        int j = dp2.size() - 1;

        // 如果最後一個節點不相等，直接返回 NULL
        if (dp[i] != dp2[j]) {
            return NULL;
        }

        // 如果數組的末尾值相等，在向左移動，不斷遍歷
        i--;
        j--;

        // 節點值相等，不斷向左移動，不斷遍歷
        while (i >= 0 && j >= 0 && dp[i] == dp2[j]) {
            i--;
            j--;
        }

        // 返回相交節點
        return dp[i + 1];
    }
};

int main() {
    ListNode *pNode14 = new ListNode(4);
    ListNode *pNode11 = new ListNode(1);
    ListNode *pNode18 = new ListNode(8);
    ListNode *pNode_14 = new ListNode(4);
    ListNode *pNode15 = new ListNode(5);

    pNode14->next = pNode11;
    pNode11->next = pNode18;
    pNode18->next = pNode_14;
    pNode_14->next = pNode15;


    ListNode *pNode25 = new ListNode(5);
    ListNode *pNode20 = new ListNode(0);
    ListNode *pNode21 = new ListNode(1);
//    ListNode *pNode28 = new ListNode(8);
//    ListNode *pNode24 = new ListNode(4);
//    ListNode *pNode_25 = new ListNode(5);
    pNode25->next = pNode20;
    pNode20->next = pNode21;
    pNode21->next = pNode18;
//    pNode28->next = pNode24;
//    pNode24->next = pNode_25;


    Solution *pSolution = new Solution;
    ListNode *pNode = pSolution->getIntersectionNode(pNode14, pNode25);
    cout << pNode->val << endl;
    system("pause");
    return 0;
}

運行結果

有點菜，有時間再優化一下。

3、總結

  難得有時間刷一波LeetCode, 這次做一個系統的記錄，等以後複習的時候可以有章可循，同時也期待各位讀者給出的建議。算法真的是一個照妖鏡，原來感覺自己也還行吧，但是算法分分鐘教你做人。前人栽樹，後人乘涼。在學習算法的過程中，看了前輩的成果，受益匪淺。
感謝各位前輩的辛勤付出，讓我們少走了很多的彎路！
哪怕只有一個人從我的博客受益，我也知足了。
點個贊再走唄！歡迎留言哦！