996. 正方形数组的数目

给定一个非负整数数组 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/number-of-squareful-arrays
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution {
public:
//count记录每种元素剩余的数量
//Map 使用邻接表法记录图 
//第i个结点与第j个结点中间有边的条件就是两个结点的值相加后是完全平方数
//然后从每个结点开始都深度优先搜索 图中每个结点都只出现一次的路径（这里若一个数在A中出现两次 则图中可以访问两次这个结点）
    int dfs(int cur,int LeftSize,map<int,int>& count,map<int,vector<int>>& Map)
    {
        if(!LeftSize)return 1;
        int res=0;
        for(auto next:Map[cur])
        {
            if(count[next])
            {
                count[next]--;
                res+=dfs(next,LeftSize-1,count,Map);
                count[next]++;
            }
        }
        return res;
    }
    int numSquarefulPerms(vector<int>& A) {
        int N=A.size();
        map<int,int> count;
        map<int,vector<int>> Map;
        for(auto i:A)count[i]++;
        for(auto p:count)
            for(auto q:count)
            {
                int sum=p.first+q.first;
                int sqrt_val=sqrt(sum);
                if(sqrt_val*sqrt_val==sum)
                    Map[p.first].push_back(q.first);
            }
        int res=0;
        for(auto p:count)
        {
            count[p.first]--;
            res+=dfs(p.first,N-1,count,Map);
            count[p.first]++;
        }
        return res;

    }
};