补题向 | Vasya And The Matrix(思维?构造?)

D. Vasya And The Matrix

Examples

input

2 3
2 9
5 3 13

output

YES
3 4 5
6 7 8

input

3 3
1 7 6
2 15 12

output

NO

给出每一行每一列的异或和,求出符合要求的原矩阵,所有行的异或和与所有列的异或和都为整个矩阵所有数的异或和,若两者不相等,则没有符合要求的矩阵。如果有,则随意构造一个n-1*m-1的矩阵(例如:1、2、3……),再求出能使最终异或和为题目所求的最后一行和最后一列的数,(求出每一行、每一列的异或和,与最终异或和相异或即能求出最后的数)

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define ll long long
int n,m;
ll a[105];
ll b[105];
ll ans[105][105];
int main(){
	while(~scanf("%d%d",&n,&m)){
		ll rs=0,cs=0;
		for(int i=1;i<=n;i++){
			scanf("%I64d",&a[i]);
			rs^=a[i];
		}
		for(int j=1;j<=m;j++){
			scanf("%I64d",&b[j]);
			cs^=b[j];
		}
		if(rs==cs){
			printf("YES\n");
		}
		else{
			printf("NO\n");
			continue;
		}
		int k=1;
		memset(ans,0,sizeof(ans));
		for(int i=1;i<n;i++){
			for(int j=1;j<m;j++){
				ans[i][j]=k; 
				ans[i][m]^=k;//用每一行最后一个格保留每一行的异或和,以便后续运算 
				ans[n][j]^=k;//用每一列最后一个格保留每一列的异或和,以便后续运算
				k++;
				
			}
		}
		for(int j=1;j<m;j++){
			ans[n][j]^=b[j];
		}
		for(int i=1;i<n;i++){
			ans[i][m]^=a[i];
		}
		
		for(int i=1;i<m;i++){
			ans[n][m]^=ans[n][i];
		}
		ans[n][m]^=a[n];
		
		
		for(int i=1;i<=n;i++){
			for(int j=1;j<=m;j++){
				printf("%I64d",ans[i][j]);
				if(j==m)printf("\n");
				else printf(" ");
			}
		}
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/bekote/article/details/81489253