魔法少女小Scarlet#洛谷

题目描述

Scarlet 最近学会了一个数组魔法，她会在 $n\times n$ 二维数组上将一个奇数阶方阵按照顺时针或者逆时针旋转 $90^{\circ }$。

首先，Scarlet 会把 $1$ 到 $n^2$ 的正整数按照从左往右，从上至下的顺序填入初始的二维数组中，然后她会施放一些简易的魔法。

Scarlet 既不会什么分块特技，也不会什么 Splay 套 Splay，她现在提供给你她的魔法执行顺序，想让你来告诉她魔法按次执行完毕后的二维数组。

输入格式

第一行两个整数 $n,m$，表示方阵大小和魔法施放次数。

接下来 $m$ 行，每行 $4$ 个整数 $x,y,r,z$，表示在这次魔法中，Scarlet 会把以第 $x$ 行第 $y$ 列为中心的 $2r+1$ 阶矩阵按照某种时针方向旋转，其中 $z=0$ 表示顺时针，$z=1$ 表示逆时针。

输出格式

输出 $n$ 行，每行 $n$ 个用空格隔开的数，表示最终所得的矩阵

样例 #1

样例输入 #1

5 4
2 2 1 0
3 3 1 1
4 4 1 0
3 3 2 1

样例输出 #1

5 10 3 18 15
4 19 8 17 20
1 14 23 24 25
6 9 2 7 22
11 12 13 16 21

提示

对于50%的数据，满足 $r=1$

对于100%的数据 $1\le n,m\le 500$，满足 $1\le x-r\le x+r\le n,1\le y-r\le y+r\le n$。

import numpy as np


def work(key,flag,x_value,y_value,size):
    global mapp
    if flag==0:
        key=np.rot90(key,3)
        pass
    else:
        key=np.rot90(key)
        pass
    xx=x_value-size
    yy=y_value-size
    for item in range(2*size+1):
        for jtem in range(2*size+1):
            mapp[item+xx][jtem+yy]=key[item][jtem]


if __name__=="__main__":
    n, m = map(int, input().split())
    mapp = [[jtem for jtem in range(item * n + 1, item * n + n + 1)] for item in range(n + 1)]
    for item in range(m):
        x, y, r, z = map(int, input().split())
        x-=1
        y-=1
        data = [[0] * (2 * r + 1) for _ in range(2 * r + 1)]
        x0 = x - r
        y0 = y - r
        for p in range(2 * r + 1):
            for q in range(2 * r + 1):
                data[p][q] = mapp[x0 + p][y0 + q]
                pass
            pass
        work(data,z,x,y,r)
        pass
    for item in range(n):
        for jtem in range(n):
            if jtem==n-1:
                print(mapp[item][jtem])
            else:
                print(mapp[item][jtem],end=' ')
            pass

超时了，最近在准备考试，以后改吧，大佬也给看看怎么优化下