力扣练习——39 正方形数组的数目

39 正方形数组的数目

1.问题描述
给定一个非负整数数组 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

说明：若一个数能表示成某个整数的平方的形式，则称这个数为完全平方数。完全平方数是非负数。
2.输入说明
首先输入A数组的长度n，

然后输入n个整数，以空格分隔。

1 <= n<= 12

0 <= A[i] <= 1e9
3.输出说明
输出一个整数
4.范例
输入
3
1 17 8
输出
2
5.代码

#include <iostream>
#include <queue>
#include <cstdlib>
#include <string>
#include<set>
#include<algorithm>
using namespace std;


int ans;//结果
vector<int>t;
int used[12] = { 0 };
void dfs(vector<int> &nums);
int numSquarefulPerms(vector<int>& nums)
{
	sort(nums.begin(), nums.end());//从小到大排序
	dfs(nums);
	return ans;
}
void dfs(vector<int>& nums) {
	if (t.size() == nums.size()) {
		ans++;
		return;
	}
	for (int i = 0; i < nums.size(); i++) {
		if (i > 0 && nums[i] == nums[i - 1] && !used[i - 1])continue;
		if (!used[i]) {
			if (t.size() == 0 || sqrt(nums[i] + t[t.size() - 1]) - int(sqrt(nums[i] + t[t.size() - 1])) == 0) {
				used[i] = 1;
				t.push_back(nums[i]);//弹入
				dfs(nums);
				used[i] = 0;//回溯
				t.pop_back();//弹出
			}
		}
	}
}


int main()

{

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

}