ZOJ - 3212 K-Nice 思维 推公式

ZOJ - 3212 K-Nice

题意：

给你 n，m 和 k，要你输出一个矩阵n行m列，它有 k 个nice 元素。一个元素的 上下左右 四个元素的和与这个元素的值相同，则称这个元素为一个nice。 nice元素必须是非边上的元素（题目没说。。。）。

思路：

为了方便我们可以只用 0和1 两种数字来填矩阵。观察发现：

当 k=(n-2)*(m-2)时，矩阵中非边上的位置都需要满足条件，所以我们全填 0 就行。

其他情况我们可以推出一个给矩阵中放 0 元素个数的公式：sum = (k/(m-2)+2)*m+k%(m-2)+1  。一行最多可以有 m-2 个元素为nice，(k/(m-2)+2)   算出一共有多少行 0 元素（加 2 是因为第一行在边上不能算，最后一行要满足条件，则他的下（四个方向中的下一行）位置也要为0，所以需要再加一行），k%(m-2)+1 算出余下多少元素（加 1 是因为最后一行的第一个元素是边上的元素所以加 1 ）。

5 5 4
0 0 0 0 0
0 0 0 0 0
0 0 0 0 0
0 0 1 1 1
1 1 1 1 1

#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<sstream>
#include<cstring>
#include<bitset>
#include<cstdio>
#include<string>
#include<deque>
#include<stack>
#include<cmath>
#include<queue>
#include<set>
#include<map>
#define mod 1000000007
using namespace std;
typedef long long ll;
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,m,k;
        scanf("%d%d%d",&n,&m,&k);
        if(k==(n-2)*(m-2))
        {
            for(int i=0;i<n;i++)
            {
                for(int j=0;j<m-1;j++)
                    printf("0 ");
                printf("0\n");
            }
        }
        else
        {
            int flag = 0;
            int sum = (k/(m-2)+2)*m+k%(m-2)+1;
            for(int i=0;i<n;i++)
            {
                for(int j=0;j<m;j++)
                {
                    if(flag)   // 0 元素放完 其余位置全放 1
                        printf("1");
                    else
                        printf("0");
                    sum--;
                    if(sum==0)
                        flag = 1;
                    if(j==m-1)
                        printf("\n");
                    else
                        printf(" ");
                }

            }

        }
    }
    return 0;
}