算法练习(11):Reverse Nodes in k-Group

来源:互联网 发布:淘宝怎么修改手机号 编辑:程序博客网 时间:2024/06/01 20:56

题意:题目意思是说给出一个链表,同时输入一个比链表长度小的正整数k,然后我们要做的是对于这个链表,从左到右,每k个节点作为一个整体逆转,如上面的例子所示,但是有个规定,不能改变节点本身的值,我们要做的是通过改变节点间的链接方式实现。

分析与思路:一开始肯定想过投机取巧通过改变节点的值来轻松实现,但是这题明确规定了不可以那么做,那么我们就只有老老实实改变节点连接方式实现了。我的想法是,既然要逆转k个,但是不足k个又不用逆转,而且第二种情况只可能出现在最后。我是先判断,后实现,先判断从当前节点以下走能不能取出一段长为k的链,若能,则以这整段链逆转,若不能,则直接结束。好了,整体思路有了,现在进一步考虑,我们需要什么指针来标记好这个段?既然是一个链,不像数组那样有下标,我们只能通过temp=temp->next的方式遍历,这样就有问题了,我们即使找到了满足条件的链段,可是我们已经扫到了段尾,所以这里我们就需要一个start指针指示链段的首节点,同样也需要一个end指针指示链段的尾节点,同时就算逆转了我们也要把链段装回整个大链吧?所以我们又需要一个整个链段的父指针(代码中的Fa),然后在链段内再用一些临时指针实现逆转就可以了。

代码:

class Solution {public:ListNode* reverseKGroup(ListNode* head, int k) {if (head == NULL || head->next == NULL || k == 1) return head;//排除特殊情况ListNode * temp = head;//当前扫描的节点ListNode * fa = head;//当前节点的父ListNode * Fa = head;//整条链的父ListNode * start = head;//链段首节点ListNode * end = head;//链段的尾bool isHead = 1;//是否为第一条链段while(1) {int times = k;start = temp;while (times--) {//判断能否成一条满足条件的链段fa = temp;temp = temp->next;if (temp == NULL) break;}//这时候temp已经到达链段尾部的后一个end = fa;if (times == -1||(temp==NULL&× == 0)) {//说明已经取的满足条件的链段fa = start;//因为start是逆转后的链段的尾ListNode* pre = start;ListNode* prenext = start->next;ListNode* preFa = temp;while (1) {//逆转过程pre->next = preFa;preFa = pre;pre = prenext;prenext = prenext->next;pre->next = preFa;if (pre == end) break;}if (isHead) {isHead = 0;head = end;}else Fa->next = pre;//把逆转后的链段接回去Fa = start;if (temp == NULL) break;}else break;//取不到满足条件的链段说明已经结束}return head;}};


原创粉丝点击