LeetCode-24. 两两交换链表中的节点(Swap Nodes in Pairs)

用一个指针记录前一个节点(和下一题一起做了)

class Solution {
public:
    ListNode* swapPairs(ListNode* head) {
        int k = 2;
        ListNode *newHead = new ListNode(0);
        ListNode *relativeHead = newHead;
        ListNode *tail = newHead;
        ListNode *p = head;
        ListNode *pre = head;
        int count = 0;
        while(p) {
            if(count == k || count == 0) {
                relativeHead = tail;
                tail = p;
                count = 0;
            }
            pre = p;
            p = p->next;
            pre->next = relativeHead->next;
            relativeHead->next = pre;
            count++;
        }
        if(count != k){
            p = relativeHead->next;
            relativeHead->next = NULL;
            while(p) {
                pre = p;
                p = p->next;
                pre->next = relativeHead->next;
                relativeHead->next = pre;
            }
        }
        return newHead->next;   
    }
};

题目链接:https://leetcode-cn.com/problems/swap-nodes-in-pairs/

发布了42 篇原创文章 · 获赞 2 · 访问量 1413

猜你喜欢

转载自blog.csdn.net/Listen_heart/article/details/103099286