Part 3.4 搜索的剪枝

对于一些不必要搜索的部分，我们可以避免访问这些状态，从而提高搜索效率。

小木棍

题目背景

本题不保证存在可以通过满足本题数据范围的任意数据做法。可以通过此题的程序不一定完全正确（算法时间复杂度错误、或不保证正确性）

本题为搜索题，本题不接受 hack 数据。关于此类题目的详细内容

题目描述

乔治有一些同样长的小木棍，他把这些木棍随意砍成几段，直到每段的长都不超过 $50$。

现在，他想把小木棍拼接成原来的样子，但是却忘记了自己开始时有多少根木棍和它们的长度。

给出每段小木棍的长度，编程帮他找出原始木棍的最小可能长度。

输入格式

第一行是一个整数 $n$，表示小木棍的个数。
第二行有 $n$ 个整数，表示各个木棍的长度 $a_i$。

输出格式

输出一行一个整数表示答案。

样例 #1

样例输入 #1

9
5 2 1 5 2 1 5 2 1

样例输出 #1

6

提示

对于全部测试点，$1\le n\le 65$，$1\le a_i\le 50$。

代码实现

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
#define MAX_N 65
bool cmp(int a,int b)
{
	return a>b;
}
int arr[MAX_N+5],vis[MAX_N+5];
int n,cnt,ind=0;
int need;
int maxx=0,minx=10000,sum=0;
bool dfs(int stick,int now,int last)//当前拼好的数量，当前长度，下一个位置 
{
	if(stick==cnt)return true;//全部拼好 
	if(now==need)return dfs(stick+1,0,1);//当前的拼好了 
	int fail=0;//记录不可行的长度 
	for(int i=last;i<=ind;i++)
	{
		if(!vis[i]&&now+arr[i]<=need&&arr[i]!=fail)
		{
			vis[i]=1;
			if(dfs(stick,now+arr[i],i+1))
			return true;
			fail=arr[i];
			vis[i]=0;
			//第一个位置都不能放，意味着当前情况下此位置不能选择，直接退出
			//当前和为need且不能放，意味着再往后找也不能放 
			if(now==0||now+arr[i]==need)return false; 
		}
	}
	return false;
}
int main()
{
	cin>>n;
	for(int i=1,x;i<=n;i++)
	{
		cin>>x;
		if(x>50)continue;
		arr[++ind]=x;
		maxx=max(maxx,x);
		sum+=x;
	}
	int temp=sum/2;
	sort(arr+1,arr+1+n,cmp);//从大到小排序 
	int flag=0;
	for(int i=maxx;i<=temp;i++)
	{
		if(sum%i)continue;
		cnt=sum/i;
		need=i;
		memset(vis,0,sizeof vis);
		if(dfs(0,0,1))flag=1;//第一个可行的长度为答案 
		if(flag)
		{
			cout<<i;
			return 0;
		}
	}
	cout<<sum;
	return 0;
 } 

[NOIP2011 提高组] Mayan 游戏

题目描述

Mayan puzzle 是最近流行起来的一个游戏。游戏界面是一个$7$ 行 $\times 5$ 列的棋盘，上面堆放着一些方块，方块不能悬空堆放，即方块必须放在最下面一行，或者放在其他方块之上。游戏通关是指在规定的步数内消除所有的方块，消除方块的规则如下：

1. 每步移动可以且仅可以沿横向（即向左或向右）拖动某一方块一格：当拖动这一方块时，如果拖动后到达的位置（以下称目标位置）也有方块，那么这两个方块将交换位置（参见输入输出样例说明中的图 $6$ 到图 $7$）；如果目标位置上没有方块，那么被拖动的方块将从原来的竖列中抽出，并从目标位置上掉落（直到不悬空，参见下面图 $1$ 和图 $2$）；

2. 任一时刻，如果在一横行或者竖列上有连续三个或者三个以上相同颜色的方块，则它们将立即被消除（参见图1 到图3）。

注意：

a) 如果同时有多组方块满足消除条件，几组方块会同时被消除（例如下面图 $4$，三个颜色为 $1$ 的方块和三个颜色为 $2$ 的方块会同时被消除，最后剩下一个颜色为 $2$ 的方块）。

b) 当出现行和列都满足消除条件且行列共享某个方块时，行和列上满足消除条件的所有方块会被同时消除（例如下面图5 所示的情形，$5$ 个方块会同时被消除）。

3. 方块消除之后，消除位置之上的方块将掉落，掉落后可能会引起新的方块消除。注意：掉落的过程中将不会有方块的消除。

上面图 $1$ 到图 $3$ 给出了在棋盘上移动一块方块之后棋盘的变化。棋盘的左下角方块的坐标为 $(0,0)$，将位于 $(3,3)$ 的方块向左移动之后，游戏界面从图 $1$ 变成图 $2$ 所示的状态，此时在一竖列上有连续三块颜色为 $4$ 的方块，满足消除条件，消除连续 $3$ 块颜色为 $4$ 的方块后，上方的颜色为 $3$ 的方块掉落，形成图 $3$ 所示的局面。

输入格式

共 $6$ 行。

第一行为一个正整数 $n$，表示要求游戏通关的步数。

接下来的 $5$ 行，描述 $7\times 5$ 的游戏界面。每行若干个整数，每两个整数之间用一个空格隔开，每行以一个 $0$ 结束，自下向上表示每竖列方块的颜色编号（颜色不多于 $10$ 种，从 $1$ 开始顺序编号，相同数字表示相同颜色）。

输入数据保证初始棋盘中没有可以消除的方块。

输出格式

如果有解决方案，输出 $n$ 行，每行包含 $3$ 个整数 $x,y,g$，表示一次移动，每两个整数之间用一个空格隔开，其中 $(x,y)$ 表示要移动的方块的坐标，$g$ 表示移动的方向，$1$ 表示向右移动，$-1$ 表示向左移动。注意：多组解时，按照 $x$ 为第一关键字，$y$ 为第二关键字，$1$ 优先于 $-1$，给出一组字典序最小的解。游戏界面左下角的坐标为 $(0,0)$。

如果没有解决方案，输出一行 -1。

样例 #1

样例输入 #1

3
1 0
2 1 0
2 3 4 0
3 1 0
2 4 3 4 0

样例输出 #1

2 1 1
3 1 1
3 0 1

提示

【输入输出样例说明】

按箭头方向的顺序分别为图 $6$ 到图 $11$

样例输入的游戏局面如上面第一个图片所示，依次移动的三步是：$(2,1)$ 处的方格向右移动，$(3,1)$ 处的方格向右移动，$(3,0)$ 处的方格向右移动，最后可以将棋盘上所有方块消除。

【数据范围】

对于 $30\%$ 的数据，初始棋盘上的方块都在棋盘的最下面一行；

对于 $100\%$ 的数据，$0<n\le 5$。

代码实现

#include<iostream>
#include<cstring>
using namespace std;
int n;
int num[10][10];//记录当前局面 
int last[10][10][10];//方便回溯 
int ans[10][5]; //记录答案 
int rm[10][10];//记录要消除的位置 
void output(int x)
{
	for(int i=1;i<x;i++)
	{
		cout<<ans[i][1]<<" "<<ans[i][2]<<" "<<ans[i][3]<<endl;
	}
	return ;
}
void copy(int x,int flag)
{
	if(!flag)
	for(int i=1;i<=5;i++)
	for(int j=1;j<=7;j++)
	last[x][i][j]=num[i][j];
	else
	for(int i=1;i<=5;i++)
	for(int j=1;j<=7;j++)
	num[i][j]=last[x][i][j];
	return ;
}
void update()
{
	for(int i=1;i<=5;i++)
	{
		int l=0;
		for(int j=1;j<=7;j++)
		{
			if(num[i][j]==0)
			{
				l++;
				continue;
			}
			if(!l)continue;
			num[i][j-l]=num[i][j];
			num[i][j]=0;
		}
	}
	return ;
}
bool remove()
{
	int flag=0;
	for(int i=1;i<=5;i++)
	{
		for(int j=1;j<=7;j++)
		{
			if(i>1&&i<5&&num[i-1][j]==num[i][j]&&num[i][j]==num[i+1][j]&&num[i][j])
			flag=1,rm[i-1][j]=1,rm[i][j]=1,rm[i+1][j]=1;
			if(j>1&&j<7&&num[i][j-1]==num[i][j]&&num[i][j]==num[i][j+1]&&num[i][j])
			flag=1,rm[i][j-1]=1,rm[i][j]=1,rm[i][j+1]=1;
		}
	}
	if(!flag)return 0;
	for(int i=1;i<=5;i++)
	for(int j=1;j<=7;j++)
	{
		if(!rm[i][j])continue;
		num[i][j]=0;
		rm[i][j]=0;
	}
	return 1;
}
void move(int i,int j,int dir)
{
	int temp=num[i][j];
	num[i][j]=num[i+dir][j];
	num[i+dir][j]=temp;
	update();
	while(remove())update();
	return ;
}
bool check()
{
	for(int i=1;i<=5;i++)
	if(num[i][1])return 0;
	return 1;
}
void dfs(int x)
{
	if(check())
	{
		output(x);
		exit(0);
	}
	if(x==n+1)return ;
	copy(x,0);//num to last
	for(int i=1;i<=5;i++)
	{
		for(int j=1;j<=7;j++)
		{
			if(!num[i][j])continue;
			if(i<5)
			{
				move(i,j,1);
				ans[x][1]=i-1,ans[x][2]=j-1,ans[x][3]=1;				
				dfs(x+1);
				copy(x,1);//last to num
			}
			if(i>1)
			{
				move(i,j,-1);
				ans[x][1]=i-1,ans[x][2]=j-1,ans[x][3]=-1;
				dfs(x+1);
				copy(x,1);
			}
		}
	}
}
int main()
{
	cin>>n;
	memset(num,0,sizeof num);
	for(int i=1;i<=5;i++)
	{
		for(int j=1;j<=8;j++)
		{
			cin>>num[i][j];
			if(num[i][j]==0)break;
		}
	}
	dfs(1);
	cout<<-1;
	return 0;
 }