P4924 [1007] 魔法少女小Scarlet

链接：P4924 [1007] 魔法少女小Scarlet - 洛谷

题目描述

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

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

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

输入格式

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

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

输出格式

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

输入输出样例

输入 #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≤n,m≤500，满足 1≤x−r≤x+r≤n,1≤y−r≤y+r≤n。

C++代码如下：

#include<bits/stdc++.h>
using namespace std;

void fun(int x,int y,int r,int z,int a[505][505]){
    int b[505][505];
    for(int i=x-r;i<=x+r;i++){
        for(int j=y-r;j<=y+r;j++){
            if(z==0){
                b[i][j]=a[x+y-j][y-x+i];
            }else{
                b[i][j]=a[x-y+j][x+y-i];
            }
        }
    }
    for(int i=x-r;i<=x+r;i++){
        for(int j=y-r;j<=y+r;j++){
            a[i][j]=b[i][j];
        }
    }
}

int main(){
    int n,m;
    cin>>n>>m;
    int x,y,r,z;
    int a[505][505];
    int mark=1;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            a[i][j]=mark;
            mark++;
        }
    }
    for(int i=0;i<m;i++){
        cin>>x>>y>>r>>z;
        fun(x,y,r,z,a);
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            cout<<a[i][j]<<" ";
        }
        cout<<endl;
    }
    return 0;
}