hello-algo/ja/codes/javascript/chapter_backtracking/permutations_ii.js

/**
 * File: permutations_ii.js
 * Created Time: 2023-05-13
 * Author: Justin ([email protected])
 */

/* バックトラッキング：すべての順列 II */
function backtrack(state, choices, selected, res) {
    // 状態の長さが要素数と等しくなったら、解を記録する
    if (state.length === choices.length) {
        res.push([...state]);
        return;
    }
    // すべての選択肢を走査する
    const duplicated = new Set();
    choices.forEach((choice, i) => {
        // 枝刈り：同じ要素の再選択も、等しい値を持つ要素の重複選択も許可しない
        if (!selected[i] && !duplicated.has(choice)) {
            // 試行：選択を行い、状態を更新する
            duplicated.add(choice); // 選択済みの要素の値を記録する
            selected[i] = true;
            state.push(choice);
            // 次の選択に進む
            backtrack(state, choices, selected, res);
            // 復元：選択を取り消し、以前の状態に戻す
            selected[i] = false;
            state.pop();
        }
    });
}

/* すべての順列 II */
function permutationsII(nums) {
    const res = [];
    backtrack([], nums, Array(nums.length).fill(false), res);
    return res;
}

// 実行用コード
const nums = [1, 2, 2];
const res = permutationsII(nums);

console.log(`入力配列 nums = ${JSON.stringify(nums)}`);
console.log(`すべての順列 res = ${JSON.stringify(res)}`);