Description
一n*m网格,给出k种颜色及其数量,要求用这些颜色对网格染色,要求相邻两个格子颜色不同,问是否存在合法方案,如果存在输出YES并输出任一种方案,否则输出NO
Input
第一行为一整数T表示用例组数,每组用例第一行为三个整数n,m,k分别表示网格行列数和颜色种类数,第二行k个整数分别表示这k种颜色的数量numi
(1<=T<=5000,1<=n,m<=5,1<=k<=n*m,sum(numi)=n*m)
Output
对于每组用例,如果存在合法染色方案则输出YES并输出任一种方案,否则输出NO
Sample Input
4
1 5 2
4 1
3 3 4
1 2 2 4
2 3 3
2 2 2
3 2 3
2 2 2
Sample Output
Case #1:
NO
Case #2:
YES
4 3 4
2 1 2
4 3 4
Case #3:
YES
1 2 3
2 3 1
Case #4:
YES
1 2
2 3
3 1
Solution
dfs,首先对颜色数量排序,每次优先使用数量多的,之后一行一行搜索,到行尾就换行,每次搜索先判断当前剩余颜色中是否有一种颜色的数量已经超过剩余未染色格子数量的一半了,如果超过那么直接回溯,否则继续深搜
Code
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
#define INF 0x3f3f3f3f
typedef long long ll;
int T,n,m,k,map[6][6];
struct node
{
int num,id;
}c[33];
int cmp(node a,node b)
{
return a.num>b.num;
}
int dfs(int x,int y)
{
for(int i=0;i<k;i++)
{
int now=(x-1)*m+y-1;
if((n*m-now+1)/2<c[i].num)
return 0;
}
for(int i=0;i<k;i++)
if(c[i].num&&map[x-1][y]!=c[i].id&&map[x][y-1]!=c[i].id)
{
c[i].num--;
map[x][y]=c[i].id;
if(x==n&&y==m)return 1;
if(y<m)
{
if(dfs(x,y+1))
return 1;
}
else
{
if(dfs(x+1,1))
return 1;
}
c[i].num++;
}
return 0;
}
int main()
{
int res=1;
scanf("%d",&T);
while(T--)
{
memset(map,0,sizeof(map));
scanf("%d%d%d",&n,&m,&k);
for(int i=0;i<k;i++)
{
scanf("%d",&c[i].num);
c[i].id=i+1;
}
printf("Case #%d:\n",res++);
sort(c,c+k,cmp);
if(dfs(1,1))
{
printf("YES\n");
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
printf("%d%c",map[i][j],j==m?'\n':' ');
}
else printf("NO\n");
}
return 0;
}