Codeforces 1016D Vasya And The Matrix（构造）

题目链接：Vasya And The Matrix

题意

构造一个 $n$ 行 $m$ 列的矩阵，要求构造矩阵的每一行的异或值为 $a_i$，每一列的异或值为 $b_i$。

输入

第一行包含两个整数 $n,m~(2\leq n,m\leq100)$，第二行有 $n$ 个整数 $a_1,a_2,\cdots,a_n~(0\leq a_i\leq10^9)$，第三行有 $m$ 个整数 $b_1,b_2,\cdots,b_m~(0\leq b_i\leq10^9)$。

输出

如果可以构造出满足条件的矩阵，第一行输出一个 $YES$，接着输出一个 $n\times m$ 的矩阵，要求矩阵的每一格元素 $0\leq c_{ij}\leq2\times10^9$，否则输出 $NO$。

样例

输入
2 3 2 9 5 3 13
输出
YES 3 4 5 6 7 8

输入
3 3 1 7 6 2 15 12
输出
NO

题解

前 $n-1$ 行 $m-1$ 列都填 $0$，最后一行的前 $m-1$ 列分别填上 $b_1,b_2,\cdots,b_{m-1}$，最后一列的前 $n-1$ 行分别填上 $a_1,a_2,\cdots,a_{n-1}$，最后根据异或的运算规则，分别从行和列的异或结果计算出第 $n$ 行第 $m$ 列的值，如果两次计算得到的值相等，则可以构造出满足条件的矩阵，否则输出 $NO$。

过题代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <climits>
#include <cstring>
#include <string>
#include <vector>
#include <list>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <bitset>
#include <algorithm>
#include <functional>
#include <iomanip>
using namespace std;

#define LL long long
const int maxn = 100 + 100;
int n, m;
int a[maxn], b[maxn];
int ans[maxn][maxn];

int main() {
    #ifdef LOCAL
        freopen("test.txt", "r", stdin);
//    freopen("out.txt", "w", stdout);
    #endif // LOCAL
    ios::sync_with_stdio(false);

    while(scanf("%d%d", &n, &m) != EOF) {
        for(int i = 1; i < n; ++i) {
            for(int j = 1; j < m; ++j) {
                ans[i][j] = 0;
            }
        }
        for(int i = 1; i <= n; ++i) {
            scanf("%d", a + i);
            ans[i][m] = a[i];
        }
        for(int i = 1; i <= m; ++i) {
            scanf("%d", b + i);
            ans[n][i] = b[i];
        }
        int tmp = 0;
        for(int i = 1; i < n; ++i) {
            tmp ^= ans[i][m];
        }
        ans[n][m] = tmp ^ ans[n][m];
        tmp = 0;
        for(int i = 1; i <= m; ++i) {
            tmp ^= ans[n][i];
        }
        if(tmp == a[n]) {
            printf("YES\n");
            for(int i = 1; i <= n; ++i) {
                for(int j = 1; j <= m; ++j) {
                    if(j != 1) {
                        printf(" ");
                    }
                    printf("%d", ans[i][j]);
                }
                printf("\n");
            }
        } else {
            printf("NO\n");
        }
    }

    return 0;
}