游游的01矩阵 [携程]【思维题】

题目描述：

游游拿到了一个2行2列的01矩阵$a$，她每次操作可以交换$a$的相邻两个元素（同一行或者同一列均为相邻）。游游想知道至少要多少次操作可以使得矩阵$a$变成矩阵$b$？
共有$t$组询问。

输入描述：

第一行输入一个正整数 $t\le 100$，代表询问的次数。
每组询问共有4行，每行两个正整数。前两行用来表示矩阵$a$，后两行用来表示矩阵$b$。

输出描述：

输出$t$行，每行输出一行答案。
如果无论如何都不能使得矩阵$a$变成矩阵$b$，则输出-1。
否则输出一个整数，代表操作的最小次数。
示例1

输入
2
0 1
1 0
1 0
0 1
0 0
0 0
1 1
1 1
输出
2
-1

思路：

只有当两矩阵01数目不相等时才会不存在解。存在解的最大操作次数是2，当对角线不同时为2，当两行或两列不同时为2，其余情况为1。

AC代码

#include <bits/stdc++.h>
using namespace std;
const int N = 5;

int a[N][N], b[N][N];

int main()
{
    int T; cin >> T;
    while(T --)
    {
        int res = 0;
        int cnt1 = 0, cnt2 = 0;
        for(int i = 0; i < 2; i ++)
            for(int j = 0; j < 2; j ++)
            {
                cin >> a[i][j];
                if(a[i][j] == 1) cnt1 ++;
            }
        for(int i = 0; i < 2; i ++)
            for(int j = 0; j < 2; j ++)
            {
                cin >> b[i][j];
                if(b[i][j] == 1) cnt2 ++;
            }
        if(cnt1 != cnt2) res = -1;
        else
        {
            int f1 = 0, f2 = 0, f3 = 0, f4 = 0;
            if(a[0][0] != b[0][0]) f1 ++;
            if(a[0][1] != b[0][1]) f2 ++;
            if(a[1][0] != b[1][0]) f3 ++;
            if(a[1][1] != b[1][1]) f4 ++;
            int sum = f1 + f2 + f3 + f4;
            if(sum == 0) res = 0;
            else if((f2 && f3) || (f1 && f4)) res = 2;
            else if(sum == 2) res = 1;
            else res = 2;  
        }
        cout << res << endl;
    }
    return 0;
}