题目来源:https://www.luogu.org/problem/show?pid=1312
没什么可说的,就是dfs,细节很多,调试了好久才调出来。
重要的剪枝:
因为向右移动的优先级高于向左的,故除非所枚举的格子左侧为空,其他的一律为向右的。
由于是按优先级来搜的,一旦出结果,直接退出程序,因为此结果一定是最优的。
代码:
#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
using namespace std;
int a[5][8];
int n;
int ansx[6],ansy[6],ansp[6];
bool had()
{
bool vis[5][7];
memset(vis,0,sizeof(vis));
for(int i=0;i<5;i++)
{
int p=1,q=-1;
for(int j=0;j<7;j++)
{
if(a[i][j]==q)p++;
else
{
if(p>=3)for(int k=1;k<=p;k++)vis[i][j-k]=1;
p=1;q=a[i][j];
}
if(j==6)
{
if(p>=3)for(int k=0;k<p;k++)vis[i][j-k]=1;
break;
}
if(a[i][j]==0)break;
}
}
for(int j=0;j<7;j++)
{
int p=1,q=-1;
for(int i=0;i<5;i++)
{
if(a[i][j]==q)p++;
else
{
if(p>=3)for(int k=1;k<=p;k++)vis[i-k][j]=1;
p=1;q=a[i][j];
}
if(i==4)
{
if(p>=3)for(int k=0;k<p;k++)vis[i-k][j]=1;
break;
}
}
}
bool ok=0;
for(int i=0;i<5;i++)
{
for(int j=0;j<7;j++)
{
if(vis[i][j]&&a[i][j]!=0)
{
ok=1;
a[i][j]=0;
}
}
}
return ok;
}
void deal()
{
for(int i=0;i<5;i++)
{
for(int j=1;j<7;j++)
{
if(a[i][j]!=0&&a[i][j-1]==0)
{
int k=j;
while(k>=1&&a[i][k]!=0&&a[i][k-1]==0)
{
swap(a[i][k],a[i][k-1]);
k--;
}
}
}
}
}
void dfs(int s)
{
if(s==0)
{
for(int i=0;i<5;i++)if(a[i][0]!=0)return;
for(int i=n;i>=1;i--)printf("%d %d %d\n",ansx[i],ansy[i],ansp[i]);
exit(0);
}
int pos[5][7];
for(int i=0;i<5;i++)for(int j=0;j<7;j++)pos[i][j]=a[i][j];
for(int i=0;i<5;i++)
{
for(int j=0;j<7;j++)
{
if(a[i][j]==0)break;
else
{
if(i+1<5)
{
swap(a[i][j],a[i+1][j]);
deal();
while(had())deal();
ansx[s]=i;ansy[s]=j;ansp[s]=1;
dfs(s-1);
ansx[s]=ansy[s]=ansp[s]=0;
for(int k=0;k<5;k++)for(int t=0;t<7;t++)a[k][t]=pos[k][t];
}
if(i-1>=0&&a[i-1][j]==0)
{
swap(a[i][j],a[i-1][j]);
deal();
while(had())deal();
ansx[s]=i;ansy[s]=j;ansp[s]=-1;
dfs(s-1);
ansx[s]=ansy[s]=ansp[s]=0;
for(int k=0;k<5;k++)for(int t=0;t<7;t++)a[k][t]=pos[k][t];
}
}
}
}
}
int main()
{
scanf("%d",&n);
for(int i=0;i<5;i++)
{
for(int j=0;j<=7;j++)
{
scanf("%d",&a[i][j]);
if(a[i][j]==0)break;
}
}
dfs(n);
printf("-1");
return 0;
}