回溯法 求集合全排列、子集

回溯法,参见之前的blog

全排列:

全排列是将一组数按一定顺序进行排列,如果这组数有n个,那么全排列数为n!个。

从集合中依次选出每一个元素,作为排列的第一个元素,然后对剩余的元素进行全排列,如此递归处理,从而得到所有元素的全排列。

以对字符串abc进行全排列为例,我们可以这么做:以abc为例
固定a,求后面bc的排列:abc,acb,求好后,a和b交换,得到bac
固定b,求后面ac的排列:bac,bca,求好后,c放到第一位置,得到cba
固定c,求后面ba的排列:cba,cab。

这个思想和回溯法比较吻合。

代码可如下编写所示

// 回溯法搜索全排列树

#include<stdio.h>

#define M 20
int n;
int a[M];
int cnt = 0;// 记录全排列个数

void swap(int *a, int *b)//交换a,b
{
    char t;
    t = *a;
    *a = *b;
    *b = t;
}

void dfs(int cur)
{
	int i;
	if(cur == n)// 找到 输出全排列
	{
		++cnt;
		for(i=0; i<n; i++)
			printf("%d ", a[i]);
		printf("\n");
	}
	else
	{
		 // 将集合中的所有元素分别与第一个交换,这样就总是在
		// 处理剩下元素的全排列(即用递归)
		for(i=cur; i<n; i++)
		{
			swap(&a[cur], &a[i]);
			dfs(cur+1);
			swap(&a[cur], &a[i]);//回溯
		}
	}
}


int main()
{
	while(scanf("%d", &n) != EOF)
	{
		for(int i=0; i<n; i++)
			a[i] = i+1;// 假设集合S为:1 2 3 ... n
		cnt = 0;
		dfs(0);
		printf("count:%d\n", cnt);
	}
	return 0;
}


或者利用一个vis数组标识每个元素是否已经被访问,代码如下:

#include <stdio.h>

int a[10];
bool vis[10];
int n;//排列个数 n
void dfs(int dep)  //打印所有的全排列,穷举每一种方案
{
    if(dep == n)
    {
        for(int i = 0; i < n; i++)
        {
            printf("%d ",a[i]);
        }
        printf("\n");
        return ;
    }
    for(int i = 0; i < n; i++)// 找一个最小的未标记的数字,保证了字典序最小
    {
        if(!vis[i])
        {
            a[dep] = i+1;
            vis[i] = true;// 找到了就标记掉,继续下一层
            dfs(dep + 1);
            vis[i] = false;
        }
    }
}
int main()
{
    while(scanf("%d",&n) != EOF)
    {
        dfs(0);
    }
    return 0;
}

子集构造:

从n个元素的集合S中找出S满足某种性质的子集时,相应解空间树称为子集树。

求n个元素集合的子集,如A = {1, 2, 3}则A集合的子集有:  
 P(A) = {{1,2,3}, {1,2}, {1,3},{1},{2,3},{2},{3},{}}

采用位向量法,构造一个位向量vis[], vis[i] = 1 表示i在子集A中。

代码如下:

#include<stdio.h>

#define M 20
int n;
int a[M];
int vis[M];
int cnt = 0;// 记录子集个数


void dfs(int cur)
{
	int i;
	if(cur == n)// 找到 输出所有子集
	{
		++cnt;
		int flag =1;
		for(i=0; i<n; i++)
			if(vis[i])
			{
				printf("%d ", a[i]);
				flag = 0;
			}
			if(flag)//子集中的空集
				printf("φ");
		printf("\n");
	}
	else
	{
		for(i=1; i>=0; --i)//vis 中分为 选or不选即 1,0
		{
			vis[cur] = i;
			dfs(cur+1);
		}
	}
}


int main()
{
	while(scanf("%d", &n) != EOF)
	{
		for(int i=0; i<n; i++)
			a[i] = i+1;// 假设集合S为:1 2 3 ... n
		cnt = 0;
		dfs(0);
		printf("count:%d\n", cnt);
	}
	return 0;
}


此外可以采用采用增量构造法,

代码如下:

#include<stdio.h>

#define M 20
int n;
int a[M];
int vis[M];
int cnt = 0;// 记录子集个数


void subset(int cur, int A[])
{
	int i;
	++cnt;
	if(0 == cur)//子集中的空集
		printf("φ");
	for(i=0; i<cur; ++i)// 打印当前集合
	{
		printf("%d ", A[i]);
	}
	printf("\n");
	
	int min  = cur ? A[cur-1]:0;//确定当前元素的最小可能值
	for(i=min; i<n; ++i)
	{
		A[cur] = a[i];
		subset(cur+1, A);//递归构造
	}
}


int main()
{
	while(scanf("%d", &n) != EOF)
	{
		for(int i=0; i<n; i++)
			a[i] = i+1;// 假设集合S为:1 2 3 ... n
		cnt = 0;
		int A[M]={0};
		subset(0, A);
		printf("count:%d\n", cnt);
	}
	return 0;
}


参考资料:

算法竞赛入门经典

    原文作者:回溯法
    原文地址: https://blog.csdn.net/daniel_ustc/article/details/17041933
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞