势均力敌(C++ 三级题--使用vector和push_back)

势均力敌
用n(>2)个不同的个位数字组成一个n位数,显然有 n!个不同的结果。
可以证明,这 n!个数字可以被分为势均力敌的两组 -- 即平方和相等、且个数也相等的两组。
本题就请你用程序验证一下这个结论。
因为本题是一道简单题,所以规模很小,只考虑 n≤4 的情况。

输入
输入第一行给出正整数 n(2<n≤4),
随后一行给出n个不同的、在区间[1,9]内的个位数字,其间以空格分隔。

输出
将所有组成的 n! 个不同的 n 位数分为平方和相等、且个数也相等的两组。
但你只需要输出其中组就可以了。每个数字占一行,共输出 n!/2 行。 
注意:解可能不唯一,输出任何一组解就可以。

样例输入: 
3
5 2 1


样例输出:
125
512
251

#include <bits/stdc++.h>
using namespace std;

int a[10005];
vector<int> v;
long long num = 0, n;
long long cnt, flag;
bool st[10005], stt[10005];
long long pa[10005];

void dfs(int x)
{
    if (x == n) v.push_back(num);

    for (int i = 0; i < n; i++)
    {
        if (!st[a[i]])
        {
            int fa = num;
            st[a[i]] = true;
            num = num * 10 + a[i];
            dfs(x + 1);
            num = fa;
            st[a[i]] = false;
        }
    }
}

void dfs1(int x, int p)
{
    if (flag == 1) return;
    if (x == (int)v.size() / 2)
    {
        num = 0;
        for (int i = 0; i < x; i++) num += pa[i] * pa[i];
        if (num * 2 == cnt)
        {
            for (int i = 0; i < x; i++) cout << pa[i] << '\n';
            flag = 1;
            return;
        }
    }

    for (int i = p; i < (int)v.size(); i++)
    {
        if (!stt[i])
        {
            stt[i] = true;
            pa[x] = v[i];
            dfs1(x + 1, i + 1);
            stt[i] = false;
            pa[x] = 0;
        }
    }
}

int main()
{
    cin >> n;

    for (int i = 0; i < n; i++) cin >> a[i];

    dfs(0);

    for (auto t : v) cnt += t * t;

    dfs1(0, 0);
}

上一篇:Scala语言的软件工程