第七届蓝桥杯决赛真题 凑平方数

把0~9这10个数字，分成多个组，每个组恰好是一个平方数，这是能够办到的。
比如：0, 36, 5948721

再比如：
1098524736
1, 25, 6390784
0, 4, 289, 15376
等等…

注意，0可以作为独立的数字，但不能作为多位数字的开始。
分组时，必须用完所有的数字，不能重复，不能遗漏。

如果不计较小组内数据的先后顺序，请问有多少种不同的分组方案？

注意：需要提交的是一个整数，不要填写多余内容。

思路:全排列+dfs

dfs按升序记录结果(pre记录上一个数) 出现0的情况表示0在第0位，否则不合法(升序);

#include<bits/stdc++.h>
using namespace std;
int res = 0;
unordered_map<long long,int> mp;
int a[10] = {0,1,2,3,4,5,6,7,8,9};
void dfs(int pre,int s)
{
	if(s==10)
	{
		res++;
		return;
	}
	if(a[s]==0)
	{
		if(s==0) dfs(0,s+1);
		return;
	}
	long long sum = 0;
	for(int i=s;i<10;i++)
	{
		sum = sum*10+a[i];
		if(sum>pre&&mp[sum])	dfs(sum,i+1);
	}
}
int main()
{
	for(long long i=0;i<=500000;i++)
		mp[i*i] = 1;
	do{
		dfs(-1,0);
	}while(next_permutation(a,a+10));	
	cout<<res;
}