dfs练习

题目都在蓝桥云课题库中。

1.全排列（入门）

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int visit[20];
int a[20];
int n;
void dfs(int cur)
{
	if(cur==n)//因为dfs(0)开始 
	{
		for(int i=0; i<n; i++)
		{
			cout<<a[i];
		}
		cout<<endl;
		return;
	}
	for(int i=1; i<=n; i++)//需要排列的数 
	{
		if(visit[i]==0)
		{
			a[cur]=i;
			visit[i]=1;//标记已使用 
			dfs(cur+1);
			visit[i]=0;//回溯，标记未使用 
		}
	}
}

int main()
{
	cin>>n;
	memset(visit,0,sizeof(visit));
	dfs(0);
	return 0;
}

2.黄金树

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n;
int w[100010];
int sum=0;
struct tree
{
	int l,r;
} t[100010];

void dfs(int cur,int x)//x是黄金指数
{
	if(x==0)
	{
		sum+=w[cur];
	}

	if(t[cur].l!=-1)
	{
		dfs(t[cur].l,x+1); //遍历下一层树，左孩子
	}
	if(t[cur].r!=-1)
	{
		dfs(t[cur].r,x-1);
	}

}

int main()
{
	cin>>n;
	for(int i=1; i<=n; i++)//注意在树的定义中,子节点的索引必须从1开始,所以从i=1开始
	{
		cin>>w[i];
	}
	for(int i=1; i<=n; i++)
	{
		cin>>t[i].l>>t[i].r;
	}
	dfs(1,0);
	cout<<sum<<endl;
	return 0;
}

3.七步诗

#include<bits/stdc++.h>
using namespace std;
int n,m;
int sum=0;
char a[410][410];
const int dx[8]= {-1,1,-2,2,-2,2,-1,1}; // 可移动的横向距离
const int dy[8]= {2,2,1,1,-1,-1,-2,-2}; // 可移动的纵向距离
void dfs(int x,int y)
{
	if(x<0||y<0||x>=n||y>=m)//终止条件
	{
		return;
	}
	if(a[x][y]=='b')
	{
		sum++;
	}
	a[x][y]='1';//标记已经被访问(不管吃没吃到豆子，都不需要再走了)
	for(int i=0; i<8; i++) //遍历选择，有8个方向可以选择
	{
		int x1,y1;//下一个落脚点
		x1=dx[i]+x;
		y1=dy[i]+y;
		if(x1>=0&&y1>=0&&x1<n&&y1<m&&a[x1][y1]!='x'&&a[x1][y1]!='1')//处理节点操作
		{
			dfs(x1,y1);//递归
			//不需要往回走，所以无回溯过程
		}
	}
}


int main()
{
	int x,y; 
	cin>>n>>m;
	for(int i=0; i<n; i++)
	{
		for(int j=0; j<m; j++)
		{
			cin>>a[i][j];
			if(a[i][j]=='S')
			{
				x=i;
				y=j;
			}
		}
	}
	dfs(x,y);
	cout<<sum<<endl;
	return 0;
}

4.车牌

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int sum=0;
string s,s1,s2;

void dfs(int cur,char ch,int sign)//cur纵向遍历，x横向遍历，sign追踪合法情况
{
	if(sign==3)//不合法
	{
		return;
	}
	if(cur==6)//合法
	{
		sum++;
		return;
	}
	if(cur<3)
	{
		for(int i=0; i<s.size(); i++)
		{
			if(s[i]==ch)//第一次与空字符比没有相同的，所以起始cur，sign都从0开始
			{
				dfs(cur+1,s[i],sign+1);
			}
			else
			{
				dfs(cur+1,s[i],1);
			}
		}
	}
	else
	{
		for(int i=0; i<s1.size(); i++)
		{
			if(s1[i]==ch)
			{
				dfs(cur+1,s1[i],sign+1);
			}
			else
			{
				dfs(cur+1,s1[i],1);
			}
		}
	}
	return ;//超过6位
}

int main()
{
	s="0123456789ABCDEF";
	s1="0123456789";
	dfs(0,'\0',0);//'\0'表示没有字符，即空字符
	cout<<sum<<endl;
	return 0;
}

5. 仙境诅咒

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int x[1010],y[1010];
int n,d;
int visit[1010];

void dfs(int cur)//本质是判断两点之间的距离
{
	visit[cur]=1;//被感染
	for(int i=0; i<n; i++)
	{
		if(visit[i])//已感染的跳过，否则dfs(i)死循环
		{
			continue;
		}
		if(((x[i]-x[cur])*(x[i]-x[cur])+(y[i]-y[cur])*(y[i]-y[cur]))<=(d*d))
		{
			dfs(i);//不需要往回退，所以无回溯
		}
	}
}

int main()
{
	cin>>n;
	for(int i=0; i<n; i++)
	{
		cin>>x[i]>>y[i];
	}
	cin>>d;
	memset(visit,0,sizeof(visit));//表示修仙者开始都未被感染
	dfs(0);//从第一位修仙者开始
	for(int i=0; i<n; i++)
	{
		cout<<visit[i]<<endl;
	}
	return 0;
}

 6.分糖果

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll sum=0;
int a[7];
void dfs(int cur,int g1,int g2)
{
	if(cur==7)
	{
		if(g1==0&&g2==0)
		{
			sum++;
		}
		return;
	}
	for(int i=0; i<=g1; i++)//开始想法 for(int i=2; i<=5; i++)没有考虑到所有情况，两种糖果分别取再组合成2~5
	{
		for(int j=0; j<=g2; j++)
		{
			if(i+j>=2&&i+j<=5)
			{
				dfs(cur+1,g1-i,g2-j);
			}
		}
	}
}

int main()
{
	dfs(0,9,16);
	memset(a,0,sizeof(a));
	cout<<sum<<endl;
	return 0;
}

7.开心

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
string s;
ll k;
ll temp=0,sum=0,max1=0,min1=1e+9;
char a[110];
void dfs(int start,int cur)
{
	if(cur==k)
	{
		for(int i=0; i<s.size(); i++)
		{
			temp=temp*10+(int)(s[i]-'0');//加号前的数字
			if(a[i]=='+')
			{
				sum+=temp;
				temp=0;
			}
		}
		sum+=temp;//最后一个加号后的数字
		max1=max(max1,sum);
		min1=min(min1,sum);
		temp=0,sum=0;
		return;
	}
	for(int i=start; i<s.size()-1; i++) //因为先进行了对加号前的数字计算，所以放加号的位置可以从start==0开始
	{
		a[i]='+';
		dfs(i+1,cur+1);
		a[i]='.';//回溯完成多个选择
	}

}
int main()
{
	cin>>s>>k;
	memset(a,'.',sizeof(a));
	dfs(0,0);
	cout<<max1-min1<<endl;
	return 0;
}