小猴编程周赛C++ | 2022年秋第1周T3平方序列

学习C++从娃娃抓起！记录下在学而思小猴编程学习过程中的题目，记录每一个瞬间。侵权即删，谢谢支持！

附上汇总贴：小猴编程C++ | 汇总-CSDN博客

---

【题目描述】

小X非常热爱信息学相关的比赛，他知道只是单纯的喜欢这门科目还远远不够，需要自己把更多的时间来投入到信息学中，并且他深刻的意识到在学习信奥的路上不能纸上谈兵、光听不练。因此为了准备今年的CSP-J/S，小X决定利用暑假的时间给自己制定一个高强度的刷题计划，但是在刷题的过程中遇到一个自己无法解决的题目：

给定一个长度为n的序列$A=a_1,a_2,\dots ,a_n$。对于序列中任意相邻的两个元素的和如果是完全平方数，那么称这个序列为平方序列。请问序列A所有的排列中有多少个排列是平方序列（注意：如果两个排列中所有位置上的元素都相同时，只能算作一个排列），并输出平方序列的个数。

你是小X的好朋友，同样也和小X一样热爱信息学奥赛，请你帮助小X完成这个题目。

【输入】

第一行一个整数n；

第二行n个整数$a_1,a_2,\dots ,a_n$。

【输出】

一行一个整数，表示序列A所有排列中平方序列的个数。

【输入样例】

3
3 1 15

【输出样例】

2

【代码详解】

#include <bits/stdc++.h>
using namespace std;
int n, a[15], cnt;
bool vis[15];
bool check(int v)
{
    int x = sqrt(v);
    return x * x == v;
}
void dfs(int step, int last)
{
    if (step>n) {
        cnt++;
        return;
    }
    for (int i=1; i<=n; i++) {
        if (vis[i]) continue;
        if (i>1 && a[i]==a[i-1] && !vis[i-1]) continue;  // 去重
        vis[i] = true;
        if (step<2) dfs(step+1, a[i]);
        else if (check(a[i]+last)) dfs(step+1, a[i]);  // 剪枝
        vis[i] = false;
    }
}

int main()
{
    cin >> n;
    for (int i=1; i<=n; i++) cin >> a[i];
    sort(a+1, a+n+1);
    dfs(1, 0);
    cout << cnt << endl;
    return 0;
}

【运行结果】

3 
3 1 15 
2