AcWing 4519. 正方形数组的数目

原题链接

题目大意

题目规定任意两个相邻数相加都是完全平方数的排列称为正方形数组，问数组A的正方形排列的数目。

思路

因为这道题的数据范围很少，只有12，那么所有的情况最大就是12的阶乘，也就是 4e8 左右，所以需要剪枝优化，然后不难发现重复的数字会有大量的相同排列，根据这一点可以进行优化。优化原理：对排完序以后的序列只用不同数的第一个数

代码

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>

using namespace std;

const int N = 15;

int a[N];
bool st[N];
int ans, n;

bool check(int x)
{
    int t = sqrt(x);
    return x == t * t;
}

void dfs(int u, int last)
{
    if (u == n) 
    {
        ans ++;
        return;
    }
    
    for (int i = 0; i < n; i ++ )
    {
        if (i && !st[i - 1] && a[i] == a[i - 1]) continue;
        if (st[i]) continue;
        
        if (check(last + a[i]))
        {
            st[i] = true;
            dfs(u + 1, a[i]);
            st[i] = false;
        }
    }
}

int main()
{
    cin >> n;
    for (int i = 0; i < n; i ++ ) cin >> a[i];
    sort(a, a + n);
    
    for (int i = 0; i < n; i ++ )
    {
        if (i && a[i] == a[i - 1]) continue;
        st[i] = true;
        dfs(1, a[i]);
        st[i] = false;
    }
    
    cout << ans << '\n';
    return 0;
}

y总讲解