dfs五颜六色求条
查看原帖
dfs五颜六色求条
1393199
Folden_xiaoming楼主2024/12/16 19:28

记录 f数组记录有没有开灯,h数组记录有没有来过。 node连接开关和灯

#include<bits/stdc++.h>
using namespace std;
#define int long long
struct node{
	int a,b,c,d;
}ls[210000];
int ans=0,h[200][200],f[200][200],n,m;
void print(){
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cout<<h[i][j]<<" ";
		}
		cout<<endl;
	}
	cout<<endl;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cout<<f[i][j]<<" ";
		}
		cout<<endl;
	}
	cout<<endl<<endl;
}
void dfs(int i,int j){
	if(f[i][j]==0||h[i][j]==1)return ;
	if(i<=0||j<=0||i>n||j>n)return ;//边界
	for(int k=0;k<m;k++){//遍历
		if(ls[k].a==i&&ls[k].b==j&&f[ls[k].c][ls[k].d]==0)f[ls[k].c][ls[k].d]=1;
	}
	//cout<<i<<" "<<j<<endl;
	//print();
	h[i][j]=1;
	dfs(i,j+1);
	dfs(i,j-1);
	dfs(i+1,j);
	dfs(i-1,j);
	h[i][j]=0;
	return;
}
signed main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
	cin>>n>>m;
	for(int i=0;i<m;i++)cin>>ls[i].a>>ls[i].b>>ls[i].c>>ls[i].d;
	//for(int i=0;i<m;i++)cout<<ls[i].a<<" "<<ls[i].b<<" "<<ls[i].c<<" "<<ls[i].d<<"\n";
	//cout<<endl;
	f[1][1]=1;
	dfs(1,1);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(f[i][j]==1){
				ans++;
			}
		}
	}
	cout<<ans;
    return 0;
}
2024/12/16 19:28
加载中...