我有一个N×M的网格,每个单元格都被染上了一种颜色。
当玩家点击网格中任何一个颜色为α的单元格时,位于网格左上角且颜色为β的单元格将接收颜色α,但不仅如此:所有通过只使用颜色α或β的路径与源相连的单元格也将接收颜色α。
连接单元格应仅考虑水平和垂直方向以形成路径。例如,当玩家单击左侧图中突出显示的单元格时,网格将获得右侧图中的着色。游戏的目标是使网格单色。
当玩家点击网格中任何一个颜色为α的单元格时,位于网格左上角且颜色为β的单元格将接收颜色α,但不仅如此:所有通过只使用颜色α或β的路径与源相连的单元格也将接收颜色α。
连接单元格应仅考虑水平和垂直方向以形成路径。例如,当玩家单击左侧图中突出显示的单元格时,网格将获得右侧图中的着色。游戏的目标是使网格单色。
输入描述
输入的第一行包括2个整数N和M(1 ≤ N ≤ 4,1 ≤ M ≤ 5),分别表示网格的行数和列数。接下来的N行描述了网格的初始配置,用0到9之间的整数表示每种颜色。输入不包含任何其他行。
输出描述
打印一行包含单个整数的内容,该整数表示玩家必须点击的最小次数,以使网格单色。
示例输入
1:
4 5
01234
34567
67890
901232:
4 5
01234
12345
23456
345673:
4 5
00162
30295
45033
01837
输出示例
1:
12
2:
7
3:
10
我正在尝试使用回溯法来解决这个问题(因为时间限制为8秒,而且网格的大小很小)。但是它会超时。有些人只用了0秒就解决了。
还有其他算法可以解决这个问题吗?
#include <stdio.h>
#include <string.h>
#define MAX 5
#define INF 999999999
typedef int signed_integer;
signed_integer n,m,mink;
bool vst[MAX][MAX];
signed_integer flood_path[4][2] = {
{-1,0},
{1,0},
{0,1},
{0,-1}
};
//flood and paint all possible cells... the root is (i,j)
signed_integer flood_and_paint(signed_integer cur_grid[MAX][MAX],signed_integer i, signed_integer j, signed_integer beta, signed_integer alpha, signed_integer colors[]){
//invalid cell
if (vst[i][j] || i < 0 || i >= n || j < 0 || j >= m)
return 0;
//mark existent colors
colors[cur_grid[i][j]] = 1;
//only alpha and beta colors counts
if (cur_grid[i][j] != beta && cur_grid[i][j] != alpha)
return 0;
//mark (i,j) as visited and change its color
vst[i][j] = true;
cur_grid[i][j] = alpha;
//floodit !
signed_integer ret = 1;
for (signed_integer k = 0; k < 4; k++)
ret += flood_and_paint(cur_grid,i + flood_path[k][0], j + flood_path[k][1], beta, alpha, colors);
//how many cells change
return ret;
}
void backtrack(signed_integer cur_grid[MAX][MAX],signed_integer k,signed_integer _cont, signed_integer alpha) {
//bigger number of clicks for this solution ? ... getting back
if(k >= mink)
return;
signed_integer colors[10];
memset(vst, false, sizeof(vst));
memset(colors, 0, sizeof(colors));
signed_integer beta = cur_grid[0][0];
signed_integer cont = flood_and_paint(cur_grid, 0, 0, beta, alpha, colors);
//there are alpha colors to change and no beta colors to change
colors[alpha] = 1;
colors[beta] = 0;
//all squares on same color
if (cont == n * m) {
mink = k;
return;
}
//this solution is equals to another ? ... getting back
if (cont == _cont)
return;
++k;//new click
//copy this matrix and backtrack
signed_integer copy[MAX][MAX];
for (signed_integer c = 0; c < 10; ++c){
if (colors[c] && c != cur_grid[0][0]) {
memcpy(copy, cur_grid,n*m*sizeof(signed_integer));
backtrack(copy,k,cont,c);
}
}
}
void cleanBuffer(){
while (getchar() != '\n');
}
int main(void) {
signed_integer grid[MAX][MAX];
scanf("%d %d",&n,&m);
for (signed_integer i = 0; i < n; ++i) {
cleanBuffer();
for (signed_integer j = 0; j < m; ++j){
grid[i][j] = getchar() - '0';
}
}
mink = INF;
backtrack(grid,0, 0, grid[0][0]);
printf("%d\n",mink);
return 0;
}
memset
调用。同样也可以在每个递归调用之前消除memcpy
。 - IVlad