G - Matrix Repair (拓扑排序)

Problem - G - Codeforces

 Bella现在正在学习网络数据传输。她需要传输一个N × N位矩阵。第i行第j列的元素用Ai,j E{0,1}表示。然而，在传输过程中，矩阵的一些元素可能会丢失。这些元素用-1表示。为了使传输更可靠，她决定记录下每一行和每一列的校验和。位集S的校验和是所有元素的逐位异或。当且仅当集合中有奇数个1时它是1。例如，{0,0,1,1,1}的校验和为1。第i行校验和表示为Ri。第j列的校验和表示为Cy，在传输过程中校验和不会丢失。现在Bella收到了矩阵和校验和，她想知道原始矩阵是什么。请帮她找出原始矩阵。输入第一行包含一个正整数N(1<N<103)。接下来的N行每一行包含N个整数，表示矩阵a。第j个元素在第i行表示Aij E{- 1,0,1}。下一行包含n个整数，表示行的校验和。第i个元素表示Ri E{0,1}。下一行包含n个整数，表示列的校验和。第j个元素表示C;E{0,1}。可以保证至少有一种方法可以将A中的-1替换为0或1，从而满足R和C。输出如果可以确定原始矩阵，输出N行。每行包含N个整数，表示原始矩阵。否则，输出-1。

Sample 1

Inputcopy	Outputcopy
3 1 -1 0 0 1 0 1 1 1 1 1 1 0 0 1	1 0 0 0 1 0 1 1 1

Sample 2

Inputcopy	Outputcopy
2 -1 -1 -1 -1 1 0 0 1	-1

Sample 3

Inputcopy	Outputcopy
3 1 1 0 0 -1 -1 -1 1 1 0 1 1 0 1 1	1 1 0 0 1 0 1 1 1

题解:
题中说要我们找出一个确定的矩阵,有两种是不符合题意的

1.本身无法构成符合C[i],R[i]的矩阵

2.可以构成符合C[i],R[i]的矩阵,但是有多种情况

所以符合题意的只有一种情况既可以符合C[i],R[i]

那么我们利用拓扑排序的思想,每次首先把每一行或每一列里,只有一个为-1,那么我们可以根据这一行或一列除了他的异或和 ^ C[i]或R[i],就是这个位置原本的值,然后更新

再找这样的-1,最后看是否每一行,每一列,都有确定值

#include<iostream>
#include<string>
#include<vector>
#include<cstring>
#include<algorithm>
#include<queue>
#include<map>
using namespace std;
typedef long long ll;
#define int long long
typedef pair<int,int> PII;
//int ans[100000];
//int a[100050];
int a[1005][1005]; 
int r[1005];
int c[1005];
int R[1004];
int C[1005];
int cr[1005];
int cc[1005];
void solve()
{
	int n;
	cin >> n;
	for(int i = 1;i <= n;i++)
	{
		for(int j = 1;j <= n;j++)
		{
			cin >> a[i][j];
			if(a[i][j] != -1)
			{
				r[i] = r[i] ^ a[i][j];
				c[j] = c[j] ^ a[i][j];
				cr[i]++;
				cc[j]++;	
			}
		}
	}
	for(int i = 1;i <= n;i++)
	{
		cin >> R[i];
	}
	for(int i = 1;i <= n;i++)
	{
		cin >> C[i];
	} 
	queue<PII> q;
	for(int i = 1;i <= n;i++)
	{
		for(int j = 1;j <= n;j++)
		{
			if(a[i][j] == -1&&(cr[i] == n - 1|| cc[j] == n - 1))
			{
				q.push({i,j});
			}
		}
	}
	while(q.size())
	{
		auto [x,y] = q.front();
		q.pop();
		if(a[x][y] != -1)
		continue;
		if(cr[x] == n - 1)
		a[x][y] = r[x] ^ R[x];
		else
		a[x][y] = c[y] ^ C[y];
		r[x] ^= a[x][y];
		c[y] ^= a[x][y];
		cr[x]++;
		cc[y]++;
		if(cr[x] == n - 1)
		{
			for(int i = 1;i <= n;i++)
			{
				if(a[x][i] == -1)
				{
					q.push({x,i});
					break;
				}
			}
		} 
		if(cc[y] == n - 1)
		{
			for(int i = 1;i <= n;i++)
			{
				if(a[i][y] == -1)
				{
					q.push({i,y});
					break;
				}
			}
		}
	}
	int f = 1;
	for(int i = 1;i <= n;i++)
	{
		if(cc[i] != n||cr[i] != n)
		{
			f = 0;
		}
	}
	if(f)
	{
		for(int i = 1;i <= n;i++)
		{
			for(int j = 1;j <= n;j++)
			{
				cout << a[i][j] <<" ";
			}
			cout <<"\n";
		}
	}
	else
	cout <<"-1";
}
signed main() 
{
	ios::sync_with_stdio(0);
	cin.tie(0);cout.tie(0);
	int t = 1;
//	cin >> t;
//    scanf("%lld",&t);
	while (t--) 
	{
		solve();
	}
	return 0;
}