题解 | # 链表中的节点每k个一组翻转 #

链表中的节点每k个一组翻转

http://www.nowcoder.com/practice/b49c3dc907814e9bbfa8437c251b028e

本题关键在于理清思路,避免冗长的边界判断,具体地,为了解耦,专门开个函数用于反转指定区间内的链表节点,并且返回反转之后的头尾节点。在主函数中,只需要k个一组分割节点,调用函数即可,同时使用经典的虚拟头节点辅助。

import . "nc_tools"
/*
 * type ListNode struct{
 *   Val int
 *   Next *ListNode
 * }
 */

/**
  * 
  * @param head ListNode类 
  * @param k int整型 
  * @return ListNode类
*/
func reverseKGroup( head *ListNode ,  k int ) *ListNode {
    // write code here
    dummyHead := &ListNode{}
    pretail := dummyHead
    cur := head
    for  {
        tmphead := cur
        for i := 1;i < k && cur != nil;i++ {
            cur = cur.Next
        }
        if cur == nil {
            pretail.Next = tmphead
            break
        }
        tmptail := cur
        nexthead := tmptail.Next
        reversehead,reversetail := reverseList(tmphead,tmptail)
        pretail.Next = reversehead
        pretail = reversetail
        cur = nexthead
    }
    return dummyHead.Next
}

func reverseList(head *ListNode,tail *ListNode) (newhead *ListNode,newtail *ListNode){

    var pre *ListNode
    cur := head
    for cur != tail {
        tmp := cur.Next
        cur.Next = pre
        pre = cur
        cur = tmp
    }
    cur.Next = pre
    newhead = tail
    newtail = head
    return
}
全部评论

相关推荐

03-16 22:00
武汉大学 C++
幸福的小熊猫想要offer:我阿里投的 c++岗,面试官说自己是做 java 的,c++这辈子才有了
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
04-23 19:42
点赞 评论 收藏
分享
评论
点赞
1
分享

创作者周榜

更多
牛客网
牛客企业服务