/** * File: permutations_i.cpp * Created Time: 2023-04-24 * Author: krahets (krahets@163.com) */ #include "../utils/common.hpp" /* バックトラッキング:すべての順列 I */ void backtrack(vector &state, const vector &choices, vector &selected, vector> &res) { // 状態の長さが要素数と等しくなったら、解を記録する if (state.size() == choices.size()) { res.push_back(state); return; } // すべての選択肢を走査する for (int i = 0; i < choices.size(); i++) { int choice = choices[i]; // 枝刈り:同じ要素を繰り返し選んではならない if (!selected[i]) { // 試行:選択を行い、状態を更新する selected[i] = true; state.push_back(choice); // 次の選択に進む backtrack(state, choices, selected, res); // 復元:選択を取り消し、以前の状態に戻す selected[i] = false; state.pop_back(); } } } /* すべての順列 I */ vector> permutationsI(vector nums) { vector state; vector selected(nums.size(), false); vector> res; backtrack(state, nums, selected, res); return res; } /* 実行用コード */ int main() { vector nums = {1, 2, 3}; vector> res = permutationsI(nums); cout << "入力配列 nums = "; printVector(nums); cout << "すべての順列 res = "; printVectorMatrix(res); return 0; }