【Datawhale组队打卡】【Day7】第54、59、61题

第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