NOI / 1.8编程基础之多维数组 04:错误探测

描述

给定n*n由0和1组成的矩阵，如果矩阵的每一行和每一列的1的数量都是偶数，则认为符合条件。
你的任务就是检测矩阵是否符合条件，或者在仅改变一个矩阵元素的情况下能否符合条件。
"改变矩阵元素"的操作定义为0变成1或者1变成0。

输入

输入n + 1行，第1行为矩阵的大小n(0 < n < 100)，以下n行为矩阵的每一行的元素，元素之间以一个空格分开。

输出

如果矩阵符合条件，则输出OK；
如果矩阵仅改变一个矩阵元素就能符合条件，则输出需要改变的元素所在的行号和列号，以一个空格分开。
如果不符合以上两条，输出Corrupt。

样例输入

样例输入1
4
1 0 1 0
0 0 0 0
1 1 1 1
0 1 0 1

样例输入2
4
1 0 1 0
0 0 1 0
1 1 1 1
0 1 0 1

样例输入3
4
1 0 1 0
0 1 1 0
1 1 1 1
0 1 0 1

样例输出

样例输出1
OK

样例输出2
2 3

样例输出3
Corrupt

#include<bits/stdc++.h>
using namespace std;

int main()
{
    int n;
    cin>>n;
    vector<vector<int>> a(n, vector<int>(n));//二维数组
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<n; j++)
        {
            cin>>a[i][j];
        }
    }
    vector<int> b(n), c(n);//来存储每一行或列的奇偶数
    for(int i=0; i<n; i++)//行
    {
        for(int j=0; j<n; j++)
        {
            b[i] = b[i] + a[i][j];//每一行的总和
        }
        b[i] =b[i] % 2 ;//看是奇数个还是偶数个，然后存到b里
    }
    for(int j=0; j<n; j++)//列
    {
        for(int i=0; i<n; i++)
        {
            c[j] = c[j] + a[i][j];
        }
        c[j] = c[j] % 2;
    }
    bool flag=true;
    int b1=0,c1=0;//行总共有多少奇数行，列同理
    int b2=0,c2=0;//只有 b1和c1都为1的时候它才显示正确
    for(int i=0;i<n;i++)
    {
        if(b[i]!=0){//如果不是偶数
            flag=false;
            b1++;//统计所有行有多少行是奇数行
            b2=i;//这个只有b1等于1的时候显示匹配的值，记录这一行是哪一行
        }
        if(c[i]!=0){
            flag=false;
            c1++;
            c2=i;
        }
    }
    if(flag){
        cout<<"OK";
        return 0;
    }

    if(b1==1 && c1==1)
    {
        cout<<b2+1<<" "<<c2+1;//我是从0开始奇数，题目从一开始所以加1
    }
    else cout<<"Corrupt";

}