给定一个大小为n的数组,输出所有全排列。
1、数组中无重复元素:从前到后,依次将数组中的元素放置在输出数组中,若放完就输出,然后递归的改变元素的位置,直到输出所有的全排列。递归中使用一个标记数组来记录某个元素是否已经被放置过。
2、数组中有重复元素:这个和1比较类似,区别在于用一个标记数组记录某个元素的次数,每放置一次就减一。
问题1的代码如下:
void PermRecur(vector<int>& ivec, vector<int>& use, vector<int>& res, int k)
{
//如果下标k等于size,说明所有元素已经放置完毕
if (k == ivec.size())
{
for (int i = 0; i < k; ++i)
cout << res[i] << ' ';
cout << endl;
return;
}
for (int i = 0; i < ivec.size(); ++i)
{
//标记为0表示没有被放置过
if (use[i] == 0)
{
//元素被放置后将标记赋为1
use[i] = 1;
res[k] = ivec[i];
PermRecur(ivec, use, res, k + 1);
//递归结束后再将该元素的标记赋为0
use[i] = 0;
}
}
}
void Perm(vector<int>& ivec)
{
int size = ivec.size();
assert(size > 0);
//所有元素的标记为初始化为0
vector<int> use(size);
vector<int> res(size);
PermRecur(ivec, use, res, 0);
}
void main()
{
int a[] = {1, 2, 3};
vector<int> ivec(a, a + 3);
Perm(ivec);
}
问题2的代码如下:
#include <iostream>
#include <vector>
#include <assert.h>
using namespace std;
void PermRecur(vector<int>& num, vector<int>& use, vector<int>& res, int k)
{
//如果下标k等于size,说明所有元素已经放置完毕
if (k == res.size())
{
for (int i = 0; i < k; ++i)
cout << res[i] << ' ';
cout << endl;
return;
}
for (int i = 0; i < num.size(); ++i)
{
//标记大于0说明还有剩余的元素没有放置
if (use[i] > 0)
{
//元素被放置后次数减1
--use[i];
res[k] = num[i];
PermRecur(num, use, res, k + 1);
//递归结束后再将该元素的次数加1
++use[i];
}
}
}
void Perm(vector<int>& ivec)
{
assert(!ivec.empty());
int size = ivec.size();
vector<int> res(size);
vector<int> use;
vector<int> num;
//根据ivec统计每个元素的出现次数,并将去重后的所有元素存在num中
use.push_back(1);
num.push_back(ivec[0]);
for (int i = 1; i < size; ++i)
{
int j = 0;
for (; j < i; ++j)
{
if (ivec[i] == ivec[j])
{
++use[j];
break;
}
}
if (j == i)
{
use.push_back(1);
num.push_back(ivec[i]);
}
}
PermRecur(num, use, res, 0);
}
void main()
{
int a[] = {1, 2, 2};
vector<int> ivec(a, a + 3);
Perm(ivec);
}
给定一个大小为n的数组,输出所有的子集。子问题是从n个元素中选择m个元素,然后循环调用即可。
1、数组中无重复的元素:从前到后依次将数组中的元素放置到输出数组中,当元素个数达到m时输出即可。然后进行递归,将剩余的元素和m个元素进行交换。
2、数组中有重复的元素:和全排列类似,先去重并计算元素的次数,然后进行递归。
问题1的代码如下:
#include <iostream>
#include <vector>
#include <assert.h>
using namespace std;
void CombRecur(vector<int>& ivec, vector<int>& res, int m, int begin, int k)
{
//当k等于m的时候输出结果
if (k == m)
{
for (int i = 0; i < res.size(); ++i)
cout << res[i] << ' ';
cout << endl;
return;
}
for (int i = begin; i < ivec.size(); ++i)
{
res[k] = ivec[i];
CombRecur(ivec, res, m, i + 1, k + 1);
}
}
void Comb(vector<int>& ivec)
{
assert(!ivec.empty());
for (int i = 1; i <= ivec.size(); ++i)
{
vector<int> res(i);
CombRecur(ivec, res, i, 0, 0);
}
}
void main()
{
int a[] = {1, 2, 3};
vector<int> ivec(a, a + 3);
Comb(ivec);
}
问题2的代码如下:
#include <iostream>
#include <vector>
#include <assert.h>
using namespace std;
void CombRecur(vector<int>& num, vector<int>& use, vector<int>& res,
int m, int begin, int k)
{
//当k等于m时,说明已经选够m个元素
if (k == m)
{
for (int i = 0; i < res.size(); ++i)
cout << res[i] << ' ';
cout << endl;
return;
}
for (int i = begin; i < num.size(); ++i)
{
//如果元素次数大于0就选择
if (use[i] > 0)
{
//选择之后就将元素次数减1
--use[i];
res[k] = num[i];
//这里递归的时候begin仍然是i,因为第i个元素有可能重复
CombRecur(num, use, res, m, i, k + 1);
++use[i];
}
}
}
void Comb(vector<int>& ivec)
{
assert(!ivec.empty());
int size = ivec.size();
vector<int> num;
vector<int> use;
//根据ivec统计每个元素的出现次数,并将去重后的所有元素存在num中
num.push_back(ivec[0]);
use.push_back(1);
for (int i = 1; i < size; ++i)
{
int j = 0;
for (; j < i; ++j)
{
if (ivec[i] == ivec[j])
{
++use[j];
break;
}
}
if (j == i)
{
use.push_back(1);
num.push_back(ivec[i]);
}
}
//从1~n循环调用即可求出大小为1~n的所有子集
for (int i = 1; i <= size; ++i)
{
vector<int> res(i);
CombRecur(num, use, res, i, 0, 0);
}
}
void main()
{
int a[] = {1, 2, 2};
vector<int> ivec(a, a + 3);
Comb(ivec);
}