FJNU校赛B题(dfs + bfs)

题意:

给出几个城市,以及街道,现在有几座城市着火了,尼克扑灭其中其中一些.然后剩下的火每天会往相邻的城市蔓延;问怎么选择扑灭的城市使全部烧完用时最长,输出时间;

思路:

dfs找到所有可能的扑灭情况,每种情况bfs算出要几天;

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;

int fire[10];
int vis[10];
int ans, n, m, l, p;
vector<int> v[105];

struct point {
	int pos;
	int step;
};
int bfs();
void dfs(int cur, int tar) {
	if(cur == p) {
//		for(int i = 0 ; i < l; i++) {
//			printf("%d ",vis[i]);
//		}
//		printf("\n");
		int tmp = bfs();
//		printf("%d\n",tmp);
		if(tmp > ans)
			ans = tmp;
		return;
	}
	if(tar >= l)
		return ;
	vis[tar] = 1;
	dfs(cur + 1,tar + 1);
	vis[tar] = 0;
	dfs(cur, tar + 1);
	return ;
}
int bfs() {
	queue<point> q;
	int viss[105];
	memset(viss, 0, sizeof(viss));
	while(!q.empty())
		q.pop();
	for(int i = 0; i < l; i++) {
		if(!vis[i]) {
			point a;
			a.pos = fire[i];
			a.step = 0; 
			q.push(a);
			viss[fire[i]] = 1;
		}
	}
	int res = 0;
	while(!q.empty()) {
		point cur = q.front();
		if(cur.step > res) {
			res = cur.step;
		}
		q.pop();
		for(int i = 0; i < v[cur.pos].size(); i++) {
			if(viss[v[cur.pos][i]])
				continue;
			point a;
			a.pos = v[cur.pos][i];
			a.step = cur.step + 1;
			q.push(a);
			viss[v[cur.pos][i]] = 1;		
		}
	}
	return res;
}
int main() {
	int t;
	int cas = 1;
	scanf("%d",&t);
	while(t--) {
		ans = 0;
		memset(vis, 0, sizeof(vis));
		scanf("%d%d%d%d",&n, &m, &l, &p);
		for(int i = 0; i <= n; i++) {
			v[i].clear();
		}
		for(int i = 0; i < l; i++) {
			scanf("%d",&fire[i]);
		}
		int a,b;
		for(int i = 0; i < m; i++) {
			scanf("%d%d",&a,&b);
			v[a].push_back(b);
			v[b].push_back(a);
		}
		dfs(0,0);
		printf("Case %d: %d\n",cas++, ans + 1);
	}
}