Submission #1234052

#	Time	Username	Problem	Language	Result	Execution time	Memory
1234052		Hanksburger	Gardening (RMI21_gardening)	C++20	100 / 100	11 ms	968 KiB

Main

#include <bits/stdc++.h>
#define int long long
using namespace std;
vector<vector<int>> ans;
void draw(int xl, int xr, int yl, int yr, int val)
{
    for (int i=yl; i<=yr; i++)
        ans[xl][i]=ans[xr][i]=val;
    for (int i=xl; i<=xr; i++)
        ans[i][yl]=ans[i][yr]=val;
}
signed main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    int t;
    cin >> t;
    while (t--)
    {
        int n, m, k, swapped=0, cnt=0, nn, mm;
        cin >> n >> m >> k;
        if (n&1 || m&1)
        {
            cout << "NO\n";
            continue;
        }
        n/=2, m/=2, k=n*m-k;
        if (n>m)
        {
            swap(n, m);
            swapped=1;
        }
        if (n==m && !(k==0 || (2<=k && k<=(n-1)*m-2) || k==(n-1)*m))
        {
            cout << "NO\n";
            continue;
        }
        if (!(k==0 || (2<=k && k<=(n-1)*m)))
        {
            cout << "NO\n";
            continue;
        }
        nn=n, mm=m, ans=vector<vector<int>>(n*2, vector<int>(m*2, 0));
        while (k && (k==nn+mm-2 || k>=nn+mm))
        {
            draw(cnt, cnt+nn*2-1, cnt, cnt+mm*2-1, cnt+1);
            k-=nn+mm-2, nn--, mm--, cnt++;
        }
        if (k==nn+mm-1)
        {
            draw(cnt, cnt+5, cnt, cnt+3, cnt+1);
            draw(cnt, cnt+nn*2-1, cnt+4, cnt+mm*2-1, cnt+2);
            cnt+=2;
        }
        else if (k)
        {
            draw(cnt, cnt+max(2LL, k-mm+2)*2-1, cnt, cnt+min(mm, k)*2-1, cnt+1);
            cnt++;
        }
        for (int i=0; i<n*2; i++)
            for (int j=0; j<m*2; j++)
                if (!ans[i][j])
                    draw(i, i+1, j, j+1, ++cnt);
        cout << "YES\n";
        for (int i=0; i<(swapped?m:n)*2; i++)
        {
            for (int j=0; j<(swapped?n:m)*2; j++)
                cout << (swapped?ans[j][i]:ans[i][j]) << ' ';
            cout << '\n';
        }
    }
}

• Subtask 1: 5 / 5
• Subtask 2: 6 / 6
• Subtask 3: 10 / 10
• Subtask 4: 18 / 18
• Subtask 5: 39 / 39
• Subtask 6: 22 / 22