蓝桥杯dfs搜索专题
2018激光样式
#include<bits/stdc++.h>
using namespace std;
/*
dfs(i) 第i个激光机器 有两种选择:vis[i-1] == 0 时 可选,无论vis[i-1]为何值都不选
vis[i] 回溯标记是否用过
*/
int n = 30;
int vis[35];
int ans = 0;
int dp[35];
void dfs(int x){
if(x == n+1){
ans++;
return;
}
dfs(x+1); //这个点不开激光
if(vis[x-1] == 0){//前一个点没开激光 那么这个点可以开激光: vis[x] = 1就表示开激光
vis[x] = 1;
dfs(x+1);
vis[x] = 0;//回溯
}
}
int main(){
for(int i=1;i<=30;i++) vis[i] = 0;
dfs(1);
cout<<ans<<endl;
return 0;
}
//2178309
2017磁砖样式
#include<bits/stdc++.h>
using namespace std;
int n,m;
const int maxn = 10;
int g[maxn][maxn];
vector<int> v;
set<vector<int> > se;
set<vector<int> > se2;
map<int, int> Hash;
int ans = 0;
bool check_color() {
for(int i = 1; i <= n; i++)
for(int j = 1; j <= m; j++) {
if(i+1 <= n && j+1 <= m) {
//1 1 1 1 2 2 2 2 1 2 1 2
if((g[i][j]+g[i][j+1]+g[i+1][j]+g[i+1][j+1]) % 4 == 0)
return false;
}
}
return true;
}
bool check2(){
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
if(g[i][j] == 0){
return false;
}
}
}
for(int i=1;i<=n-1;i++){
for(int j=1;j<=m-1;j++){
int aa = g[i][j];
int bb = g[i+1][j];
int cc = g[i][j+1];
int dd = g[i+1][j+1];
if(aa == bb && aa ==cc && bb== cc && cc == dd && bb == dd && aa == dd){
return false;
}
}
}
return true;
}
bool check(){
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
if(g[i][j] == 0){
return false;
}
}
}
for(int i=1;i<=n-1;i++){
for(int j=1;j<=m-1;j++){
if(g[i][j] == g[i+1][j] == g[i][j+1] == g[i+1][j+1])
return false;
}
}
return true;
}
void dfs(int x,int y){
if(x == n+1 && y == 1){
// for(int i=1;i<=n;i++){
// for(int j=1;j<=m;j++){
// cout<<g[i][j]<<" ";
// }
// cout<<endl;
// }
if(check_color()){
v.clear();
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
v.push_back(g[i][j]);
}
}
se.insert(v);
}
if(check2()){
v.clear();
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
v.push_back(g[i][j]);
}
}
se2.insert(v);
}
return;
}
if(g[x][y]){
if(y == m)
dfs(x+1,1);
else
dfs(x,y+1);
}else{
if(y+1 <= m && !g[x][y+1]){
for(int i=1;i<=2;i++){
g[x][y+1] = i;
g[x][y] = i;
if(y == m){
dfs(x+1,1);
}else{
dfs(x,y+1);
}
g[x][y] = 0;
g[x][y+1] = 0;
}
}
if(x+1 <= n && !g[x+1][y]){
for(int i=1;i<=2;i++){
g[x+1][y] = i;
g[x][y] = i;
if(y == m){
dfs(x+1,1);
}else{
dfs(x,y+1);
}
g[x+1][y] = 0;
g[x][y] = 0;
}
}
}
}
int main(){
n = 3, m =10;
dfs(1,1);
cout<<se2.size()<<endl;
cout<<se.size()<<endl;
set<vector<int> >::iterator it = se2.begin();
vector<int> vv;
while(it != se2.end()){
if(se2.find(*it) != se2.end() && se.find(*it) == se.end() ){
vv = *it;
break;
}
it++;
}
int t = 0;
for(int i=0;i<vv.size();i++){
if(t == 10) {
t = 0;
cout<<endl;
}
cout<<vv[i]<<" ";
t++;
}
cout<<endl;
return 0;
}
//123996我的答案 check函数 是错的?!! check2函数是对的
//101466网上答案 是对的!!
//原因:检查颜色的函数出错 为什么? 不能连等判断。。。。。这语法
//已改正
/*
1 1 1 1 1 1 1 1 1 1
1 2 1 2 2 1 2 2 1 2
1 2 2 2 2 2 1 1 1 2
*/
2016凑平方数
#include<bits/stdc++.h>
using namespace std;
/*
分成几组? k组 1 ~ 10;
每组:dfs搜索0~9这几个没用过的数;
if 完全平方数
1.x+1
2.继续加值 (0不能作为第一个数 单独考虑)
到了k组 先对结果排序存到vector数组中 再set去重(因为递归回溯 结果有大量重复)
注意:必须用long long...用int会出错 因为int的取值范围为:-2147483648 ~ 2147483647
*/
typedef long long ll;
int vis[15];
ll a[15];
vector<ll> v;
int vis2[10];
int k;
int ans = 0;
set<vector<ll> > se;
inline bool check(ll x){
if(x == 9814072356){
int eeeeee = 1;
}
double d = sqrt(x);
return d == (ll)d;
}
//因为递归回溯有大量重复 改成set去重
void dfs(int x,ll cur){
if(x == k){
for(int i=0;i<10;i++){
vis2[i] = 0;
}
for(int i=0;i<k;i++){
ll d = a[i];
if(d == 0) vis2[d] = 1;
else{
while(d){
vis2[d%10] = 1;
d = d/10;
}
}
}
for(int i=0;i<=9;i++){
if(!vis2[i]) return;
}
for(int i=0;i<k;i++) v.push_back(a[i]);
sort(v.begin(),v.end());
if(se.find(v) == se.end()){
for(int i=0;i<k;i++) cout<<v[i]<<" ";
cout<<endl;
se.insert(v);
}
v.clear();
ans++;
return;
}
for(int i=0;i<=9;i++){
if(!vis[i]){
vis[i] = 1;
if(cur == 0 && i == 0){//如果是以0开头 并且当前搜索的是一个新的分组(cur值为0) 就直接搜索下一组
a[x] = 0;
dfs(x+1,0);
vis[i] = 0;
continue;
}
ll num = cur*10+i;
if(check(num)){
a[x] = num;
dfs(x+1,0);
} //搜索下一分组
dfs(x,cur*10+i);//继续搜索当前分组
vis[i] = 0;
}
}
}
int main(){
//freopen("out1.txt","w",stdout);
//枚举分组的次数
for(k = 1;k <= 10;k++){
memset(vis,0,sizeof(vis));
dfs(0,0);
}
cout<<ans<<endl;
cout<<se.size()<<endl;
return 0;
}
//3085
//300
2015完美正方形
#include<bits/stdc++.h>
using namespace std;
int n = 47 + 46 + 61;//边长
int a[19] = {2, 5, 9, 11, 16, 17, 19, 21, 22, 24, 26, 30, 31, 33, 35, 36, 41, 50, 52};
int g[500][500];//大正方形地图
int vis[30];
set<int> se;//集合存储正方形最后一行边长数据结果
void fill(int x,int y,int l,int num){
for(int i=x;i<=x+l-1;i++){
for(int j=y;j<=y+l-1;j++){
g[i][j] = num;
}
}
}
bool ok(int x,int y,int l){
if(x+l-1 > n) return false;
if(y+l-1 > n) return false;
for(int i=x;i<=x+l-1;i++){
for(int j=y;j<=y+l-1;j++){
if(g[i][j] != 0) return false;
}
}
return true;
}
bool check(){
return true;
}
void dfs(int x,int y){
if(x == n+1){//递归出口
if(check()){
for(int i=1;i<=n;i++){
se.insert(g[n][i]);//set集合存储最后一层正方形边长数据
}
}
return;
}
if(g[x][y] != 0 ){//当前正方形填充过了
if(y == n)
dfs(x+1,1);//dfs下一个
else
dfs(x,y+1);//dfs下一个
}else{//当前正方形没有填充过
for(int i=0;i<19;i++){//枚举19块正方形
if(!vis[i]){
if(ok(x,y,a[i])){
fill(x,y,a[i],a[i]);//填充正方形成a[i]边长 以(x,y)为左上顶点
vis[i] = 1;
if(y == n){
dfs(x+1,1);//dfs下一个
}else{
dfs(x,y+1);//dfs下一个
}
vis[i] = 0;//回溯
fill(x,y,a[i],0);//填充正方形成0 以(x,y)为左上顶点
}else{
break;//剪枝 因为a数组按顺序排的 当前边长不行 后面边长更不行了
}
}
}
}
}
int main(){
fill(1,1,47,47);//填充以(1,1)为左上顶点的正方形 边为47
fill(1,47+1,46,46);
fill(1,47+46+1,61,61);
dfs(1,1);//从(1,1)点开始搜索
set<int>::iterator it = se.begin();
while(it!=se.end()){
cout<<*it<<" ";
it++;
}
return 0;
}
//30 33 41 50