【codeforces 1136C】Nastya Is Transposing Matrices（思维+矩阵）

题意：给出一个n×m的原矩阵和一个改变后的矩阵，
问能否通过交换多次原矩阵中的任意k×k矩阵，
使变成新矩阵，如果可以输出YES，不可以输出NO
交换矩阵按照上图中的规则

思路：
交换某个k×k的矩阵后主对角线的元素是不变的，
而次对角线的元素顺序变了，
但是每个数的个数是不会变的
所以题目就变成了
检查原矩阵的后来的矩阵的每个次对角线是不是严格相等的
在这里用map存每条次对角线的元素
一共有n+m-1条次对角线

代码：

#include <iostream>
#include <cstdio>
#include <map>
using namespace std;
#define ll long long
#define memset(a,b) memset(a,b,sizeof(a))

int a[505][505];
int b[505][505];
map <int ,int > num1[1010];
map <int ,int > num2[1010];
int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    for(int i=1; i<=n; i++)
        for(int j=1; j<=m; j++)
            scanf("%d",&a[i][j]);

    for(int i=1; i<=n; i++)
        for(int j=1; j<=m; j++)
            scanf("%d",&b[i][j]);

    int s=n+m-1;
    for(int i=1;i<=s;i++){
        if(i>m-1){
            int k=i-m+1;
            for(int j=k;j<=n;j++){
                num1[i][a[j][i+1-j]]++;
                num2[i][b[j][i+1-j]]++;
            }
        }
        else{
            for(int j=1;j<=i;j++){
                num1[i][a[j][i+1-j]]++;
                num2[i][b[j][i+1-j]]++;
            }
        }
    }


    map <int ,int>::iterator it1,it2;
    for(int i=1;i<=s;i++){
        for(it1 = num1[i].begin(),it2 = num2[i].begin();it1!=num1[i].end(),it2!=num2[i].end();it2++,it1++){
           int flag =0;
           if(it1->first == it2->first && it1->second == it2->second)
              flag =1;
         if(flag == 0)
         {
             printf("NO\n");
             return 0;
         }
    }
}

     printf("YES\n");
    return 0;
}
/*
5 4
6 3 2 11
5 9 4 2
3 3 3 3
4 8 2 2
7 8 6 4
6 3 2 11
5 9 3 8
3 4 3 2
4 2 3 2
7 8 6 4
*/