乐团站位---Leetcode练习

这道题理解了就不难

题目描述

某乐团的演出场地可视作 num * num 的二维矩阵 grid（左上角坐标为 [0,0])，每个位置站有一位成员。乐团共有 9 种乐器，乐器编号为 1~9，每位成员持有 1 个乐器。

为保证声乐混合效果，成员站位规则为：自 grid 左上角开始顺时针螺旋形向内循环以 1，2，…，9 循环重复排列。例如当 num = 5 时，站位如图所示

请返回位于场地坐标 [Xpos,Ypos] 的成员所持乐器编号。

思路

这道题卡人的地方就是时间复杂度，当矩形的边长比较小时我们甚至可以模拟建立一个二维数组来找到我们需要的位置。

但是当nums很大时，就会超时了，这个方法pass

数字排列的规律很好找到，可以看成一个一维数组一圈一圈向内卷成的二维数组，而目标坐标在一维数组中的位置取9的模就是答案。

1、找出目标点在第几圈，n=min(xPos,yPos,num-xPos-1,num-yPos-1)

2、找出目标点的位置，上侧 or 下侧 or 左侧 or 右侧

if n==xPos:
   kinds+=yPos-n+1
elif n==yPos:
    kinds+=(num-2*n)*3-3+num-n-xPos
elif n==num-xPos-1:
    kinds+=(num-2*n)*2-2+num-n-yPos
elif n==num-yPos-1:
    kinds+=(num-2*n)*1-1+xPos-n+1

代码

class Solution:
    def orchestraLayout(self, num: int, xPos: int, yPos: int) -> int:
        n=min(xPos,yPos,num-xPos-1,num-yPos-1)
        kinds=0
        kinds+=4*n*num-4*n-4*n*(n-1)
        if n==xPos:
            kinds+=yPos-n+1
        elif n==yPos:
            kinds+=(num-2*n)*3-3+num-n-xPos
        elif n==num-xPos-1:
            kinds+=(num-2*n)*2-2+num-n-yPos
        elif n==num-yPos-1:
            kinds+=(num-2*n)*1-1+xPos-n+1

        if kinds%9==0:
            return 9
        else:
            return kinds%9