Codeforces D. Grid-00100(构造 / 思维) (Round #654 Div.2)

传送门

题意： 构造一个n*n的01方格A(即方格内只填0或1)，且1的个数正好为k。规定f(A) = (max( R ) - min( R )) ^ 2 + (max( C ) - min( C )) ^ 2.(其中R表示某一行1的个数，C表示某一列1的个数)。输出f(A)值最小的方格A。

思路：

• 若 k % n == 0，那么肯定每行每列的1都相等，这是f(A) = 0.
• 若k % n != 0的话，注定不能每行每列的1都相等，这无可厚非了，那么就去最小的 1^2 + 1^2 = 2岂不美哉！只不过前 k % n行比后面的多一个1，其他还是像前面那么构造。

代码实现：

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#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;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 2e5 + 5;

int n, k, t, a[500][500];

signed main()
{
	cin >> t;
	while(t--){
        me(a);
		cin >> n >> k;
		int q = k % n, h = k / n;
		cout << (!q ? '0' : '2') << endl;
		for(int i = 0; i < q; i ++)//前q行每行放s+1个
		    for(int j = i; j <= i + h; j ++)
			    a[i][j % n] = 1;
		for(int i = q; i < n; i ++)//后面的行每行放s个
		    for(int j = i; j < i + h; j ++)
			   a[i][j % n] = 1;
		for(int i = 0; i < n; i ++){
			for(int j = 0; j < n; j ++)
                cout << a[i][j];
			cout << endl;
		}
	}
	return 0;
}