RCC 2017 Qual 1 Mail.Ru, April 2, 2017 Problem B. Painting the Wall

题意：给一个n*m(<=100)的图，图中0表示台灯，1表示空地，以台灯和墙作为边界，问能否使用k种使得联通的线段上没有重复的颜色

分析：只要连续的空地不超过k个就必然存在一个解，可以选择如下构造，假设n=5,m=5,k=3

1 2 3 1 2

2 3 1 2 3

3 1 2 3 1

1 2 3 1 2

2 3 1 2 3

这样，把灯的位置都换为0，如果检测到没有连续的超过k的线段，那么这就是一个可行解

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e2+5;
int g[maxn][maxn],res[maxn][maxn];
int t,n,m,k;

void init(){
    for(int i=1;i<=n;i++){
        res[i][1]=(res[i-1][1]+1)%k;
        for(int j=2;j<=m;j++)res[i][j]=(res[i][j-1]+1)%k;
    }
}

void print(int x,int y){
    if(!g[x][y])cout<<0;
    else cout<<res[x][y]+1;
}

int main(){
     int t;cin>>t;
     while(t--){
         cin>>n>>m>>k;init();
         
         for(int i=1;i<=n;i++)for(int j=1;j<=m;j++){
             cin>>g[i][j];
             if(!g[i][j])res[i][j]=0;
         }
         
         bool flag=1;
         
         for(int i=1;i<=n;i++){
             int j=1;
             for(;j<=m;j++){
                 int cnt=0;
                 while(j<=m&&g[i][j])cnt++,j++;
                 if(cnt>k){flag=0;break;}
             }
         }
         
         for(int i=1;i<=m;i++){
             int j=1;
             for(;j<=n;j++){
                 int cnt=0;
                 while(j<=n&&g[j][i])cnt++,j++;
                 if(cnt>k){flag=0;break;}
             }
         }
         if(!flag){puts("NO");continue;}
         puts("YES");
         
         for(int i=1;i<=n;i++){
             print(i,1);
             for(int j=2;j<=m;j++)cout<<" ",print(i,j);
             cout<<endl;
         }
     }
    return 0;
}