C ++中平方数组的数量
假设我们有一个由正整数组成的数组A,可以说该数组是正方形的,如果对于每对相邻元素,它们的和是一个完美的平方。我们必须找到平方的A的排列数。当且仅当存在一些索引i使得A1[i]与A2[i]不同时,两个置换A1和A2才会相同。
因此,如果输入像[3,30,6],那么输出将为2,因为我们有两个排列,例如[3,6,30],[30,6,3]。
为了解决这个问题,我们将遵循以下步骤-
定义一个函数isSqr(),将花费n,
x:=n的平方根
当(x*x)与n相同时返回true
定义一个函数solve(),它将使用一个数组a,idx,
如果(idx与0或isSqr(a[idx-1]+a[i]))和a[i]未访问,则-
swap(a[idx],a[i])
解决(a,idx+1)
swap(a[idx],a[i])
在访问过的地方插入一个[i]
(增加1)
返回
如果idx与a的大小相同,则-
定义一组参观
对于初始化i:=idx,当i<a的大小时,更新(将i增加1),执行-
从主要方法中,执行以下操作-
计数:=0
解决(a,0)
返回计数
让我们看下面的实现以更好地理解-
示例
#include <bits/stdc++.h>
using namespace std;
typedef long long int lli;
class Solution {
public:
int count;
bool isSqr(lli n){
lli x = sqrt(n);
return x * x == n;
}
void solve(vector<int>& a, int idx){
if (idx == a.size()) {
count++;
return;
}
set<int> visited;
for (int i = idx; i < a.size(); i++) {
if ((idx == 0 || isSqr(a[idx - 1] + a[i])) &&
!visited.count(a[i])) {
swap(a[idx], a[i]);
solve(a, idx + 1);
swap(a[idx], a[i]);
visited.insert(a[i]);
}
}
}
int numSquarefulPerms(vector<int>& a){
count = 0;
solve(a, 0);
return count;
}
};
main(){
Solution ob;
vector<int> v = {3,30,6};
cout << (ob.numSquarefulPerms(v));
}输入值
{3,30,6}输出结果
2