P4924 [1007]魔法少女小Scarlet

#include<bits/stdc++.h>
using namespace std;
int square[505][505],temp[505][505];
void rotation(int x,int y,int r) //顺时针旋转90°
{
    for(int i=x-r;i<=x+r;i++)
    {
        for(int j=y-r;j<=y+r;j++)
        {
            temp[i][j]=square[i][j];
        }
    }
    int x1=x+r,y1=y-r;
    for(int i=x-r;i<=x+r;i++)
    {
        for(int j=y-r;j<=y+r;j++)
        {
            square[i][j]=temp[x1][y1];
            x1--;
        }
        x1=x+r;
        y1++;
    }
}
void rotation_r(int x,int y,int r) //逆时针旋转90°
{
    for(int i=x-r;i<=x+r;i++)
    {
        for(int j=y-r;j<=y+r;j++)
        {
            temp[i][j]=square[i][j];
        }
    }
    int x1=x-r,y1=y+r;
    for(int i=x-r;i<=x+r;i++)
    {
        for(int j=y-r;j<=y+r;j++)
        {
            square[i][j]=temp[x1][y1];
            x1++;
        }
        x1=x-r;
        y1--;
    }
}
int main()
{
    int n,m,t=1;
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            square[i][j]=t++;
        }
    }
    int x,y,r,z;
    for(int i=0;i<m;i++){
        cin>>x>>y>>r>>z;
        if(z==0)
            rotation(x,y,r);
        else if(z==1)
            rotation_r(x,y,r);
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            cout<<square[i][j]<<" ";
        }
        cout<<endl;
    }
    return 0;
}