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;
}