JAVA程序设计:正方形数组的数目(LeetCode: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
 

提示:

1 <= A.length <= 12
0 <= A[i] <= 1e9

思路:我们将能够组成完全平方数的一对数字连边,之后通过深搜进行遍历得到数组中元素的排列,为了满足时间复杂度的要求,我们需要对DFS进行剪枝,我们可以对于前i个元素组成的排列存储起来避免重复枚举。

class Solution {
	
	private int ans;
	private Set st;
	private boolean[] flag;
	private List> list;
	private Map used;
	
    public int numSquarefulPerms(int[] A) {
    	
    	int n=A.length;
    	
    	if(n<=1) return n;
    	
    	ans=0;
    	flag=new boolean[n];
    	st=new HashSet<>();
    	used=new HashMap<>();
    	list=new ArrayList<>();
    	
    	for(int i=0;i());
    	
    	for(int i=0;i

 

你可能感兴趣的:(JAVA程序设计:正方形数组的数目(LeetCode:996))