洛谷 P2853 Cow Picnic S（DFS）

https://www.luogu.com.cn/problem/P2853

题目大意

k头牛在n个点上，m条有向边，统计那些所有牛都能到达的点的个数

思路

很容易就想到，以每头牛为起点去深搜，能走到的地方数值加1，最后那些数值为k的点就是所有牛都能走到的点。

ps: 本来还顺着这样想了一个优化，如果在牛 a 的路径上有另外一头牛 b，那么 b能走到的路径必定包含在a能走到的以内，那么我dfs加一个参数num，如果多碰到一头牛，num++，这样在牛a的深搜中就把牛b的也走完了，再标记一下后面不用走牛b了，然后走到的点的数值变成加num，深搜完了后再判断一下恢复num，感觉挺可行的，但是写出来却wa了，不知道自己是想法有误还是代码敲错。

以下为ac代码

代码

#include 
using namespace std;
const int N = 1005;

int k, n, m, ans;
vector <int> g[N];
vector <int> cow;
int flag[N];
int f[N];

void dfs(int x) {
	flag[x] = 1;
	f[x]++;
	for (int i = 0; i < g[x].size(); i++) {
		int fp = 0;
		if (flag[g[x][i]]) continue;
		dfs(g[x][i]);
	}
	return ;
}

int main() {
	cin >> k >> n >> m;
	for (int i = 0; i < k; i++) {
		int c; cin >> c;
		cow.push_back(c);
	}
	for (int i = 0; i < m; i++) {
		int u, v;
		cin >> u >> v;
		g[u].push_back(v);
	}
	for (int i = 0; i < cow.size(); i++) {
		memset(flag, 0, sizeof(flag));
		dfs(cow[i]);		
	}
	for (int i = 1; i <= n; i++) {
		if (f[i] == k) ans++;
	}
	cout << ans << endl;
	return 0;
}