一、题目
二、解法
考虑如果我们确定了行和列(行小),怎么填?最优的方法是斜着填,如下图(不同的颜色代表不同的值):
每个数最多能填
n
n
n次,可以看出只要个数够就一定能填。
不确定行和列,我们可以枚举。枚举行,我们就知道每个数最多能填多少次,就知道最多填的总数(联系上文),可以反算出最大的列数。知道了行和列后答案就用斜着填的方法构造就行了,时间复杂度 O ( n ) O(n) O(n)
#include <cstdio>
#include <algorithm>
using namespace std;
const int M = 400005;
int read()
{
int num=0,flag=1;
char c;
while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
while(c>='0'&&c<='9')num=(num<<3)+(num<<1)+(c^48),c=getchar();
return num*flag;
}
int n,m,r,cc,tot,a[M],b[M],vis[M],ans[M];
struct node
{
int x,y;
bool operator < (const node &b) const
{
return x<b.x;
}
}c[M];
signed main()
{
n=read();
for(int i=1;i<=n;i++)
a[i]=read();
sort(a+1,a+1+n);
for(int i=1;i<=n;i++)
{
if(a[i]!=a[i-1]) b[i]=++m;
else b[i]=m;
vis[b[i]]=a[i];
c[b[i]].x++;
c[m].y=m;
}
sort(c+1,c+1+m);tot=n;
for(int i=n;i>=1;i--)
{
for(int j=m;j>=1 && c[j].x>i;j--)
{
c[j].x--;
tot--;
}
int j=tot/i;
if(i>j) continue;
if(r*cc<i*j)
{
r=i;cc=j;
}
}
printf("%d\n%d %d\n",r*cc,r,cc);
for(int i=1;i<=m;i++)
{
c[i].x=0;
c[i].y=i;
}
for(int i=1;i<=n;i++) c[b[i]].x++;
sort(c+1,c+1+m);
for(int i=m;i>=1;i--)
if(c[i].x>=r) c[i].x=r;
else break;
for(int i=1;i<=cc;i++)
for(int j=1;j<=r;j++)
{
if(c[m].x==0) m--;
ans[(j-1)*cc+(i+j)%cc+1]=vis[c[m].y];
c[m].x--;
}
for(int i=1;i<=r*cc;i++)
{
printf("%d ",ans[i]);
if(i%cc==0) puts("");
}
}