正方形数组的数目

问题描述 :
给定一个非负整数数组 A，如果该数组每对相邻元素之和是一个完全平方数，则称这一数组为正方形数组。返回 A 的正方形排列的数目。两个排列 A1 和 A2 不同的充要条件是存在某个索引 i，使得 A1[i] != A2[i]。

示例 1：
输入：[1,17,8]
输出：2
解释：
[1,8,17] 和 [17,8,1] 都是有效的排列。

示例 2：
输入：[2,2,2]
输出：1

说明：若一个数能表示成某个整数的平方的形式，则称这个数为完全平方数。完全平方数是非负数。

输入说明 :
首先输入A数组的长度n，
然后输入n个整数，以空格分隔。
1 <= n<= 12
0 <= A[i] <= 1e9

输出说明 :
输出一个整数

输入范例 :
3
1 17 8

输出范例：
2

#include<iostream>
#include<vector>
#include<map>
#include<math.h>
using namespace std;

bool isSquare(int n){
	int t=sqrt(n);
	return t*t==n;
}

void dfs(vector<int> &nodes,vector<vector<int>> &g,map<int,int> &counts,int i,int m,int M,int &res){
	if(m==M){
		res++;
		return ;
	}
	for(auto j:g[i]){
		if(counts[nodes[j]]>0){
		    counts[nodes[j]]--;
			dfs(nodes,g,counts,j,m+1,M,res);
			counts[nodes[j]]++;	
		}
	}
} 
int numSquarefulPerms(vector<int> &nums){
	map<int,int> counts;
	for(auto x:nums){
		counts[x]++;
	}
	vector<int> nodes;
	for(auto p:counts){
		nodes.push_back(p.first);
	}
	int N=nodes.size();
	vector<vector<int>> g(N);
	for(int i=0;i<N;i++){
		if(counts[nodes[i]]>1&&isSquare(nodes[i]*2)){
			g[i].push_back(i);
		}
		for(int j=i+1;j<N;j++){
			if(isSquare(nodes[i]+nodes[j])){
				g[i].push_back(j);
				g[j].push_back(i);
			}
		}
	}
	int res=0;
	for(int i=0;i<N;i++){
		counts[nodes[i]]--;
		dfs(nodes,g,counts,i,1,nums.size(),res);
		counts[nodes[i]]++;
	}
	return res;
}

int main(){
	int n,tmp;
	cin>>n;
	vector<int> nums;
	for(int i=0;i<n;i++){
		cin>>tmp;
		nums.push_back(tmp);
	}
	int count=numSquarefulPerms(nums);
	cout<<count;
	return 0;
}