Fire Net HDU - 1045-（DFS）

最新推荐文章于 2021-08-19 13:58:09 发布

转载最新推荐文章于 2021-08-19 13:58:09 发布 · 123 阅读

博客给出一段代码，用于解决炮台放置问题。通过深度优先搜索（DFS）算法，判断在给定地图中炮台的可放置位置，避免同一行和列有炮台冲突，最终输出可放置炮台的最大数量。代码包含判断函数和 DFS 函数，在主函数中循环读取输入并调用 DFS 求解。

#include <stdio.h>
#include <cstring>

/*
   本题要注意坐标可以用一个数来表示，p, x = p / n; y = p % n;
*/

char map[4][4];
bool mark[4][4];
int n, ans;

bool Judge(int nx, int ny)
{
   if(mark[nx][ny] == 1)
       return false;
   //判断与（nx, ny）在同一行和同一列上的点
   for(int k = nx; k >= 0; k--)
   {
       if(map[k][ny] == 'X')
           break;
       if(mark[k][ny] == 1) //同一行上已经有炮台
           return false;
   }
   for(int k = ny; k >= 0; k--)
   {
       if(map[nx][k] == 'X')
           break;
       if(mark[nx][k] == 1) //同一列上已经有炮台
           return false;
   }
   return true;
}

void DFS(int p, int t)
{
   if(p == n*n)
   {
       if(ans < t)
           ans = t;
       return;
   }

   int x = p / n, y = p % n;
   if(map[x][y] == '.' && Judge(x, y))
   {
       mark[x][y] = 1;
       DFS(p+1, t+1);
       mark[x][y] = 0;
   }
   DFS(p+1, t); //在(x,y)处没有放炮台
}

int main()
{
   while(~scanf("%d", &n) && n != 0)
   {
       getchar();
       for(int i = 0; i < n; i++)
           scanf("%s", map[i]);
       memset(mark, 0, sizeof(mark));
       DFS(0,0);
       printf("%d\n", ans);
       ans = 0;
   }
   return 0;
}