[LintCode]450.K组翻转链表

最新推荐文章于 2021-05-08 23:04:08 发布

温酒知莲华

最新推荐文章于 2021-05-08 23:04:08 发布

阅读量342

点赞数

分类专栏：数据结构链表文章标签：算法

本文链接：https://blog.csdn.net/zwy1258432405/article/details/77717167

版权

数据结构链表专栏收录该内容

23 篇文章 0 订阅

订阅专栏

给你一个链表以及一个 k ,将这个链表从头指针开始每 k 个翻转一下。

链表元素个数不是k的倍数，最后剩余的不用翻转。

样例

给出链表 1->2->3->4->5

k = 2, 返回 2->1->4->3->5

k = 3, 返回 3->2->1->4->5

思路：从左向右遍历链表，如果栈的大小不等于k，就将结点压入栈中。每当栈的大小达到k时，从栈中依次弹出这些结点，根据弹出的顺序重新联接。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    /**
     * @param head a ListNode
     * @param k an integer
     * @return a ListNode
     */
    ListNode *reverseKGroup(ListNode *head, int k) {
        if(head==NULL||k<2)return head;
        int count=0;
        stack<int> temp;
        ListNode* p1=head;
        ListNode* p2=head;
        while(p1){
            temp.push(p1->val);
            count++;
            p1=p1->next;
            if(count%k==0){
                for(int i=0;i<k;i++){
                    p2->val=temp.top();
                    temp.pop();
                    p2=p2->next;
                }
            }
        }
        return head;
    }
};