CF1521E Nastia and a Beautiful Matrix

CF1521E Nastia and a Beautiful Matrix

其实这题并不需要排序，下面会给出证明

答案满足单调性，可以通过二分找出边长 (len)。如下图所示将网格染成四种颜色：

饱和的情况就是把所有非白色的格子都填上数，共 (S=len^2-lfloor frac{len}{2} floor^2) 个，那么二分第一个条件是 (Sgesum a_i)

然后考虑众数 (x)，为了满足对角线不重复，它最多只能放在蓝色和红(或黄)色的格子里，共 (T=len imes lceil frac{len}{2} ceil) 个，需满足 (Tge a[x])

对于满足以上两个条件的 (len)，一定可以构造出一个解，方法如下：

把所有数排成一列，相同的数排在一起，(x) 排在最前面，其他数顺序随意，然后先往红格子放，红的放完了放蓝的，最后放黄的。

简单的证明一下所有 (2 imes 2) 对角线不重复：所有对角线都由一个黄的一个红的组成，进行分类，

1、如果 (x) 填满了红格子，并且根据二分条件 (x) 最多填满蓝的，不会填到黄的，所以黄和红不重复

2、如果 (x) 没填满红格子，若出现重复，则存在一个 (y)，在红格子里填了，又填满了蓝的，还得填黄的。而格子数量 蓝 (ge) 红，(x) 又是众数，显然不可能

所以其他数的填放顺序根本无所谓，不需要排序。

#include <bits/stdc++.h>
using namespace std;
#define int long long
void read (int &x) {
    char ch = getchar(); x = 0; while (!isdigit(ch)) ch = getchar();
    while (isdigit(ch)) x = x * 10 + ch - 48, ch = getchar();
} const int N = 2e5 + 5, M = 2000, NN = N * 5;
int n, s, mx, a[N], res[M][M], sr, sb, sy, id[N];
pair<int, int> r[NN], b[NN], y[NN];
#define xx first
#define yy second
int get () {
    int l = 0, r = 1000, mid, res = 0;
    while (l <= r) {
        mid = l + r >> 1;
        if (s <= mid * mid - (mid / 2) * (mid / 2) && a[1] <= mid * ((mid + 1) / 2))
            res = mid, r = mid - 1;
        else l = mid + 1;
    }
    return res;
}
signed main() {
    int T; read (T);
    while (T--) {
        read (s), read (n); mx = 0;
        for (int i = 1; i <= n; ++i) {
            read (a[i]), id[i] = i;
            if (a[i] > a[mx]) mx = i;
        }
        id[1] = mx, id[mx] = 1; swap (a[1], a[mx]);
        int len = get (); sr = sb = sy = 0;
        for (int i = 2; i <= len; i += 2)
            for (int j = 1; j <= len; j += 2) r[++sr] = {i, j};
        for (int i = 1; i <= len; i += 2)
            for (int j = 1; j <= len; j += 2) b[++sb] = {i, j};
        for (int i = 1; i <= len; i += 2)
            for (int j = 2; j <= len; j += 2) y[++sy] = {i, j};
        for (int i = 1; i <= len; ++i)
            for (int j = 1; j <= len; ++j) res[i][j] = 0;
        for (int i = 1; i <= n; ++i) {
            for (int j = 1; j <= a[i]; ++j) {
                if (sr) res[r[sr].xx][r[sr].yy] = id[i], --sr;
                else if (sb) res[b[sb].xx][b[sb].yy] = id[i], --sb;
                else res[y[sy].xx][y[sy].yy] = id[i], --sy;
            }
        }
        printf ("%d
", len);
        for (int i = 1; i <= len; ++i) {
            for (int j = 1; j <= len; ++j) printf ("%d ", res[i][j]); puts ("");
        }
    }
    return 0;
}