【NAIPC2018】Recovery 题解

思路： 对于每一行，如果元素个数为奇数，输入值为0，那么就得有一个0；如果元素个数为偶数，输入值为1，也得有一个0。列同理。我们就得到需要0的行数Cntr和列数Cntc，显然当Cntr==Cntc时，解很容易求，继续推导，当Cntr+Cntc为偶数时也有解，我们假设分别为5 和 3， 我们可以将第这5行的前2行的第一列对应元素赋值为0（改变了偶数个对应列的奇偶性不变）；显然Cntr+Cntc为奇数时无解

代码：

 #include<iostream>
#include<cstdio>
#include<string>
#include<string.h>
using namespace std;
char r[51], c[51];
int ans[51][51];
int main()
{
	scanf("%s%s",r, c);
	int n=strlen(r), m=strlen(c), Cntr=0, Cntc=0;
	for(int i=0; i<n; i++)
	{
		for(int j=0; j<m; j++)
		{
			ans[i][j]=1;
		}
	}

	if(m&1)
	{
		for(int i=0; i<n; i++)
		{
			r[i]=r[i]=='0' ? '1' : '0';  //r[i]==1代表该行缺0
		}
	}
	if(n&1)
	{
		for(int j=0; j<m; j++)
		{
			c[j]=c[j]=='0' ? '1' : '0'; //列缺0
		}
	}
	for(int i=0; i<n; i++)
	{
		if(r[i]=='1') Cntr++;
	}
	for(int j=0; j<m; j++)
	{
		if(c[j]=='1') Cntc++;
	}
	//串中1的个数和为奇数时无解	
	if((Cntr+Cntc)&1)
	{
		printf("-1\n");
		return 0;
	}
	//将行或者列中多余的1在第一行或者列中消耗掉(消耗偶数个) 
	for(int i=0; Cntr>Cntc; i++)
	{
		if(r[i]=='1') 
		{
			r[i]='0';
			ans[i][0]=0;
			Cntr--;
		}
	}
	for(int i=0; Cntc>Cntr; i++)
	{
		if(c[i]=='1') 
		{
			c[i]='0';
			ans[0][i]=0;
			Cntc--;
		}
	}
	//剩下的Cntr==Cntc对应匹配
	for(int i=0, j=0; i<n; i++)
	{
		if(r[i]=='1')
		{
			while(c[j]=='0') ++j;
			ans[i][j]=0;
			++j;
		}
	}
	for(int i=0; i<n; i++)
	{
		for(int j=0; j<m; j++)
		{
			printf("%d", ans[i][j]);
		}
		printf("\n");
	}
	return 0;
}