Задача: 996. Number of Squareful Arrays
Сложность: hard

Массив является квадратным, если сумма каждой пары соседних элементов является совершенным квадратом. Если задан целочисленный массив nums, верните количество перестановок nums, которые являются квадратными. Две перестановки perm1 и perm2 различны, если существует некоторый индекс i такой, что perm1[i] != perm2[i].

Пример:

Input: nums = [1,17,8]
Output: 2

👨‍💻 Алгоритм:

1⃣Генерация перестановок:
Сгенерируйте все возможные перестановки массива nums.
Для каждой перестановки проверьте, является ли она квадратной.

2⃣Проверка квадратности:
Для каждой перестановки проверьте, является ли сумма каждой пары соседних элементов совершенным квадратом.
Для этого используйте функцию для проверки, является ли число совершенным квадратом.

3⃣Подсчет квадратных перестановок:
Подсчитайте количество перестановок, которые являются квадратными, и верните это значение.

😎 Решение:

public class Solution {
    public int NumSquarefulPerms(int[] nums) {
        Array.Sort(nums);
        var used = new bool[nums.Length];
        var result = new HashSet<string>();
        var path = new List<int>();
        Backtrack(nums, used, path, result);
        return result.Count;
    }
    
    private void Backtrack(int[] nums, bool[] used, List<int> path, HashSet<string> result) {
        if (path.Count == nums.Length) {
            if (IsSquareful(path)) {
                result.Add(string.Join(",", path));
            }
            return;
        }
        
        for (int i = 0; i < nums.Length; i++) {
            if (used[i] || (i > 0 && nums[i] == nums[i - 1] && !used[i - 1])) continue;
            path.Add(nums[i]);
            used[i] = true;
            Backtrack(nums, used, path, result);
            path.RemoveAt(path.Count - 1);
            used[i] = false;
        }
    }
    
    private bool IsSquareful(List<int> perm) {
        for (int i = 0; i < perm.Count - 1; i++) {
            int sum = perm[i] + perm[i + 1];
            int root = (int)Math.Sqrt(sum);
            if (root * root != sum) return false;
        }
        return true;
    }
}

Ставь 👍 и забирай 📚 Базу знаний