P4924 [1007]魔法少女小Scarlet

// Problem: P4924 [1007]魔法少女小Scarlet
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P4924
// Memory Limit: 125 MB
// Time Limit: 1000 ms
// User: Pannnn
 
#include <bits/stdc++.h>
 
using namespace std;
 
void move(int (*dst)[510], int (*src)[510], int x, int y, int r) {
    for (int i = -r; i <= r; ++i) {
        for (int j = -r; j <= r; ++j) {
            dst[x + i][y + j] = src[x + i][y + j];
        }
    }
}
 
void rotateL(int (*info)[510], int x, int y, int r) {
    int tmp[510][510] = { 0 };
    for (int i = -r; i <= r; ++i) {
        for (int j = -r; j <= r; ++j) {
            tmp[x + i][y + j] = info[x + j][y - i];
        }
    }
    move(info, tmp, x, y, r);
}
 
void rotateR(int (*info)[510], int x, int y, int r) {
    int tmp[510][510] = { 0 };
    for (int i = -r; i <= r; ++i) {
        for (int j = -r; j <= r; ++j) {
            tmp[x + i][y + j] = info[x - j][y + i];
        }
    }
    move(info, tmp, x, y, r);
}
 
void init(int (*info)[510], int n) {
    int idx = 0;
    for (int i = 1; i <= n; ++i) {
        for (int j = 1; j <= n; ++j) {
            info[i][j] = ++idx;
        }
    }
}
 
void print(int (*info)[510], int n) {
    for (int i = 1; i <= n; ++i) {
        for (int j = 1; j <= n; ++j) {
            cout << info[i][j] << " ";
        }
        cout << endl;
    }
}
 
// int info[510][510] = { 0 };
 
/*
    开辟临时数组，将旋转的结果存值临时空间中，再复制回来
*/
int main() {
    int n, m;
    cin >> n >> m;
    // 这里开局部数组可能会爆栈，如果爆栈用全局数组
    int info[510][510] = { 0 };
    init(info, n);
    
    int x, y, r, z;
    for (int i = 0; i < m; ++i) {
        cin >> x >> y >> r >> z;
        if (z == 0) {
            rotateR(info, x, y, r);
        } else if (z == 1) {
            rotateL(info, x, y, r);
        }
    }
    print(info, n);
    return 0;
}