题目大意:一个N*M的地图哟,有好多烦人的黑块块。这些玩意可以遮光。
然后我们要把好多啊好多啊灯上去,使得用最少的灯能把整个地图照亮。
还有一个要求,就是砖块上有一个值,要让上下左右四个方向的灯的数量加起来等于那个值。
思路:用DFS枚举每一个点,能放就放,当然要面临着剪枝的问题。(我将黑块用 -1 表示 放灯用 -2 表示)
我这里用了一个OPLIGHT函数,当传入的FLAG是TRUE时,就执行开灯操作,把能找到的地方的地图上全部+1,
反之就把地图-1.。
剪枝:当地图上点不为零的时候,就代表这个点没有光照到,而且也不是什么黑块或是已经有灯了,直接跳过。
当你搜索的当前节点是一个黑块的时候,我们就去寻找黑块上方有没有未被照亮的点,如果还有的话,那么之后不会再有等可以将它照亮。所以就RETUEN了~
#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <cmath>
#define INF 0x3f3f3f3f
int n,m,b;
int min=INF;
int map[9][9];
int dx[] = {-1,1,0,0};
int dy[] = {0,0,-1,1};
struct light
{
int x,y;
int w;
}l[50];
void debug()
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
printf("%d ",map[i][j]);
putchar(10);
}
putchar(10);
}
bool is_ok()
{
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
if(map[i][j]==0)return false;
for(int i=1;i<=b;i++)
{
if(l[i].w==-1)continue;
int cnt=0;
for(int j=0;j<4;j++)
if(map[l[i].x+dx[j]][l[i].y+dy[j]]==-2)cnt++;
if(cnt!=l[i].w)return false;
}
return true;
}
void oplight(int lx,int ly,bool flag)
{
for(int i=lx+1;i<=n;i++)
{
if(map[i][ly]==-2)continue;
if(map[i][ly]==-1)break;
if(flag)map[i][ly]++;
else map[i][ly]--;
}
for(int i=lx-1;i>=1;i--)
{
if(map[i][ly]==-2)continue;
if(map[i][ly]==-1)break;
if(flag)map[i][ly]++;
else map[i][ly]--;
}
for(int i=ly+1;i<=m;i++)
{
if(map[lx][i]==-2)continue;
if(map[lx][i]==-1)break;
if(flag)map[lx][i]++;
else map[lx][i]--;
}
for(int i=ly-1;i>=1;i--)
{
if(map[lx][i]==-2)continue;
if(map[lx][i]==-1)break;
if(flag)map[lx][i]++;
else map[lx][i]--;
}
}
void dfs(int posx,int posy,int cnt)
{
if(is_ok())
{
if(cnt<min)min=cnt;
return;
}
for(int s=posx;s<=n;s++)
{
for(int j=(s==posx ? posy:1);j<=m;j++)
{
if(is_ok())
{
if(cnt<min)min=cnt;
return;
}
if(map[s][j]==-1)
{
for(int k=1;k<s;k++)
if(map[k][j]==0)return;
}
if(map[s][j]==0)
{
map[s][j]=-2;
oplight(s,j,true);
dfs(s,j,cnt+1);
oplight(s,j,false);
map[s][j]=0;
}
}
}
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF && n && m)
{
memset(map,0,sizeof(map));
min=INF;
scanf("%d",&b);
for(int i=1;i<=b;i++)
{
int a,d,c;
scanf("%d%d%d",&a,&d,&c);
l[i].x=a;
l[i].y=d;
l[i].w=c;
map[a][d]=-1;
}
dfs(1,1,0);
if(min==INF){printf("No solution\n");continue;}
printf("%d\n",min);
}
}