C组 group3：D - Grid-00100

题目链接：https://vjudge.net/contest/397104#problem/D
题解：填一个01矩阵，满足sum和为k，且f(A)=(max®−min®)2+(max©−min©)2最小。
这个题思路是尽可能让每一行，每一列的1的个数都相等，因此f（A）只存在两种情况，一种是每一行，每一列个数相等，那么的话f（A）即为0；当无法满足的时候max减去min也是1，所以f（A）即为2。
现在考虑如何分布，因为均匀分布，所以每一行至少分布sum/n个。当无法整除时，前几行应该分布sum/n+1个。那么如何分布呢？即从以该行数为第一列，往后分配。
代码：

#include<iostream>
#include<algorithm>
#include<string>
#include<cstdio>
#include<cstring>
#include<cmath>
#define maxn 400
using namespace std;
int r[maxn];
int c[maxn];
int a[maxn][maxn];
int main()
{
    int t;//循环个数
    int n;//阶数
    int sum;//和
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&sum);
        memset(a,0,sizeof(a));
        if(sum%n==0)
            printf("0\n");
        else
            printf("2\n");
        int k=sum/n;//每行的个数
        int duo=sum%n;
        for(int i=1; i<=duo; i++)
        {
            for(int j=i; j<=i+k; j++)
            {
                if(j%n!=0)
                    a[i][j%n]=1;
                else
                    a[i][n]=1;
            }
        }
        for(int i=duo+1; i<=n; i++)
        {
            for(int j=i; j<i+k; j++)
            {
                if(j%n!=0)
                    a[i][j%n]=1;
                else
                    a[i][n]=1;
            }
        }
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=n; j++)
                cout<<a[i][j];
            cout<<endl;
        }
    }
    return 0;
}