蓝桥杯 dfs 平方组合数
把0~9这10个数字,分成多个组,每个组恰好是一个平方数,这是能够办到的。
比如:0, 36, 5948721
再比如:
1098524736
1, 25, 6390784
0, 4, 289, 15376
等等…
注意,0可以作为独立的数字,但不能作为多位数字的开始。
分组时,必须用完所有的数字,不能重复,不能遗漏。
如果不计较小组内数据的先后顺序,请问有多少种不同的分组方案?
注意:需要提交的是一个整数,不要填写多余内容。
在每个组之间加个+,用set去重,先要排序,因为跟顺序是无关的。
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<vector>
#include<queue>
#include<string>
#include<set>
#define IL inline
#define x first
#define y second
typedef long long ll;
using namespace std;
const int N=10;
bool st[N];
vector<ll>vv;
ll cnt;
int a[N];
bool stt[1000000];
set<string>s;
bool judge(ll x)
{
if(x==0 ) return 1;
ll t=(ll)sqrt(x);
return t*t==x;
}
void dfs(int u,ll n)
{
if(u==10)
{
if(n) //最后一层还存在别的数也是不行的
return ;
vector<ll>vv2=vv;
sort(vv2.begin(),vv2.end());
string ss="";
for(int i=0;i<vv2.size();i++)
{
// cout<<vv[i]<<endl;
string t=to_string(vv2[i]);
ss+=t;
ss+="+";
}
bool flag=0;
for(int i=0;i<ss.size();i++)
if(ss[i]=='0') //没有0的话是不行的
flag=true;
if(!flag)
return ;
s.insert(ss);
return ;
}
ll x=n*10+a[u];
if(judge(x))
{
vector<ll>vvv=vv;
vv.push_back(x);
dfs(u+1,0);
vv.pop_back();
}
dfs(u+1,x);
}
int main()
{
for(int i=0;i<=9;i++)
a[i]=i;
do
{
dfs(0,0);
vv.clear();
}while(next_permutation(a,a+N));
cout<<s.size()<<endl;
return 0;
}