BZOJ 1142: [POI2009]Tab

1142: [POI2009]Tab

Time Limit: 40 Sec  Memory Limit: 162 MB
Submit: 213  Solved: 80
[Submit][Status][Discuss]

Description

　　2个n*m矩阵，保证同一个矩阵中元素两两不同。问能否通过若干次交换两行或交换两列把第一个矩阵变成第二个。

Input

　　第一行正整数 T (1≤T≤10) 表示数据组数. 每组数据包括：第一行n m （1≤n,m≤1000） 2个n行m列的整数矩阵，元素绝对值均在10^6以内

Output

　　每组数据输出“TAK”/“NIE”表示能/不能.

Sample Input

2
4 3
1 2 3
4 5 6
7 8 9
10 11 12
11 10 12
8 7 9
5 4 6
2 1 3
2 2
1 2
3 4
5 6
7 8

Sample Output

TAK
NIE

HINT

 

Source

鸣谢WangTianYi制作数据

[Submit][Status][Discuss]

分析

对一个矩阵交换两行的时候，显然行内元素没有发生变化；而交换两列的时候，行内元素也只是变换了顺序而已。所以得出——不论对矩阵进行什么样的变换，原本在一行内的元素现在还在一行，原本在一列的元素现在还在一列。而对于两个矩阵，如果它们每行的元素相同，定能通过若干次对列的交换使得其行内元素顺序也相同；显然列也具有相同的性质。

由此得出，我们只需要分析两个矩阵的行列是否满足元素相同即可。当然，这个问题的做法不一，或排序，或哈希。因为题目满足元素大小在-1000000到1000000之间，且一个矩阵内不存在相同元素，所以不妨直接用数组记录每个元素在A矩阵中出现的位置。假如一个元素在A矩阵的(a,b)位置出现，在B矩阵的(c,d)位置出现，我们就认为A的a行和B的c行是匹配的，A的b列和B的d列是匹配的。如果出现了一行匹配两行，就是非法的。这样就能做到稳定的O(N*M + 1000000)，显然可以过掉了。另外，最好加上读入优化，如果想上榜的话。

代码

#include <bits/stdc++.h>

#define N 1005
#define M 1000000
#define K 2000005

int n, m;
int a[N][N];
int b[N][N];
int posX[K];
int posY[K];
int matchX[N];
int matchY[N];

signed main(void)
{
    int cas; scanf("%d", &cas);
    
    while (cas--)
    {
        scanf("%d%d", &n, &m);
        
        for (int i = 1; i <= n; ++i)
            for (int j = 1; j <= m; ++j)
                scanf("%d", &a[i][j]), a[i][j] += M;
        
        for (int i = 1; i <= n; ++i)
            for (int j = 1; j <= m; ++j)
                scanf("%d", &b[i][j]), b[i][j] += M;
                
        memset(posX, 0, sizeof(posX));
        memset(posY, 0, sizeof(posY));
        
        for (int i = 1; i <= n; ++i)
            for (int j = 1; j <= m; ++j)
                posX[a[i][j]] = i, posY[a[i][j]] = j;
                
        bool answer = true;
        
        memset(matchX, 0, sizeof(matchX));
        memset(matchY, 0, sizeof(matchY));
                
        for (int i = 1; i <= n; ++i)
            for (int j = 1; j <= m; ++j) {
                const int v = b[i][j];
                
                if (!posX[v])
                    { answer = false; break; }
                else if (matchX[i] && matchX[i] != posX[v])
                    { answer = false; break; }
                else matchX[i] = posX[v];
                
                if (!posY[v])
                    { answer = false; break; }
                else if (matchY[j] && matchY[j] != posY[v])
                    { answer = false; break; }
                else matchY[j] = posY[v];
            }
            
        puts(answer ? "TAK" : "NIE");
    }
}

@Author: YouSiki