第54题
题目
给定一个包含 m x n 个元素的矩阵(m 行, n 列),请按照顺时针螺旋顺序,返回矩阵中的所有元素。
输入:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
输出: [1,2,3,6,9,8,7,4,5]
输入:
[
[1, 2, 3, 4],
[5, 6, 7, 8],
[9,10,11,12]
]
输出: [1,2,3,4,8,12,11,10,9,5,6,7]
思路
class Solution:
def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
result = []
while len(matrix) > 1:
result.extend(matrix[0])
matrix = self.trans(matrix[1:])
result.extend(matrix[0])
return result
def trans(self, matrix: [[int]]) -> [[int]]:
# https://blog.csdn.net/qq_30615903/article/details/80932737
return list(zip(*matrix))[::-1]
第59题
题目
给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
输入: 3
输出:
[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]
思路
class Solution:
def generateMatrix(self, n: int) -> List[List[int]]:
res = [[0 for _ in range(n)] for _ in range(n)]
i, j = 0, 0
# 方向数组 右下左上
ds = [(0, 1), (1, 0), (0, -1), (-1, 0)]
d = 0 # 方向指示变量
cur = 1
while cur < n ** 2 + 1:
res[i][j] = cur
x, y = i + ds[d][0], j + ds[d][1] # 下一步坐标
# 需要变向
if x < 0 or x == n or y < 0 or y == n or res[x][y] != 0:
d = (d + 1) % 4
x, y = i + ds[d][0], j + ds[d][1] # 重置下一步坐标
i, j = x, y
cur += 1
return res
第61题
题目
给定一个链表,旋转链表,将链表每个节点向右移动 k 个位置,其中 k 是非负数。
示例 1:
输入: 1->2->3->4->5->NULL, k = 2
输出: 4->5->1->2->3->NULL
解释:
向右旋转 1 步: 5->1->2->3->4->NULL
向右旋转 2 步: 4->5->1->2->3->NULL
代码
注意k可以大于链表的长度,快慢指针可能会有问题,因此使用循环链表
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def rotateRight(self, head: ListNode, k: int) -> ListNode:
l = head
count = 1
if head == None:return None
while head.next:
count+=1
head = head.next
if count == 1 : return l
head.next = l
t = head
k = count*(k//count + 1) - k
while k:
t = l
l = l.next
k-=1
t.next = None
return l