飞道的博客

数据结构与算法_五大算法之--回溯算法

255人阅读  评论(0)

1 回溯算法

回溯算法具有通用性,但是算法的效率不高,通常可以通过剪枝等操作提高算法的效率。
算法思想:
在包含问题的所有解空间树中,按照深度优先搜索的策略,从根节点出发,深度搜索解空间树。当搜索到某一个节点时候,先判断该节点是否包含问题的解,如果包含就从该节点触发继续深度搜索下去,否则回溯;
解空间:
解空间就是所有解的可能取值构成的空间,一个解往往包含了得到这个解的每一步,往往就是对应解空间树中一条从根节点到叶子节点的路径。子集树和排列树都是一种解空间,它们不是真实存在的数据结构,也就是说并不是真的有这样一颗树,只是抽象出的解空间树。

1.1 解空间—子集树

子集树通常用于求某个问题的子集,涉及到子集的问题,看是否可以用子集树方式。

void func(int arr[], int i, int length,int x[])
{
   
	if (i == length)
	{
   
		for (int j = 0; j < length; j++)
		{
   
			if (x[j] == 1)
			{
   
				cout << arr[j] << " ";
			}
		}
		cout << endl;
	}
	else
	{
   
		x[i] = 1;  // 选择i节点
		func(arr, i + 1, length,x); // 遍历 i 的左孩子
		
		x[i] = 0; // 不选择i节点
		func(arr, i + 1, length,x); // 遍历i的右孩子
		//func(arr, i + 1, length); // 三叉树
	}
}

int main()
{
   
	int arr[] = {
    1, 2, 3 };
	int length = sizeof(arr) / sizeof(arr[0]);
	int x[3] = {
    0 };
	func(arr, 0, length,x);
	system("pause");
	return 0;
}

 

对代码的理解:

子集树中的两个循环,在逻辑上生成二叉树。
上面在递归过程中,一个节点(栈帧),向左递归标记为1,向右标记为0,打印时候,按照条件打印。
上边对递归左右走做一个标记,出来的就是子集树。算法时间复杂度,O(2^n)。

子集树总结

1 , 新开辟的栈帧中,从第一句代码开始执行;即使,第二句代码开辟的栈帧,仍然从新开辟的栈帧的第一句开始执行;
2 , 回溯就是结束,回溯到上一个节点后,如果后边有语句,继续执行后边的语句;
3 , 每一层栈帧中i节点相同,也就是层数相同。



1.2 解空间— 排列树

排列树就是获取某个集合的全排列,比如求1 2 3这组数的全排列;
假如求1 2 3 全排列

void swap(int arr[], int i, int j)
{
   
	int temp = arr[i];
	arr[i] = arr[j];
	arr[j] = temp;
}

void func(int arr[], int i, int length)
{
   
	if (i == length)
	{
   
		for (int j = 0; j < length; j++)
		{
   
			cout << arr[j] << " ";
		}
		cout << endl;
	}
	else
	{
   
		for (int k = i; k < length; k++)
		{
   
			swap(arr, i, k);
			func(arr, i + 1, length);
			swap(arr, i, k);
		}
	}
}

int main()
{
   
	int arr[] = {
    1,2,3 };
	int length = sizeof(arr) / sizeof(arr[0]);

	func(arr, 0, length);

	system("pause");
	return -1;
}

 



1.3 整数选择问题

整数选择:给定一组整数,从里面挑选出一组整数,让选择的整数的和 与 剩下的整数的和的差最小;

int min = 10000;
int result = 0;
int Xsum = 0;   //  记录选择的元素之和
int Ysum = 0;	//  记录没有选择的元素之和
void func(int arr[],int i,int length,int x[])
{
   
	if (i == length)
	{
   
		for (int j = 0; j < length; j++)
		{
   
			if (x[j] == 1)
			{
   
				
			}
			else
			{
   
				Ysum += arr[j];	// 记录未选择的元素之和
			}
		}
		result = abs(Xsum - Ysum);
		Xsum = 0;
		Ysum = 0;
		if (result <= min)
		{
   
			min = result;
		}
		cout << endl;
	}
	else
	{
   
		x[i] = 1;// 1 从当前节点向左走
		Xsum += arr[i];
		func(arr, i + 1, length,x);

		Ysum += arr[i];
		x[i] = 0;		// 2 回溯到父节点,然后从父节点向右走
		func(arr, i + 1, length,x);
	}
}

int main()
{
   
	int arr[] = {
    12,6,7,11,16,3,8 };
	int length = sizeof(arr) / sizeof(arr[0]);
	int x[7] = {
    0 };
	func(arr, 0, length,x);
	cout << "选择和剩下的整数之和为:" << min << endl;
	system("pause");
	return 0;
}

 

1.4 2N整数选择问题-加入剪枝操作

给2n个整数,从里面挑选出n个整数,让选择的整数的和 与 剩下的整数的和的差值最小;

int arr[] = {
    12,6,7,11,16,3,5,23};
int length = sizeof(arr) / sizeof(arr[0]);
vector<int> bestM;		// 记录选择的最小值
vector<int> M;			// 记录子集中选择的元素 
unsigned int minval = 0xffffffff;
int sumR = 0;	// 记录数组中未被选择的元素之和
int sumS = 0;	// 记录选中的数组元素总和
int cnt = 0;  // 记录遍历的子集的个数
int left1 = length;
// 未加入剪枝操作
void func1(int i)
{
   
	if (i == length)
	{
   
		cnt++;
		if (M.size() != length / 2)
		{
   
			return;
		}

		int result = abs(sumS - sumR);
		if (result < minval)
		{
   
			minval = result;
			bestM = M;
		}
	}
	else
	{
   
		M.push_back(arr[i]);   // 记录选择的元素
		sumR -= arr[i];
		sumS += arr[i];
		func1(i + 1);
		// a
		cout << endl;
		sumS -= arr[i];
		sumR += arr[i];
		M.pop_back();

		func1(i + 1);
	}
}
// 加入剪枝操作 
void func(int i)
{
   
	if (i == length)
	{
   	
		cnt++;
		if (M.size() != length / 2)
		{
   
			return;
		}
	
		int result = abs(sumS - sumR);
		if (result < minval)
		{
   
			minval = result;
			bestM = M;
		}
	}
	else
	{
   
		left1--;
		if (M.size() < length / 2)   // 减左树枝,还未选择够n个整数
		{
   
			// 下面三行,递归做的事情
			M.push_back(arr[i]);   // 记录选择的元素
			sumR -= arr[i];
			sumS += arr[i];

			func(i + 1);
			// 回溯
			sumS -= arr[i];
			sumR += arr[i];
			M.pop_back();
		}
		// 右树枝可以选吗?已选择的数字的个数 + 未来能选择的所有数组个数(i+1,i+2,...i+n) >= n个元素
		//				   已选择的数字的个数 + 未来能选择的所有数组个数(i+1,i+2,...i+n) <= n个元素  则不执行 
		if (M.size() + left1 >= length / 2)
		{
   
			func(i + 1);
		}
		left1++;
	}
}

int main()
{
   
	// 计算出数组总和
	for (int v : arr)
	{
   
		sumR += v;
	}
	func(0);
	
	cout << "最小值" << minval << endl;
	for (int v : bestM)
	{
   
		cout << v << " ";
	}
	cout << endl;
	cout << "cnt:  " << cnt << endl;
	system("pause");
	return 0;
}

 

1.5 挑选一组数字,让他们的和等于指定的值

挑选数字:有一组整数,请挑选出一组数字,让他们的和等于指定的值,存在解打印,不存在打印

int arr[] = {
    12,6,7,11,16,3,5,23 };
const int number = 34;
int length = sizeof(arr) / sizeof(arr[0]);
vector<int> x;  // 记录选择的数字
int sum = 0;	// 记录选择的数字之和
int r = 0;		// 记录未处理的数字的和

void func(int i)
{
   
	if (i == length)
	{
   
		if (sum != number)
		{
   
			return;
		}
		for (int v : x)
		{
   
			cout << v << " ";
		}
		cout << endl;

	}
	else
	{
   
		r -= arr[i];
		// 加入剪枝操作,减左枝:  如果已选择的数字的和 + 加上即将选择的的数字的和 < number 
		if (sum + arr[i] <= number)
		{
   
			sum += arr[i];		 // 记录所选数字之和
			x.push_back(arr[i]);
			func(i + 1);

			sum -= arr[i];		 // 减去右子树中的元素
			x.pop_back();
		}
		if (sum + r >= number)
		{
   
			func(i + 1);
		}

		r += arr[i];
	}
		
}

int main()
{
   
	for (int v : arr)
	{
   
		r += v;
	}

	func(0);
	// 计算出数组总和
	system("pause");
	return 0;
}

 

1.6 穷举法 (挑选数字)

穷举法和子集树的关系:穷举法所有的可能是子集树的子集;子集树能解决的问题,穷举法可能解决不了;
对代码的理解:
for的第一层 1 2 3 4 ;以1为确定的值第二层 2 3 4 ,以2为可能值的第2层,3,4;等等。

int arr[] = {
    12,6,7,11,16,5,5,23 };
int number = 34;
int length = sizeof(arr) / sizeof(arr[0]);
vector<int> vec;  // 存放选择的数字

void func(int i, int number)
{
   
	if (number == 0)
	{
   
		for (int v : vec)
		{
   
			cout << v << " ";
		}
		cout << endl;
	}
	else
	{
   
		for (int k = i; k < length; k++)
		{
   
			if (number >= arr[k])// number大于下一个元素,比如number=8,下一个元素为13,不满足;如果下一个元素为3,number>=3,满足
			{
   
				vec.push_back(arr[k]);

				func(k + 1, number - arr[k]); // Andy: k+1这种方式表示不允许重复选择元素,只能遍历当前元素的孩子节点
				//func(k, number - arr[k]);// Andy: k这种方式表示允许重复选择元素,只能遍历当前元素的孩子节点

				vec.pop_back();
			}
		}
	}
}

int main()
{
   
	func(0,number);
	// 计算出数组总和
	system("pause");
	return 0;
}


 

1.7 0-1背包

有一组物理,其重量分别是w1 w2 w3,,其价值分别是v1 v2 ,vn,现在有一个背包,其容量是C,问怎么把物品装入背包,能够使背包的价值最大化。
解法:用子集树求解

int w[] = {
    12,5,8,9,6 };  // 物品的重量
int v[] = {
    9,2,4,7,8 };   // 物品的价值,物品和价值是一对一关系 
int C = 25;

const int length = sizeof(w) / sizeof(w[0]);

int cw = 0; // 已经选择的物品的重量;
int cv = 0; // 已经选择的物品的价值 

vector<int> x; // 记录已经选择的物品 
vector<int> bestx; // 记录最优解 
int bestv = 0;
int r = 0; // 记录剩余元素的总价值
int cnt = 0;

void func(int i)
{
   
	if (i == length)
	{
   
		cnt++;
		if (C == cw)
		{
   
			if (bestv < cv)
			{
   
				bestv = cv;
				bestx = x;
			}
		}

	}
	else
	{
   
		r -= v[i];
		// 开始加入剪枝操作,减左枝
		if (cw + w[i] <= C)  // 已选择物品重量 + 即将选择的物品重量,小于总容量再操作 
		{
   
			x.push_back(w[i]);
			cw += w[i];
			cv += v[i];

			func(i + 1);

			x.pop_back();
			cw -= w[i];
			cv -= v[i];
		}
		
		// 加入减右枝操作
		if (cv + r > bestv)
		{
   
			func(i + 1);
		}
		r += v[i];
	}
}

int main()
{
   

	for (int v : v)
	{
   
		r += v;
	}
	func(0);
	cout << "cnt" <<  cnt << endl;
	cout << "选择的物品重量为:";
	for (int w : bestx)
	{
   
		cout << w << " ";
	}
	cout << endl;
	cout << "最优值为:" << bestv << endl;

	system("pause");
	return 0;

}

 

1.8 排列树解决N-皇后问题

8*8的格子中,任意两个棋子不能出现在同一行,同一列,同一条斜线上。

int cnt = 0;
// 8皇后问题
bool judge(int arr[], int i)
{
   
	for (int j = 0; j < i; j++)
	{
   
		if (i == j || arr[i] == arr[j] || abs(i - j) == abs(arr[i] - arr[j]))
		{
   
			return false;
		}
	}
	return true;
}

// 
void swap(int arr[], int i, int j)
{
   
	int temp = arr[i];
	arr[i] = arr[j];
	arr[j] = temp;
}

void func(int arr[], int i, int length)
{
   
	if (i == length)
	{
   
		cnt++;
		for (int j = 0; j < length; j++)
		{
   
			cout << arr[j] << " ";
		}
		cout << endl;
	}
	else
	{
   
		for (int k = i; k < length; ++k)
		{
   
			
			swap(arr, i, k);
			if (judge(arr, i))
			{
   
				func(arr, i + 1, length);   // 每次从上层遍历到下层的过程中,每向下一层,理解为选择每行每列的过程;
			}								// 遍历到第i层时候,都要判断0 -- i-1层,是否满足条件
			swap(arr, i, k);
		}
	}
}


int main()
{
   
	int arr[] = {
    1,2,3,4 ,5,6,7,8};   // arr数组下标表示行,下标对应位置表示列;
	int n = 8;
	int length = sizeof(arr) / sizeof(arr[0]);
	func(arr, 0, length);

	cout << "cnt = " << cnt << endl;
	system("pause");
	return -1;
}

 

1.9 穷举法生成排列树

int arr[] = {
    1,2,3 };
int length = sizeof(arr) / sizeof(arr[0]);
vector<int> vec;
int state[3] = {
    0 };  // 1表示选择了,0表示未选择

void func(int i)
{
   
	if (i == length)
	{
   
		for (int v : vec)
		{
   
			cout << v << " ";
		}
		cout << endl;
	}
	else
	{
   
		for (int k = 0; k < length; ++k)
		{
   
			if (state[k] == 0)
			{
   
				state[k] = 1;
				vec.push_back(arr[k]);
				func(i + 1);
				vec.pop_back();
				state[k] = 0;
			}
		}
	}
}

int main()
{
   
	func(0);

	system("pause");
	return -1;
}

 

1.10穷举法func(k+1) 与 排列树中的fun(i+1)中,i 和 k区别?

穷举法中的k表示可选择的元素的起始下标,并且选择的数量不同,树的深度不同。

排列树中的i表示的是层数,排列树最后生成的是平衡树,树的深度相同。


转载:https://blog.csdn.net/weixin_43916755/article/details/128500593
查看评论
* 以上用户言论只代表其个人观点,不代表本网站的观点或立场