AcWing92-dfs-递归-状压
题目描述
从 1~n 这 n 个整数中随机选取任意多个,输出所有可能的选择方案。
输入格式
输入一个整数n。
输出格式
每行输出一种方案。
同一行内的数必须升序排列,相邻两个数用恰好1个空格隔开。
对于没有选任何数的方案,输出空行。
本题有自定义校验器(SPJ),各行(不同方案)之间的顺序任意。
数据范围
1≤n≤15
输入样例:
3
输出样例:
3
2
2 3
1
1 3
1 2
1 2 3
做法1:利用深搜,填坑方式将一个个数填进
#include <iostream>
using namespace std;
int n;
int a[20];
bool vis[20];
void dfs(int pos, int start, int tar){
if(pos == tar + 1){
for(int i = 1; i <= tar; i++) cout << a[i] << " ";
cout << endl;
return;
}
for(int i = start; i <= n; i++){
if(! vis[i]){
vis[i] = true; a[pos] = i;
dfs(pos + 1, i + 1, tar);
vis[i] = false;
}
}
}
int main(){
cout << endl;
cin >> n;
for(int i = 1; i <= n; i++){
dfs(1,1,i);
}
return 0;
}
做法2:在做法一的基础上,不开数组a,利用二进制使用一个变量的空间进行存储情况。
#include <iostream>
using namespace std;
int n;
bool vis[20];
void dfs(int pos, int start, int tar, int state){
if(pos == tar + 1){
for(int j = 0; j < n; j++){
if(state >> j & 1) cout << j + 1 << " ";
}
cout << endl;
return;
}
for(int i = start; i <= n; i++){
if(!vis[i]){
vis[i] = true; state |= 1 << (i - 1);
dfs(pos + 1, i + 1, tar, state);
vis[i] = false; state ^= 1 << (i - 1);
}
}
}
int main(){
cout << endl;
cin >> n;
for(int i = 1; i <= n; i++){
dfs(1, 1, i, 0);
}
return 0;
}
做法三(学习自lovebecky):利用状态压缩:如果输入数字为3, 那么就有2的3次方个 结果,通过0,1来判断该位置是否需要输出。
例中n=3,即
000 -> \n
001 -> 1
010 -> 2
100 -> 3
011 -> 1 2
101 -> 1 3
110 -> 2 3
111 -> 1 2 3
#include <iostream>
using namespace std;
int n;
void dfs(int u, int state){
if(u == n){
for(int i = 0; i < n; i++){
if(state >> i & 1){
cout << i + 1 << " ";
}
}cout << endl;
return;
}
dfs(u + 1, state);
dfs(u + 1, state | (1 << u));
}
int main(){
cin >> n;
dfs(0,0);
return 0;
}