随意组合 dfs

小明被绑架到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)}
是同一种方案。
注意:需要提交的是一个整数,不要填写任何多余内容(比如,解释说明文字等)

 

 

#include <iostream>
using namespace std;
int a[2][4] = { 2,3,5,8,1,4,6,7 };
//2 3 5 8
//1 4 6 7
int b[2][4];
int vis[4] = { 0 };
int sum = 0;
int verify()
{
    return b[0][0] * b[0][0] + b[0][1] * b[0][1] + b[0][2] * b[0][2] + b[0][3] * b[0][3] == b[1][0] * b[1][0] + b[1][1] * b[1][1] + b[1][2] * b[1][2] + b[1][3] * b[1][3];
}
void dfs(int n)
{
    if (n == 4)
    {
        if (verify())
            sum++;
        return;
    }
    for (int i = 0; i <= 3; i++)
    {
        if (!vis[i])
        {
            //cout << n << " " << i <<endl;
            b[0][n] = a[0][n] * 10 + a[1][i];
            b[1][n] = a[1][i] * 10 + a[0][n];
            vis[i] = 1;
            dfs(n + 1);
            vis[i] = 0;
        }
    }
    return;
}
int main()
{
    dfs(0);
    cout << sum;
    return 0;
}

 

posted @ 2018-11-09 08:45  uzi2008  阅读(263)  评论(0编辑  收藏  举报