2016年第七届 蓝桥杯 全国总决赛B题(完全平方数)

题目意思就是：
给你0，1，2，3，4，5，6，7，8，9十个数字，要你选出任意一个或几个组合在一起成为完全平方数，每个数字都必须选且只能选一次，求可能的方案。

比如有其中几种符合题意的情况:
0 16 25 73984
0 1 625 73984
0 4 16 537289
0 16 784 5329
0 25 784 1936

ans=300;

#include<iostream>using namespace std;#include<cstring>#include<cstdio>#include<vector>#include<algorithm>#include<cmath>typedef long long ll;const ll inf=(ll)(0x3f3f3f3f);int ans=0;ll t[105];vector<ll> vec[500];int len=0;ll b[15];int lens[600];const int maxn=900005;ll pri[maxn];bool judge(ll x) {    ll o1=(ll)(sqrt(x));    return o1*o1==x;}void solve(ll *a,int num) {    memset(b,inf,sizeof(b));    int u=0,pos=0;    bool is=true;    for(int i=0; i<num; ++i) {        ll s=0;        if(!a[pos]&&t[i]>1) {            is=false;            break;        }        for(int j=pos; j<t[i]+pos; ++j) {            s=s*10+a[j];        }        b[u]=s;        ++u;        pos=t[i]+pos;    }    if(is) {        sort(b,b+u);        bool issame=true;        for(int i=0; i<len; ++i) {            int sj=lens[i];            if(sj!=u)continue;            bool same1=true;            for(int j=sj-1; j>=0; --j) {                if(vec[i][j]!=b[j]) {                    same1=false;                    break;                }            }            if(same1) {                issame=false;                break;            }        }//判重        if(issame) {            ++len;            lens[len-1]=u;            for(int i=0; i<u; ++i) {                vec[len-1].push_back(b[i]);//                cout<<b[i]<<" ";            }//            cout<<endl;            ++ans;        }    }}ll ppp=(ll)maxn;void dfs(ll *a,int sum,int num,int pos) {    if(!sum) {        solve(a,num);    } else if(sum<0);    else {        for(int i=sum; i>=1; --i) {//求划分数            t[num]=i;            ll ts=0;            bool can=true;            if(i>1&&a[pos]==0)continue;            for(int j=pos; j<pos+i&&can; ++j) {                ts=ts*10+a[j];            }            if((ts<ppp&&(!pri[ts]))||!judge(ts))continue;//剪枝            dfs(a,sum-i,num+1,pos+i);        }    }}int main() {    ll a[]= {0,1,2,3,4,5,6,7,8,9};    for(int i=0; i<500; ++i)        vec[i].clear();    for(ll i=0; i<ppp; ++i) {        pri[i]=judge(i);    }    int op=0;    do {        dfs(a,10,0,0);    } while(next_permutation(a,a+10));    cout<<ans<<endl;    return 0;}