O(1)计算螺旋矩阵对应位置的元素

螺旋矩阵

螺旋矩阵是指一个呈螺旋状的矩阵，它的数字由右上角开始向下依次填充，到边界之后继续向左边填充吗，然后依次是上边和右边。如此循环。图例如下：

我们规定左下角的坐标是$(1,1)$，右上角的坐标是$(n,n)$，$(x,y)$代表位于从左往后数第x列，从下往上数第y行的元素。

计算方法

螺旋矩阵具有很强的规律性，因为元素是一圈一圈填充进去的，所以，我们可以先计算出来所求位置元素位于第几圈(从外向内数，第一圈的边长是$n$，第二圈的边长是$n-2$，以此类推…)上(假设是$T$)，然后先计算出前$T-1$圈的总的数字数量，然后再计算出第T圈中小于等于所求位置数字的数字数量，就可以得到对应位置的值了。具体情况如下：
首先第$T$圈的边长为$n-2(T-1)=n+2T+2$，当前点$(x,y)$位于从下往上数第$y$行，从上往下数第$n-y+1$行，从左往右数第$x$列，从右往左数第$n-x+1$列。每一圈的数字可以看作由四个长度为$当前边长-1$的数字段组成。那么，我们可得$(x,y)$位置的数字的值(即$val(x,y)$)为：
首先，设该元素所在的圈为第$T$圈，可得$T=min(n-x+1,n-y+1,x,y)$。那么，前$m$圈的数字个数为：
$$4\ast \frac{m\ast ((n-1)+(n-2\ast (m-1)-1))}{2}=4\ast m\ast (n-m)$$
(每一圈的数字可以看作由四个长度为$当前边长-1$的数字段组成，然后等差数列求和计算即可)
则前$T-1$圈的数字个数为$ans=4\ast (T-1)\ast (n-T+1)$(代入$m=T-1$即可)。
$$val(x,y)=ans+\{\begin{array}{ll}(n-y+1)-(T-1)& x=n-T+1\\ 2\ast (n-2\ast T+1)+(y-(T-1))& x=T\\ 3\ast (n-2\ast T+1)+(x-(T-1))& y=n-T+1\\ (n-2\ast T+1)+(n-x+1-(T-1))& y=T\end{array}$$
即(上述算式含义为4段中的某几整段的数字个数+剩下的数字个数)
$$val(x,y)=ans+\{\begin{array}{ll}n-y-T+2& x=n-T+1\\ 2\ast n+y-5\ast T+3& x=T\\ 3\ast n+x-7\ast T+4& y=n-T+1\\ 2\ast n-x-3\ast T+3& y=T\end{array}$$
按照上述算式进行计算即可得到对应位置的值。

今年南京网络赛的A题就用到了这个东西
我的博客链接