leetcode-54 蛇形矩阵推导

刷到关于蛇形矩阵， 不知道通项怎么出来的，

就自己进行了推导，为防后续忘记，记录一下。

1 3 6 10 15

2 5 9 14

4 8 13

7 12

11

先计算每行的通项公式(i代表列，j代表行)：

a2 - a1 = 2

a3 - a2 = 3

a4 - a3 = 4

...

ai - ai-1 = n-1

可以发现右边是一个首项为2，公差为1的等差数列。

所有式子相加求和后，根据等差数列前n项和公式，ai - a1 = (2+i)(i-1)/2

化简后，可以得到第一行通项，ai1 = i(i+1)/2

同理，第二行通项，ai2 = i(i+3)/2

第三行通项，ai3 = i(i+5)/2 + 1

第四行通项，ai4 = i(i+7)/2 + 3

令bj = aij，

b2 - b1 = i

b3 - b2 = i + 1

b4 - b3 = i + 2

...

bj - bj-1 = i + j -2

同样可以发现右边是一个首项为i，公差为1的等差数列。

所有式子相加求和后，根据等差数列前n项和公式， bj - b1 = [i + (i + j - 2)](j - 1)/2,

由于b1 = ai1 =  i(i+1)/2,

最终通项，aij = bj = (2i - 2 + j)(j - 1)/2 + i(i + 1)/2