蓝桥杯-随意组合

随意组合

小明被绑架到X星球的巫师W那里。

其时，W正在玩弄两组数据 (2 3 5 8) 和 (1 4 6 7)
他命令小明从一组数据中分别取数与另一组中的数配对，共配成4对（组中的每个数必被用到）。
小明的配法是：{(8,7),(5,6),(3,4),(2,1)}

巫师凝视片刻，突然说这个配法太棒了！

因为：
每个配对中的数字组成两位数，求平方和，无论正倒，居然相等：
87^2 + 56^2 + 34^2 + 21^2 = 12302
78^2 + 65^2 + 43^2 + 12^2 = 12302

小明想了想说：“这有什么奇怪呢，我们地球人都知道，随便配配也可以啊！”
{(8,6),(5,4),(3,1),(2,7)}

86^2 + 54^2 + 31^2 + 27^2 = 12002
68^2 + 45^2 + 13^2 + 72^2 = 12002

巫师顿时凌乱了。。。。。

请你计算一下，包括上边给出的两种配法，巫师的两组数据一共有多少种配对方案具有该特征。
配对方案计数时，不考虑配对的出现次序。
就是说：
{(8,7),(5,6),(3,4),(2,1)}
与
{(5,6),(8,7),(3,4),(2,1)}
是同一种方案。

注意：需要提交的是一个整数，不要填写任何多余内容（比如，解释说明文字等）

经验技巧：
凡是阶乘组合方案，均想到用dfs枚举组合。
如n!,432*1想到用dfs枚举方案。
因为，dfs每次会book标记一下，下一次就少一次，正好构成阶乘组合方案。

#include<bits/stdc++.h>
using namespace std;
int ans,num1,num2,book[10];
int a[5]={2,3,5,8};
int b[5]={1,4,6,7};
map<int,int> mp;
void dfs(int step)
{
    if(step==4)
    {
        num1=num2=0;
        for(int i=0;i<4;i++)
        {
            num1+=(a[i]*10+mp[a[i]])*(a[i]*10+mp[a[i]]);
            num2+=(mp[a[i]]*10+a[i])*(mp[a[i]]*10+a[i]);
        }
        cout << num1 << " " <<num2<<endl;
        if(num1==num2)
            ans++;

        return ;
    }
    for(int i=0;i<4;i++){
        if(book[i]==0)
        {
            book[i]=1;
            mp[a[step]]=b[i];
            dfs(step+1);
            book[i]=0;
        }
    }
}
int main()
{
    dfs(0);
    cout << ans << endl;
    return 0;
}