欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

#leetcode刷题之路25- k个一组翻转链表

程序员文章站 2022-05-14 10:49:12
给出一个链表,每 k 个节点一组进行翻转,并返回翻转后的链表。k 是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍,那么将最后剩余节点保持原有顺序。 示例 :给定这个链表:1->2->3->4->5当 k = 2 时,应当返回: 2->1->4->3->5当 k = 3 时, ......

给出一个链表,每 k 个节点一组进行翻转,并返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍,那么将最后剩余节点保持原有顺序。

示例 :
给定这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5

说明 :
你的算法只能使用常数的额外空间。
你不能只是单纯的改变节点内部的值,而是需要实际的进行节点交换。

 

思路:先把是不是从头开始区分一下,然后每一次翻转之前,都要判断数量够不够k

#include <iostream>
using namespace std;
struct listnode {
    int val;
    listnode *next;
    listnode(int x) : val(x), next(null) {}
};

listnode* createlist(int n)//生成链表
{
    if (n == 0) return nullptr;
    listnode* head = (listnode*)malloc(sizeof(listnode));
    cin >> head->val;
    listnode* pre = head;
    for (int i = 0; i < n - 1; i++)
    {
        listnode* p = (listnode*)malloc(sizeof(listnode));
        cin >> p->val;
        pre->next = p;
        pre = pre->next;
    }
    pre->next = nullptr;
    return head;
}

listnode* reverselist(listnode* head, listnode *p, int k)
{
    int tk = k;//暂时保存k的值
    if (head == p)//是头
    {
        listnode* temp = head;//
        listnode* endd = head;//
        listnode* ttemp=nullptr;
        listnode* pre = nullptr;//保存已反转的下一个
        listnode* end = nullptr;//
        listnode* t = head;//保存反转过的最后一个
        while (tk > 0)//说明从头节点开始有k个节点可供反转
        {
            if (temp != nullptr)
            {
                temp = temp->next;
                tk--;
            }
            else
                return head;
        }//while循环后temp指针指向链表的第k+1个节点
        //cout<<temp->val<<endl;
        //开始反转,让头指向空
        pre = head->next;
        head->next = nullptr;
        //cout<<temp->val<<endl;
        while (pre != temp)
        {
            //t = pre;
            end = pre->next;
            pre->next = head;
            head = pre;
            pre = end;
        }
        //cout << t->val << endl;
        endd->next = temp;//连接后面剩下的
        //带头的前k个处理完了,判断接下来的够不够k个
        tk = k;
        //temp = t;
        while (tk != 0)//是否可继续处理
        {
            if (temp != nullptr)
            {
                temp = temp->next;
                tk--;
            }
            else
                return head;
        }
        //cout << t->val <<t->next->val<< endl;
        //cout << head->val << head->next->val << endl;
        reverselist(head, t, k);//够的话就递归反转
    }
    else
    {
        listnode* pre = p;//保存已经反转过的最后一个--------------------
        listnode*cur = pre->next;//保存待反转的第一个
        listnode* cur_next=cur->next ;//保存待反转的下一个
        listnode* end = nullptr;//存放待反转的最后一个节点的下一个节点----------------------
        listnode* thead=nullptr;//存放转后的头节点---------------
        listnode* tcur = cur_next->next;//存放当前的cur_next的下一个节点,即下次要反转的那个------------------
        listnode* temp = nullptr;//记录反转的链表尾
        listnode* t = nullptr;//
        //-----------------------------先反转第一次

        cur_next->next = cur;
        cur->next = nullptr;
        temp = cur;
        thead = cur_next;
        tk = k-2;
        while (tk > 0)
        {
            cur = tcur;
            tcur = tcur->next;
            cur->next = thead;
            thead = cur;
            tk--;
        }
        pre->next = thead;
        temp->next = tcur;
        t = temp->next;
        //带头的前k个处理完了,判断接下来的够不够k个
        tk = k;
        //temp = t;
        while (tk != 0)//是否可继续处理
        {
            if (t != nullptr)
            {
                t = t->next;
                tk--;
            }
            else
                return head;
        }
        reverselist(head, temp, k);//够的话就递归反转
    }
    return head;
}

listnode* reversekgroup(listnode* head, int k) {
    if (head == nullptr || k == 0 || k == 1) return head;
    listnode *temp = head;
    head = reverselist(head, head, k);
    return  head;
}

int main() {
    listnode* head = createlist(4);
    listnode*ans = reversekgroup(head, 2);
    while (ans != nullptr)
    {
        cout << ans->val << endl;
        ans = ans->next;
    }
    return 0;
}