当前位置: 代码迷 >> 综合 >> G. Grid Coloring (思维 / 构造) 2020牛客暑期多校训练营(第六场)
  详细解决方案

G. Grid Coloring (思维 / 构造) 2020牛客暑期多校训练营(第六场)

热度:94   发布时间:2023-12-22 13:28:19.0

传送门

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
思路:

  • 给一个n*n的网格上色(n为边数),需1.满足k种颜色出现的次数相同,2.不会出现同色环,3.同行/列相邻两条边不同色且出现至少两种颜色。
  • 先考虑无解的情况:1.n == 1,不满足至少两种颜色的要求。2.k == 1,只有一种颜色肯定不能满足2,3要求。3. 2n(n+1) % k != 0,肯定无法平分给k中颜色。
  • 只需满足同行 / 列没有相邻的同色边,且相邻两行 / 列同一列 / 行的边颜色也不相同即可。
  • 当 n%k != 0,直接1~k 按顺序依次分配即可;而当 n%k == 0,在偶数行 / 列上循环移 动一 位即可。

代码实现:

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {
    {
    1, 0}, {
    -1, 0}, {
    0, 1}, {
    0, -1}};
using namespace std;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 2e5 + 5;int t, n, k;signed main()
{
    IOS;cin >> t;while(t --){
    cin >> n >> k;if(n == 1 || k == 1 || 2*n*(n+1) % k){
    cout << -1 << endl;continue;}if(n % k){
    int x = 0;for(int i = 1; i <= 2*(n+1); i ++){
    for(int j = 1; j <= n; j ++){
    x = x % k + 1;cout << x << " ";}cout << endl;}}else{
    int c = 0;for(int i = 1; i <= 2*(n+1); i ++){
    int x = c;for(int j = 1; j <= n; j ++){
    x = x % k + 1;cout << x << " ";}c ^= 1;cout << endl;}}}return 0;
}
  相关解决方案