[CF1276C]Beautiful Rectangle

题目

传送门 to luogu

思路

部分枚举是不错的思路，考虑枚举其行数。不妨设行数 $r\le$ 列数 $c$ 。

若数字 $x$ 出现了 $v_x$ 次，则 $x$ 可以在矩阵中填入 $\min (v_x,r)$ 次，原因是直接按照斜线填。形如：

$$\left|\begin{array}{cccc}x& & & y\\ & x& & \\ & & x& \end{array}\right|$$

下一步就接着往下填 $y$ 。不难发现，任意连续的 $t(t\le r)$ 个格子满足不在同一行、同一列上。只有一个例外，就是 $n\times n$ 的矩阵，可能有

$$\left|\begin{array}{ccc}y& x& \\ & x& \\ & & x\end{array}\right|$$

连续的 $3$ 个 $x$ 却有两个在同一列。所以这种情况先填出现次数多的，以避免单独的 $y$ 卡在上面的情况。

形式化地，我们用 $f_x(t),f_y(t)$ 表示第 $t$ 个数所在的行列坐标。因为有取模，我们规定坐标范围是 $[0,n)$ 与 $[0,m)$ ，则

$$f_x(t)=t\mathrm{m}\mathrm{o}\mathrm{d}n$$

$$f_y(t)=[\lfloor t\cdot \gcd (n,m)/nm\rfloor +t]\mathrm{m}\mathrm{o}\mathrm{d}m$$

其中，$\frac{t}{\mathtt{l}\mathtt{c}\mathtt{m}(n,m)}$ 是进行的修正。因为移动 $\mathtt{l}\mathtt{c}\mathtt{m}$ 次之后就会重复。

既然如此，我们可以直接用 ${\sum }_{i=1}^n\min (v_i,r)$ 计算列数，然后更新答案即可。

复杂度 $\mathcal{O}(n\log n)$ 。因为要排序离散化。

代码

#include <cstdio>
#include <iostream>
#include <vector>
#include <cmath>
#include <algorithm>
using namespace std;
inline int readint(){
	int a = 0; char c = getchar(), f = 1;
	for(; c<'0'||c>'9'; c=getchar())
		if(c == '-') f = -f;
	for(; '0'<=c&&c<='9'; c=getchar())
		a = (a<<3)+(a<<1)+(c^48);
	return a*f;
}
template < typename T >
void getMax(T&a,const T&b){if(a<b)a=b;}
template < typename T >
void getMin(T&a,const T&b){if(b<a)a=b;}

const int MaxN = 400005;
int a[MaxN], n;

int tmp[MaxN];
void input(){
	n = readint();
	for(int i=1; i<=n; ++i)
		a[i] = readint();
	for(int i=1; i<=n; ++i)
		tmp[i] = a[i];
	sort(tmp+1,tmp+n+1);
	int *r = unique(tmp+1,tmp+n+1);
	# define lb lower_bound
	for(int i=1; i<=n; ++i)
		a[i] = lb(tmp+1,r,a[i])-tmp;
	# undef lb
}

int cnt[MaxN]; // cnt[num]为num出现次数
int bucket[MaxN]; // [cnt]几个数出现cnt次
struct Node{
	int x, y, val;
	bool operator < (const Node &that) const {
		if(x == that.x)
			return y < that.y;
		return x < that.x;
	}
} node[MaxN];
Node want[MaxN];
void solve(){
	for(int i=1; i<=n; ++i)
		++ cnt[a[i]];
	int now = 0, delta = 0;
	for(int i=1; i<=n; ++i){
		++ bucket[cnt[i]];
		if(cnt[i]) ++ delta;
	}
	int ans = 0, row, col;
	for(int i=1; i*i<=n; ++i){
		now += delta, delta -= bucket[i];
		if(now/i < i) continue;
		if(ans < now/i*i)
			ans = now/i*i, row = i;
	}
	col = ans/row;
	printf("%d\n%d %d\n",ans,row,col);
	for(int i=1; i<=n; ++i){
		getMin(cnt[i],row);
		want[i].x = cnt[i];
		want[i].y = 0;
		want[i].val = i;
	}
	sort(want+1,want+n+1);
	for(int i=1; i<n+1-i; ++i)
		swap(want[i],want[n+1-i]);
	int T = ans/__gcd(row,col);
	for(int i=0,x=0,y=0,j=1; i<ans; ++i){
		if(i != 0 && i%T == 0) ++ y;
		x = x%row+1, y = y%col+1;
		while(cnt[want[j].val] == 0) ++ j;
		node[i].x = x, node[i].y = y;
		node[i].val = tmp[want[j].val];
		-- cnt[want[j].val];
	}
	sort(node,node+ans);
	for(int i=0; i<row; ++i){
		printf("%d",node[i*col].val);
		for(int j=1; j<col; ++j)
			printf(" %d",node[i*col+j].val);
		putchar('\n');
	}
}

int main(){
	input(), solve();
	return 0;
}