前言
我们社区陆续会将顾毅(Netflix 增长黑客,《iOS 面试之道》作者,ACE 职业健身教练。微博:@故胤道长[1])的 Swift 算法题题解整理为文字版以方便大家学习与阅读。
LeetCode 算法到目前我们已经更新了 24 期,我们会保持更新时间和进度(周一、周三、周五早上 9:00 发布),每期的内容不多,我们希望大家可以在上班路上阅读,长久积累会有很大提升。
不积跬步,无以至千里;不积小流,无以成江海,Swift社区 伴你前行。如果大家有建议和意见欢迎在文末留言,我们会尽力满足大家的需求。
难度水平:困难
1. 描述
给你一个链表,每 k
个节点一组进行翻转,请你返回翻转后的链表。
k
是一个正整数,它的值小于或等于链表的长度。
如果节点总数不是 k
的整数倍,那么请将最后剩余的节点保持原有顺序。
进阶:
- 你可以设计一个只使用常数额外空间的算法来解决此问题吗?
- 你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。
2. 示例
示例 1
代码语言:javascript复制输入:head = [1,2,3,4,5], k = 2
输出:[2,1,4,3,5]
示例 2
代码语言:javascript复制输入:head = [1,2,3,4,5], k = 3
输出:[3,2,1,4,5]
示例 3
代码语言:javascript复制输入:head = [1,2,3,4,5], k = 1
输出:[1,2,3,4,5]
示例 4
代码语言:javascript复制输入:head = [1], k = 1
输出:[1]
约束条件:
- 列表中节点的数量在范围
sz
内 1 <= sz <= 5000
0 <= Node.val <= 1000
1 <= k <= sz
3. 答案
代码语言:javascript复制/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class ReverseNodesInKGroup {
func reverseKGroup(_ head: ListNode?, _ k: Int) -> ListNode? {
let dummy = ListNode(0)
dummy.next = head
var prev = dummy // prev = A
while prev.next != nil { // A->B->C->D->E => A->D->C->B->E
var groupTail : ListNode? = prev
for _ in 1...k {
groupTail = groupTail?.next
}
guard let _ = groupTail else { // groupTail = D
break
}
let nextGroupHead = groupTail!.next // nextGroupHead = E
var last = nextGroupHead // last = E
var current : ListNode? = prev.next // current = B
while current != nil && current !== nextGroupHead {
let next = current!.next // next = C
current!.next = last // B -> E
last = current // last = B
current = next // current = C
}
groupTail = prev.next
prev.next = last
prev = groupTail!
}
return dummy.next
}
}
- 主要思想:在一个循环中反转一组。 从后往前构造一个反向组。
- 时间复杂度: O(n)
- 空间复杂度: O(1)
该算法题解的仓库:LeetCode-Swift[2]
点击前往 LeetCode[3] 练习
特别感谢 Swift社区 编辑部的每一位编辑,感谢大家的辛苦付出,为 Swift社区 提供优质内容,为 Swift 语言的发展贡献自己的力量,排名不分先后:张安宇@微软[4]、戴铭@快手[5]、展菲@ESP[6]、倪瑶@Trip.com[7]、杜鑫瑶@新浪[8]、韦弦@Gwell[9]、张浩@讯飞[10]、张星宇@ByteDance[11]、郭英东@便利蜂[12]
参考资料
[1] @故胤道长: https://m.weibo.cn/u/1827884772
[2] LeetCode-Swift: https://github.com/soapyigu/LeetCode-Swift
[3] LeetCode: https://leetcode.com/problems/reverse-nodes-in-k-group/
[4] 张安宇: https://blog.csdn.net/mobanchengshuang
[5] 戴铭: https://ming1016.github.io
[6] 展菲: https://github.com/fanbaoying
[7] 倪瑶: https://github.com/niyaoyao
[8] 杜鑫瑶: https://weibo.com/u/3878455011
[9] 韦弦: https://www.jianshu.com/u/855d6ea2b3d1
[10] 张浩: https://github.com/zhanghao19920218
[11] 张星宇: https://github.com/bestswifter
[12] 郭英东: https://github.com/EmingK