打印子集树。比如说有三个元素,用0和1表示子集有或者没有这个元素,向左分支走为1,向右分支走为0,那么如下图所有路径都可以用0和1表示出来,可以用0和1完整表示子集。0就不打印对应的元素,1就打印对应的元素。比如三个元素是{1,2,3}路径为{0,1,0}的话就打印{2};路径为{1,1,0}的话就打印{1,2}。
递归代码:
void fun(int* ar, int* br, int i, int n)
{
if (i == n)//如果一条路径走完了就打印
{
for (int j = 0; j < n; j++)
{
if (br[j] == 1)
{
cout << ar[j]<<" ";
}
}
cout << endl;
}
else//左为1,右为0
{
br[i] = 1;
fun(ar, br, i + 1, n);//left
br[i] = 0;
fun(ar, br, i + 1, n);//right
}
}
int main()
{
const int n = 3;
int ar[n] = { 1,2,3 };
int br[n] = { 0,0,0 };
fun(ar, br, 0, n);
return 0;
}
运行结果:
下面我们用非递归方法,用回溯法代码打印子集树。
先上代码:
void Print(int* ar, int* br, int n)
{
for (int i = 0; i < n; ++i)
{
if (br[i] == 1)
{
printf("%2d", ar[i]);
}
}
printf("\n");
}
int main()
{
const int n = 3;
int ar[n] = { 1,2,3 };
int br[n] = { 0,0,0 };//初始化为-1,0 1,2是越界
int i = 0;
while (i >= 0)
{
if (br[i] >= 2)//遇到2就回溯
{
--i;//回溯
if (i != -1 && br[i] != 2)
{
br[i] += 1;
}
}
else//没遇到2,向后走,如果整个数组都没2就打印
{
if (i == n - 1)//i走到了最后
{
Print(ar, br, n);
br[i] += 1;
}
else//i没有走到最后
{
++i;
br[i] = 0;
}
}
}
return 0;
}
运行结果:
分析:
br数组用1,0表示ar集合的子集和中元素的有无。
比如ar = {1,2,3}
br = {0,0,0}表示ar一个子集不含任何元素
br = {1,0,1}表示ar一个子集含有第一个元素1,和第三个元素3
…
br数组中在代码循环中可能出现2,2是为了回溯用的,遇到2就回溯。
第一次的时候数组没有2,i直接走到最后,br = {0,0,0}所以ar子集为空集,什么也没打印。
第二次的时候打印完空集后,br[2]赋值为1,进入到下一次循环 ,没有遇到2,不需要回溯,br = {0,0,1},打印。
第三次的时候,第二次打印完后,br[2] += 1,变成2,继续进入下一次循环,i = 2,br[i] = 2,回溯i–,br[1]被赋值为1,此时br = {0,1,2},进入到下一次循环,将br[2]置为0。br = {1,0,1}打印。
第四次的时候,第三次打印完后,br[2] += 1, 变成1,进入下一次循环,没遇到2,br = {0,1,1}打印。
第五次的时候,第四次打印完后,br[2] += 1, 变成2,进入下一次循环,回溯i–,br[1]+=1,变成2,继续回溯,br[0]+=1,变成1,此时br = {1,2,2},i = 0,进入下两次循环,将2都置为0。br = {1,0,0},打印。
第六次循环的时候,第五次打印完,br[2] += 1,变成1,进入下一次循环,没有遇到2,没有回溯,br = {1,0,1},打印。
第七次的时候,第六次打印完,br[2] += 1, 变成2,进入下一次循环,i = 2,br[2] = 2,回溯i–,变为1,br[1] += 1,变为1,进入下一次循环,将br[2]置为0,br = {1,1,0},打印。
第八次的时候,第七次打印完,br[2] += 1,变成1,进入下一次循环,没有2,br = {1,1,1},打印。
第九次的时候,第八次打印完,br[2] +=1,变成1,进入下一次循环,br[2] = 2,回溯i–,变为1,br[1]+=1,变为2,进入下一次循环,br[1] = 2,回溯i–,变为0,br[0] += 1,变为2,再进入下一次回溯,i–,小于0, 退出循环,结束。