回溯的一般模板:
int t[1000]={0},n;
void Recall(int p)
{
if(...)//回溯到目标,记录或者返回等
{ }
else
{
for(int i=0;i<n;++i)
{//n为全局,i的起始值可按题目要求不一样,有时i=p
if(!t[i])//t[i]记录回溯
{
t[i]=1;
...//进行一些操作,记录等
Recall(p+1);//下一层,不能写++p或p++
t[i]=0;
}
}
}
}
H-1:比较基础的回溯题
从 m (0<m≤26) 个大写字母中任意选出 n (0<n≤m) 个字母组合到一起,一共有多少种组合?
请编写程序,输入 m 和 n,输出从 A 开始的连续 m 个字母中任取 n 个字母的所有组合。
要求:每行输出一个组合(组合内的字母按升序排序),按字典序输出。
输入样例
4 3
输出样例
ABC
ABD
ACD
BCD
#include <stdio.h>
int a[26] = {0};
int tag[26] = {0};
int m, n, t = 0;
void Show()
{
int k;
for(k = 0; k < n; k++)
{
putchar(a[k] + 'A');
}
putchar('\n');
}
void Compose(int p)
{
int k;
if(p == n)
{
Show();
}
else
{
for(k = t; k <= (m -n + p); ++k)
{
if(!tag[k])
{
tag[k] = 1;
a[p] = k;
t = k + 1;
Compose(p + 1);
tag[k] = 0;
}
}
}
}
int main()
{
scanf("%d%d", &m, &n);
Compose(0);
return 0;
}
还有个有意思的回溯可看PTA–整数拆分
H-2: 回溯要进行一些必要的剪枝
子集和问题
设集合S={x1,x2,…,xn}是一个正整数集合,c是一个正整数,子集和问题判定是否存在S的一个子集S1,使S1中的元素之和为c。试设计一个解子集和问题的回溯法,并输出利用回溯法在搜索树(按输入顺序建立)中找到的第一个解。
输入格式:
输入数据第1行有2个正整数n和c,n表示S的大小,c是子集和的目标值。接下来的1行中,有n个正整数,表示集合S中的元素。 是子集和的目标值。接下来的1 行中,有n个正整数,表示集合S中的元素。
输出格式:
输出利用回溯法找到的第一个解,以空格分隔,最后一个输出的后面有空格。当问题无解时,输出“No Solution!”。
输入样例:
在这里给出一组输入。例如:
5 10
2 2 6 5 4
输出样例:
在这里给出相应的输出。例如:
2 2 6
#include <iostream>
using namespace std;
int a[10000], t[1000]={0},b[1000];
int m,n,ok=0;
void dp(int p,int s)
{
int i,j;
if(ok)//跳出,减掉不必要的计算
return ;
if(s == m)
{
for(j = 0; j < p; ++j)
{
printf("%d ",b[j]);
}
putchar('\n');
ok=1;
}
else if(s > m)//跳出
{
return ;
}
else
{
for(i = p; i < n; ++i)//i=p,不重复算同一个子集
{
if(!t[i])
{
t[i] = 1;
b[p]=a[i];//记录
dp(p+1,s+a[i]);
t[i] = 0;
}
}
}
}
int main()
{
int i,s=0;
cin >> n >> m;
for(i = 0; i < n; ++i)
{
cin >> a[i];
s+=a[i];
}
if(s<m)//不写,有个测试点超时
{
printf("No Solution!");
}
else
{
dp(0,0);
if(!ok)
{
printf("No Solution!");
}
}
return 0;
}
这个题的剪枝解法应用树结构,上面的解法不属于真正的剪枝,主要要理解要必要的剪枝(剪去不必要的情况),待我更新。
H-3:最佳调度问题(*)
假设有n(n<=20)个任务由k(k<=20)个可并行工作的机器完成。完成任务i需要的时间为ti。 试设计一个算法,对任意给定的整数n和k,以及完成任务i 需要的时间为ti ,i=1~n。计算完成这n个任务的最佳调度,使得完成全部任务的时间最早。
输入格式:
输入数据的第一行有2 个正整数n和k。第2 行的n个正整数是完成n个任务需要的时间。
输出格式:
将计算出的完成全部任务的最早时间输出到屏幕。
输入样例:
在这里给出一组输入。例如:
7 3
2 14 4 16 6 5 3
输出样例:
在这里给出相应的输出。例如:
17
#include <iostream>
#include <algorithm>
using namespace std;
int a[22],b[22],s=100000,t[100]={0};
int n,m;
bool cmp(int c,int d)
{
return c>d;
}
void dpf(int p, int mm)
{
int i;
if(mm>s) return;
if(p==n)//记录所有符合时间最短的
{
if(s>mm)
s=mm;
}
else
{
for(i=0;i<m;++i)//可记录m个线程的时间和
{
if(!t[p])
{
t[p]=1;
b[i]+=a[p];//记录某个线程的时间和
dpf(p+1,max(b[i],mm));//进入所有线程时间最长的
t[p]=0;
b[i]-=a[p];//记得剪去
}
}
}
}
int main()
{
int i;
cin >> n >>m;
for(i=0;i<n;++i)
{
cin>>a[i];
}
dpf(0,0);
cout << s;
return 0;
}
迷宫问题
跳马问题
给定m*n(mn<=100)的棋盘,左上角的点坐标(1,1),一匹马从(1,2)点开始沿着日字型(有8种)跳完棋盘上所有点,且每一个点都只能跳一次,马每一步从(x,y)点起跳时优先选择的方向对应坐标偏移为:{2,1},{1,2},{-1,2},{-2,1},{-2,-1},{-1,-2},{1,-2},{2,-1}。有的棋盘会有很多种跳法,现在比较感兴趣的是马的第k种跳法是什么?
输入格式:
一行三个整数m,n,k。
输出格式:
输出马的第k种跳法,如果马无法跳完棋盘所有点则输出impossible,如果所有跳法数不到k种,则输出最后一种跳法 。
输入样例:
在这里给出一组输入。例如:
4 4 1
4 5 6
输出样例:
在这里给出相应的输出。例如:
impossible
20 1 16 9 12
15 8 11 4 17
2 19 6 13 10
7 14 3 18 5
说明:矩阵中整数1~m*n依次表示马所在位置,;例如1在第1行第2列,表示马起始位置,并且下一步在整数2所对应的第3行第1列。
#include <iostream>
using namespace std;
int a[100][100]={0},b[100][100];
int A[8]={2,1,-1,-2,-2,-1,1,2};
int B[8]={1,2,2,1,-1,-2,-2,-1};//A[]和B[]要对其
//或者写二维数组
//int A[4]={-1,1,0,0};
//int B[4]={0,0,-1,1};在迷宫四个方向这样写
int ok=0,t=0,n,m,k,okk=1;
void dpf(int x, int y, int s)
{
if(s==n*m&&okk)//判断是否符合情况
{
for (int i = 0; i < m; i++)
{
for (int j = 0; j < n; j++)
{
b[i][j] = a[i][j];
}
}
t++;
ok=1;
if(t==k)
{
okk=0;
return ;
}
}
else
{
for(int i = 0; i < 8; ++i)//几个方位,迷宫是四个方位
{
if(x+A[i]<m&&x+A[i]>=0&&y+B[i]<n&&y+B[i]>=0&&!a[x+A[i]][y+B[i]]&&okk)
{//剪枝
a[x+A[i]][y+B[i]]=s+1;
dpf(x+A[i],y+B[i],s+1);
a[x+A[i]][y+B[i]]=0;
}
}
}
}
int main()
{
cin>>m>>n>>k;
b[0][1]=1;
a[0][1]=1;
dpf(0,1,1);
if(ok)
for(int i = 0; i< m;++i)
{
for(int j=0;j< n;++j)
{
if(j)
printf(" %d",b[i][j]);
else
cout << b[i][j];
}
cout << endl;
}
else
{
cout << "impossible";
}
return 0;
}