环形链表检测、数组交集与随机链表复制实战

代码实现

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        // 去重
        std::set<int> s1(nums1.begin(), nums1.end());
        std::set<int> s2(nums2.begin(), nums2.end());
        vector<int> v;
        
        // 将一个数组中的值依次到另一个数组中查找
        for(auto e : s1) {
            if(s2.count(e)) {
                v.push_back(e); // 在另一个数组中插入到一个容器中
            }
        }
        return v;
    }
};

对比算法优化版：由于 set 遍历是有序的，我们可以利用双指针依次比较，效率更高。

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        std::set<int> s1(nums1.begin(), nums1.end());
        std::set<int> s2(nums2.begin(), nums2.end());
        vector<int> ret;
        auto it1 = s1.begin();
        auto it2 = s2.begin();
        
        // 有序值，依次比较，小的 ++，相等的就是交集
        while(it1 != s1.end() && it2 != s2.end()) {
            if(*it1 < *it2) {
                it1++;
            } else if(*it1 > *it2) {
                it2++;
            } else {
                ret.push_back(*it1);
                it1++;
                it2++;
            }
        }
        return ret;
    }
};

三、随机链表的复制

题目描述

给你一个长度为 n 的链表，每个节点包含一个额外增加的随机指针 random，该指针可以指向链表中的任何节点或空节点。

构造这个链表的深拷贝。深拷贝应该正好由 n 个全新节点组成，其中每个新节点的值都设为其对应的原节点的值。新节点的 next 指针和 random 指针也都应指向复制链表中的新节点，并使原链表和复制链表中的这些指针能够表示相同的链表状态。复制链表中的指针都不应指向原链表中的节点。

解题思路

这个问题难点在于 random 指针的映射关系。如果直接遍历赋值，无法确定 random 指向的新节点地址。

方法一：C 语言穿插法（空间复杂度 O(1)）

利用原链表结构，在每个原节点后插入一个拷贝节点。这样原节点 A 的拷贝就在 A->next。通过 A->random->next 即可找到拷贝节点的 random 目标。最后将新旧链表断开。

/**
 * Definition for a Node.
 * struct Node {
 *     int val;
 *     struct Node *next;
 *     struct Node *random;
 * };
 */
typedef struct Node Node;

// 构造一个新节点
Node* buyNode(int x) {
    Node* newnode = (Node*)malloc(sizeof(Node));
    newnode->val = x;
    newnode->next = newnode->random = NULL;
    return newnode;
}

// 在原链表的基础上拷贝节点（即原链表的每个节点后面拷贝一个一样的节点）
void AddNode(Node* head) {
    Node* pcur = head;
    while(pcur) {
        Node* newnode = buyNode(pcur->val);
        Node* next = pcur->next;
        newnode->next = next;
        pcur->next = newnode;
        pcur = next;
    }
}

// 设置新节点的 random，利用关系式 copy->random = pcur->random->next
void setRandom(Node* head) {
    Node* pcur = head;
    while(pcur) {
        Node* copy = pcur->next;
        if(pcur->random) {
            copy->random = pcur->random->next;
        }
        pcur = copy->next;
    }
}

struct Node* copyRandomList(struct Node* head) {
    if(head == NULL) return head;
    
    // 1. 在原链表基础上拷贝节点并插入到原链表的每个节点之后
    AddNode(head);
    // 2. 设置新节点的 random
    setRandom(head);
    
    // 3. 断开新旧链表
    Node* pcur = head;
    Node* copyHead, *copyTail;
    copyHead = copyTail = pcur->next;
    
    while(copyTail->next) {
        pcur = copyTail->next;
        copyTail->next = pcur->next;
        copyTail = copyTail->next;
    }
    return copyHead;
}

方法二：C++ STL Map 法（空间复杂度 O(N)）

C++ 中可以使用 std::map 建立原节点到拷贝节点的映射关系。遍历时，先创建所有新节点并存入 map，再第二次遍历设置 random 指针。这种方法逻辑清晰，易于维护。

class Solution {
public:
    Node* copyRandomList(Node* head) {
        std::map<Node*, Node*> nodeMap; // 存储原节点到拷贝节点的映射
        Node* copyhead = nullptr, *copytail = nullptr;
        Node* cur = head;
        
        // 第一次遍历：创建新节点并建立映射
        while(cur) {
            if(copytail == nullptr) {
                copyhead = copytail = new Node(cur->val);
            } else {
                copytail->next = new Node(cur->val);
                copytail = copytail->next;
            }
            nodeMap[cur] = copytail;
            cur = cur->next;
        }
        
        // 第二次遍历：处理 random 指针
        cur = head;
        Node* copy = copyhead;
        while(cur) {
            if(cur->random == nullptr) {
                copy->random = nullptr;
            } else {
                // 通过 map 获取原节点 random 对应的拷贝节点
                copy->random = nodeMap[cur->random];
            }
            cur = cur->next;
            copy = copy->next;
        }
        return copyhead;
    }
};

以上三种经典算法题涵盖了链表操作、集合应用及哈希映射技巧。实际开发中，理解底层原理比死记硬背更重要，建议结合调试工具跟踪指针变化，加深印象。