算法1
DFS加等效冗余剪枝(顺便吐槽,1个int敲成了bool调了快一小时屮屮)
C++ 代码
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 10;
int n;
int g[5][7];
int st[5][7];
struct op{
int x, y, t;
}path[N], res[N];
int idx;
bool is_leaf()
{
bool tag = true;
for(int i = 0; i < 5; ++ i)
if(g[i][0])
tag = false;
return tag;
}
bool check_line(int x, int y)
{
int len = 1;
for(int i = 1; x - i >= 0; ++ i)
if(g[x - i][y] == g[x][y]) ++ len; else break;
for(int i = 1; x + i < 5; ++ i)
if(g[x + i][y] == g[x][y]) ++ len; else break;
return len >= 3;
}
void set_line(int x, int y)
{
st[x][y] = 1;
for(int i = 1; x - i >= 0; ++ i)
if(g[x - i][y] == g[x][y]) st[x - i][y] = 1; else break;
for(int i = 1;x + i < 5; ++ i)
if(g[x + i][y] == g[x][y]) st[x + i][y] = 1; else break;
}
bool check_col(int x, int y)
{
int len = 1;
for(int i = 1; y - i >= 0; ++ i)
if(g[x][y - i] == g[x][y]) ++ len; else break;
for(int i = 1; y + i < 7; ++ i)
if(g[x][y + i] == g[x][y]) ++ len; else break;
return len >= 3;
}
void set_col(int x, int y)
{
st[x][y] = 1;
for(int i = 1; y - i >= 0; ++ i)
if(g[x][y - i] == g[x][y]) st[x][y - i] = 1; else break;
for(int i = 1; y + i < 7; ++ i)
if(g[x][y + i] == g[x][y]) st[x][y + i] = 1; else break;
}
void mark()
{
memset(st, 0, sizeof st);
for(int i = 0; i < 5; ++ i)
for(int j = 0; j < 7; ++ j)
{
if(!g[i][j]) continue;
if(check_line(i, j)) set_line(i, j);
if(check_col(i, j)) set_col(i, j);
}
}
int remove()
{
int cnt = 0;
for(int i = 0; i < 5; ++ i)
for(int j = 0; j < 7; ++ j)
if(st[i][j])
g[i][j] = 0, ++ cnt;
return cnt;
}
void processing(int arr[])
{
int i = 0, j = 0;
while(i < 7 && arr[i]) ++ i, ++ j;
while(j < 7)
{
if(arr[j]) swap(arr[i], arr[j]), ++ i, ++ j;
else ++ j;
}
}
void drop()
{
for(int i = 0; i < 5; ++ i)
processing(g[i]);
}
void update()//1 右移 -1 左移
{
mark();
if(!remove()) return;
drop();
update();
}
bool dfs(int x, int y, int t, int k)
{
int bg[5][7];
memcpy(bg, g, sizeof g);
if(k > n) return false;
if(x + t < 0 || x + t >= 5) return false;
swap(g[x][y], g[x + t][y]);
drop();//!!!!!!
update();
path[idx ++ ] = {x, y, t};
if(is_leaf() && k == n)
{
memcpy(res, path, sizeof res);
memcpy(g, bg, sizeof g);
-- idx;
return true;
}
for(int i = 0; i < 5; ++ i)
for(int j = 0; j < 7; ++ j)
if(g[i][j])
{
if(dfs(i, j, 1, k + 1))
{
memcpy(g, bg, sizeof bg);
-- idx;
return true;
}
if(i > 0 && !g[i - 1][j])
if(dfs(i, j, -1, k + 1))
{
memcpy(g, bg, sizeof bg);
-- idx;
return true;
}
}
memcpy(g, bg, sizeof bg);
-- idx;
return false;
}
int main()
{
cin >> n;
for(int i = 0; i < 5; ++ i)
{
for(int j = 0; j < 8; ++ j)
{
int x;
cin >> x;
if(!x) break;
g[i][j] = x;
}
}
bool tag = false;
for(int i = 0; i < 5; ++ i)
{
for(int j = 0; j < 7; ++ j)
{
if(g[i][j])
{
if(dfs(i, j, 1, 1)){ tag = true; break; }
if(i > 0 && !g[i - 1][j])
if(dfs(i, j, -1, 1))
{ tag = true; break; }
}
}
if(tag) break;
}
if(tag)
{
for(int i = 0; i < n; ++ i)
printf("%d %d %d\n", res[i].x, res[i].y, res[i].t);
}
else printf("%d", -1);
return 0;
}