寒假蓝桥杯训练一1014不同金字塔的数量（DFS+剪枝）

传送门：点击打开链接

Problem Description

小明最近喜欢搭数字积木，
一共有10块积木，每个积木上有一个数字，0~9。
搭积木规则：
每个积木放到其它两个积木的上面，并且一定比下面的两个积木数字小。
最后搭成4层的金字塔形，必须用完所有的积木。
下面是两种合格的搭法：
___0
__1_2
_3_4_5
6_7_8_9

___0
__3_1
_7_5_2
9_8_6_4
说明：数字之间的下划线只是为了对齐，没有实际意义
请你计算这样的搭法一共有多少种？
请填表示总数目的数字。
注意：你提交的应该是一个整数，不要填写任何多余的内容或说明性文字。

思路：像这种填数问题，从0-9选数来填的，不是回溯就是next_permutation

            10！可能有点多，所以我就进行了剪枝。（这个剪枝有点捞）

#include<iostream>
#include<cstring>
using namespace std;
int a[10];
int flag[10];
int cnt=0;
bool judge(int m)
{
	if(m==3)//这个时候已经回溯出 a[0] a[1] a[2] 判断 a[0]是否满足条件 
	{
		if(a[0]<a[1]&&a[0]<a[2])
			return true;
		return false;
	}
	if(m==6)//这个时候已经回溯出 a[0] a[1] a[2] a[3] a[4] a[5] a[0]肯定是满足条件的 紧接着判断a[1] a[2] 
	{
		if((a[1]<a[3]&&a[1]<a[4])&&(a[2]<a[4]&&a[2]<a[5]))
			return true;
		return false;
	}
	if(m==10) // 同上 不仅要判断，还需要计数 
	{
		if(a[3]<a[6]&&a[3]<a[7]&&(a[4]<a[7]&&a[4]<a[8])&&(a[5]<a[8]&&a[5]<a[9]))
		{
		cnt++;
		return true;
			
		}	
		return false;
	}
}
void dfs(int m)
{
	if(m==3||m==6||m==10)
	{
		if(judge(m)==0)
			return ;
	}
	if(m==10)
		return ;
	for(int i=0;i<10;i++)
	{
		if(flag[i]==0)
		{
			flag[i]=1;
			a[m]=i;
			dfs(m+1);
			flag[i]=0;
		}
	}
}
int main()
{
	memset(flag,0,sizeof(flag));
	dfs(0);
	cout<<cnt<<endl;
	return 0;
}