CCF NOI 1048.检测矩形

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

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

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

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

样例输出
OK

---

思路：在输入的同时用两个数组分别记录每行每列1的个数，然后在通过统计行和列的奇数的情况，如果两个情况都为0，则输出OK，如果都为1，则输出需要修改的元素坐标，数组中行为1的下标代表需要修改元素的行坐标，列为1的下标代表需要修改元素的列坐标，如果都不满足，则输出Corrupt。

---

代码：

#include <iostream>
#include <cstring>
using namespace std;
int main()
{
    int n;
    cin>>n;
    int num[n][n];
    int line[n],col[n];
    memset(line,0,sizeof(line));
    memset(col,0,sizeof(col));
    int k1,k2,x1=0,x2=0;

    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
        {    cin>>num[i][j];
            if(num[i][j]==1)
            {
                line[i]++;
                col[j]++;
            }
        }
    for(int i=0;i<n;i++)
    {
        if(line[i]%2)
        {
            x1++;
            k1=i;
        }
        if(col[i]%2)
        {
            x2++;
            k2=i;
        }
    }
    if(x1==0&&x2==0)
        cout<<"OK"<<endl;

    else if(x1==1&&x2==1)
        cout<<k1+1<<" "<<k2+1<<endl;
    else
        cout<<"Corrupt"<<endl;

    return 0;
}