Codeforces Round #644 (Div. 3) G. A/B Matrix (思维，构造)

题目链接
题意:
给你一个n*m的方阵(最开始全是0)，让你构造出一个每一行的元素和为a(每一位只能为0或者1)，每一列的元素和为b。
如果不可能构造出来，则输出-1.

思路:
1.首先我们得判断不可能构造出来的情况，在这里我们会发现，无论你在数组中的哪个位置添加一个1，那么这个1既在行上，也在列上(废话…)，那么满足题意的要求我们可以得出每一行的和为a，每一列的和为b。又因为每一个1必须同时在行和列上，那么n * a 必须等于 m * b(所有行的和，必须等于所有列的和)
由此来判断是否能构造出满足题意的方阵

2.如果能构造，那么我们交叉构造数组，如何构造呢?
就是每一行连续a个1，交错开来(具体看代码)

AC代码

#include <bits/stdc++.h>
inline int read(){char c = getchar();int x = 0,s = 1;
while(c < '0' || c > '9') {if(c == '-') s = -1;c = getchar();}
while(c >= '0' && c <= '9') {x = x*10 + c -'0';c = getchar();}
return x*s;}
using namespace std;
#define NewNode (TreeNode *)malloc(sizeof(TreeNode))
#define Mem(a,b) memset(a,b,sizeof(a))
const int N = 1e8 + 5;
const long long INFINF = 0x7f7f7f7f7f7f7f;
const int INF = 0x3f3f3f3f;
const double EPS = 1e-7;
const unsigned long long mod = 998244353;
const double II = acos(-1);
const double PP = (II*1.0)/(180.00);
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef pair<ll,ll> piil;
int main()
{
    std::ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    int t,n,m,a,b;
    cin >> t;
    while(t--)
    {
        cin >> n >> m >> a >> b;
        int x = 0,arr[n+5][m+5];
        memset(arr,0,sizeof(arr));//二维数组初始化
        if(n*a != m*b) {cout << "NO" << endl;continue;}
        else cout << "YES" << endl;
        for(int i = 0;i < n;i++)
            for(int j = 0;j < a;j++)
                {arr[i][x++] = 1;x %= m;}//交错赋值，本来可以arr[i][(++x)%m] = 1的，x的初值为-1，但是感觉花里胡哨了就没写。。
        for(int i = 0;i < n;i++)
            for(int j = 0;j < m;j++)
                j != m-1 ? cout << arr[i][j] : cout << arr[i][j] << endl;
    }
}