Codeforces G. A/B Matrix (贪心 / 思维) (Round #644 Div.3)

传送门

题意： 给出 n, m, a, b四个数，试问是否可以列出一个每行a个1，每列b个1，其余全是0的 n * m的矩阵？ 如果可以输出"YES"并打出一个可行的矩阵，否则输出"NO"。

思路：

• 显然只有横着数的n * a与竖着数的m * b相等才会有可行的方案
• 然后每行依次填充a个1即可，每行的1相连，且下一行的1紧接上一行1的后面。

刚开始一直处理不好1的阶梯填充，后面才想通其实很简单，只需要每一行循环填充a次，直接找到 (p + 1) % m的位置即可。

代码实现：

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <cctype>
#include <cstring>
#include <iostream>
#include <sstream>
#include <string>
#include <list>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <algorithm>
#include <functional>
#define int long long
#define lowbit(x) (x &(-x))
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pll;
const int  INF = 0x3f3f3f3f3f3f3f3f;
const double PI = acos(-1.0);
const double EXP = 1e-8;
const ll   MOD = 1e9 + 7;
const int  N = 2e5 + 5;

int t, n, m, a, b;
int mp[55][55];

signed main()
{
    IOS;

    cin >> t;
    while(t --){
        mem(mp, 0);
        cin >> n >> m >> a >> b;
        
        if(n * a == m * b){
            cout << "YES" << endl;
            int p = 0;
            
            for(int i = 0; i < n; i ++){
                for(int j = 0; j < a; j ++){
                    mp[i][p] = 1;
                    p = (p + 1) % m;
                }
            }

            for(int i = 0; i < n; i ++){
                for(int j = 0; j < m; j ++)
                    cout << mp[i][j];
                cout << endl;
            }
        }
        else cout << "NO" << endl;
    }

    return 0;
}