#include<vector>
using namespace std;
class Solution {
private:
void perm(vector<int>& nums, int p, int q) {
if (p == q) {
answer.push_back(nums);
}
for (int i = p; i <= q; i++) {
swap(nums, p, i);
perm(nums, p + 1, q);
swap(nums, p, i);
}
}
void swap(vector<int>& nums, int p, int q) {
int temp = nums[p];
nums[p] = nums[q];
nums[q] = temp;
}
vector<vector<int>> answer;
public:
vector<vector<int>> permute(vector<int>& nums) {
perm(nums, 0, nums.size() - 1);
return answer;
}
};
作业:从长度为n(n>0)的数组A中选出k个元素(1<=k<=n),已知A中元素的值各不相同,请编写一个算法输出其全排列。
整体思路:先递归得到所有组合,再分别对这些组合进行全排列
#include<stdio.h>
class permutation {
private:
int* B;
int k;//数组总长度
void print(int begin, int end) {
for (int i = begin; i <= end; i++) {
printf("%d ", B[i]);
}
printf("\n");
}
void swap(int p, int q) {
int temp = B[p];
B[p] = B[q];
B[q] = temp;
}
void perm(int p, int q) {//给出p和q之间的所有排列方式
if (p == q) {
print(p - k + 1, q);
}
else {
for (int i = p; i <= q; i++) {
swap(p, i);
perm(p + 1, q);
swap(p, i);
}
}
}
public:
permutation(int* B, int k) :B(B), k(k) {
perm(0, k - 1);
}
};
class combination {
private:
int* A;//原始数组
int* B;//新产生数组
int n;//数组A长度
int k;//子数组长度
void print() {
for (int i = 0; i < k; i++) { printf("%d ", B[i]); }
}
//i为待取出元素在A中的下标,j为已取出元素在B中的下标,k为还需要取出的数量
void combine(int i, int j, int k) {
if (k == 0) {
print();
printf("这种组合的排列有:\n");
permutation permute(B, this->k);
}
else if (i + k <= n) {
B[j] = A[i];
j++;
combine(i + 1, j, k - 1);
combine(i + 1, j - 1, k);
}
}
public:
combination(int* A, int n, int k) :A(A), n(n), k(k) {
B = new int[k];
combine(0, 0, k);
}
};
int main() {
int A[5] = { 0,1,2,3,4 };
combination answ(A, 5, 3);
return 0;
}