python选手使用pypy3提交

题目内容

给小塔一个$n×n$的方阵，第$i$行第$j$列的元素是$r,e,d$中的一个。

定义一个矩阵权值为这个矩阵出现的$r,e,d$数量的最小值。

现在有一个值$myval$，现在想请小塔计算出，有多少个子方阵的矩阵权值不小于$myval$。

思路:前缀和+滑动窗口

1.正方形本质上就是两个在同一个对角线上的点构成的矩阵。所以先把下标根据对角线归类成不同序列,例如一个3 × 3的矩阵:

red

red

red