CodeForces - 999E Reachability from the Capital(贪心 + dfs暴力)

题目:

传送门

思路:

网上博客好像不是虚拟边就是缩点,其实只要每次对与首都不链接的点dfs一次,找到能经过点最多的点,将其与首都连边就可以了,复杂度分析其实是和虚拟边的复杂度相同.
要加个小优化,因为可能最多要进行 n² 次 dfs , 我们直接更新标记数组会T, 我们考虑新增一个 now变量,每次dfs前 now++ ,通过 标记数组是否等于 now 来判断这次dfs 是否经过该点.(这样我们就不用了每次初始化数组了).

Ac_Code:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <stack>
#include <bitset>
#include <vector>
#include <map>
#include <string>
#include <cstring>
#define fir first
#define sec second
using namespace std;

const int maxn = 5e3+7;
const int INF = 1e9+7;

int n,m,s,e,now;
int edge[maxn<<1],nex[maxn<<1],head[maxn];
int vis[maxn],inque[maxn];

void add(int l,int r) {
	edge[++e] = r;
	nex[e] = head[l];
	head[l] = e;
}

void DFS(int u) {
	vis[u] = 1;
	for(int i=head[u]; i; i=nex[i]) {
		int v = edge[i];
		if( vis[v] == 1 ) continue;
		DFS(v);
	}
}

int dfs(int u) {
	int sum=1;
	inque[u] = now;
	for(int i=head[u];i;i=nex[i]) {
		int v = edge[i];
		if(inque[v]==now || vis[v]) continue;
		sum += dfs(v);
	}
	return sum;
}

int main() {
	scanf("%d%d%d",&n,&m,&s);
	for(int i=0;i<m;i++) {
		int l,r;
		scanf("%d%d",&l,&r);
		add(l,r);
	}
	int ans = 0,max,num = -1;
	DFS(s);
	while(true) {
		for(int i=1;i<=n;i++) {
			if(vis[i] == 1) continue;
			now++;
			int cur = dfs(i);
			if(num == -1 || max < cur) {
				max = cur;
				num = i;
			}
		}
		if(max == 0) break;
		ans++;
		DFS(num);
		max = 0;
	}
	printf("%d\n",ans);
	return 0;
}