洛谷 魔法少女 python

因为最近一直很忙,包括我自己本来刷题是顺序的刷的,遇到这道题思考了很久,首先我一直知道Numpy和C++可以很快写出来,本来一开始我不想用Numpy,但是我差不多思考了一周,用C++的方法很容易超时,而我自己想的是原地翻转(具体可以参考我的 leetCode 旋转图像 这个文章) 有思路可以处理评论一下,但是很遗憾 我写不出来 我只能用numpy了 抱歉

import numpy as np

n, m = map(int, input().split())  
t1 = np.arange(1, n * n + 1).reshape(n, n)

for i in range(m):
    x, y, r, z = map(int, input().split())
    a, b = x - 1, y - 1
    if z ==1:  # 左转
        t1[a-r:a+r+1,b-r:b+r+1] = np.rot90(np.copy(t1[a-r:a+r+1,b-r:b+r+1]))
    else:  # 右转 
        t1[a - r:a + r + 1, b - r:b + r + 1] = np.rot90(np.copy(t1[a - r:a + r + 1, b - r:b + r + 1]),axes=(1,0))

for i in range(len(t1)):
    for j in range(len(t1)):
        print(t1[i][j],end=" ")
    print()

你可能感兴趣的:(算法,python)