/** * 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; }