Codeforces Round #495 (Div. 2)D. Sonya and Matrix(数学思维+构造)

D. Sonya and Matrix

题意：

输入$t(1e6)$,表示有一个矩形$n\times m=t$
接下来$a_1,a_2\dots ,a_t(0,t)$表示矩形内的数。
问构造$n,m,x,y$；其中$n,m$表示矩形，在$(x,y)$填$0$,$(x+1,y),(x-1,y),(x,y-1),(x,y+1)$填$1$,紧接着填$2,3\dots$,以此类推。要求填的数和$a$数组符合，不要求顺序。

题解：

考虑最大填的数为$(1,1),(n,m)$的距离，不妨设$(x,y)$距离$(n,m)$最远，于是就有
$mx=n+m-x-y$,又$n\times m=t$，$n,m$可枚举，$mx$根据数组可知，还有$x,y$未知。
然后，可以确定第一个小于$4\times i$的数字$i$,不妨设$x=i$,于是所有变量都已知了，最后判断一下就好了。

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+9;
int t,a[N],mx,low;
int tmp[N];
void check(int n,int m){
    int x=low,y=n+m-mx-x;
    memset(tmp,0,sizeof(tmp));
    if(x<=0||x>n||y<=0||y>m)return;
    for(int i=1;i<=n;i++)
      for(int j=1;j<=m;j++)tmp[abs(i-x)+abs(j-y)]++;
    for(int i=0;i<N;i++)if(tmp[i]!=a[i])return;
    cout<<n<<" "<<m<<endl;
    cout<<x<<" "<<y<<endl;
    exit(0);
}
int main(){
  //  freopen("tt.in","r",stdin),freopen("tt.out","w",stdout);
    cin>>t;
    for(int i=1,x;i<=t;i++)cin>>x,mx=max(mx,x),a[x]++;
    for(int i=0;;i++)if(a[i]<4*i){low=i;break;}
    for(int i=1;i<=sqrt(t);i++)if(t%i==0){
        check(i,t/i);
        check(t/i,i);
    }
    cout<<-1<<endl;
    return 0;
}