2020年团体程序设计天梯赛 L3-1 那就别担心了 (30分)

下图转自“英式没品笑话百科”的新浪微博 —— 所以无论有没有遇到难题，其实都不用担心。

博主将这种逻辑推演称为“逻辑自洽”，即从某个命题出发的所有推理路径都会将结论引导到同一个最终命题（开玩笑的，千万别以为这是真正的逻辑自洽的定义……）。现给定一个更为复杂的逻辑推理图，本题就请你检查从一个给定命题到另一个命题的推理是否是“逻辑自洽”的，以及存在多少种不同的推理路径。例如上图，从“你遇到难题了吗？”到“那就别担心了”就是一种“逻辑自洽”的推理，一共有 3 条不同的推理路径。

输入格式：
输入首先在一行中给出两个正整数 N（1<N≤500）和 M，分别为命题个数和推理个数。这里我们假设命题从 1 到 N 编号。

接下来 M 行，每行给出一对命题之间的推理关系，即两个命题的编号 S1 S2，表示可以从 S1 推出 S2。题目保证任意两命题之间只存在最多一种推理关系，且任一命题不能循环自证（即从该命题出发推出该命题自己）。

最后一行给出待检验的两个命题的编号 A B。

输出格式：
在一行中首先输出从 A 到 B 有多少种不同的推理路径，然后输出 Yes 如果推理是“逻辑自洽”的，或 No 如果不是。

题目保证输出数据不超过 10^9​​ 。

开始想用DFS遍历一遍图，到了目标点路径数就加1，但是最后一个样例会超时。

思路：用一个path数组记录每个顶点到目标点的路径条数，每次DFS遍历到节点x就让该节点的path数加上所有相邻点的path数（x->i）。用个vis数组记录访问过的所有顶点，用DFS遍历到该节点时，先判断该节点的path是否为0，不是0就直接返回该节点path数。最后遍历一遍vis数组，如果有访问过的节点path数为0，说明不是“逻辑自洽”图。

代码：

#include<bits/stdc++.h>
#define rep(i, a, n) for(int i=a;i<=n;i++)
using namespace std;
const int maxn = 500 + 2;
int n, m, cnt, a, b;
int vis[maxn], path[maxn];
vector<int> vec[maxn];
int DFS(int x) {
	vis[x] = 1;
	if(path[x]) return path[x];
	for(auto i: vec[x]) {
		path[x] += DFS(i);
	}
	return path[x];
}
int main() 	{
	cin >> n >> m;
	rep(i, 1, m) {
		cin >> a >> b;
		vec[a].push_back(b);
	}
	cin >> a >> b;
	path[b] = 1;
	cnt = DFS(a);
	int flag = 1;
	rep(i, 1, n) {
		if(vis[i] && !path[i]) {
			flag = 0;
			break;
		}
	}
	cout << cnt << " ";
	if(!flag)
		cout << "No" << endl;
	else
		cout << "Yes" << endl;
}