7-10 那就别担心了 (30 分) - 记忆化拿最后2分

题目介绍

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

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

输入格式：

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

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

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

输出格式：

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

题目保证输出数据不超过 109。

输入样例 1：

7 8
7 6
7 4
6 5
4 1
5 2
5 3
2 1
3 1
7 1

输出样例 1：

3 Yes

输入样例 2：

7 8
7 6
7 4
6 5
4 1
5 2
5 3
6 1
3 1
7 1

输出样例 2：

3 No

 

注解

如果直接开启莽夫搜索模式，结果是28分。

有一个点超时。

认真考虑就会发现，其实在自己的搜索过程中，有很多解被算了很多遍。

比如这样一种情况：

 

 如果你是直接开dfs，每走到一次B，都计数器+1，那么在从2走向5，以及从3走向5，5再向后走时，5到B的路径数就会被计算两遍。

那我能不能只计算一遍，以后再遇到就直接使用不计算了呢？

当然可以，这就是“记忆化搜索”啦！

具体就请看下面代码吧。

代码

#include<bits/stdc++.h>
using namespace std;

//可以按照树来存储
//虽然子节点有交叉，但是有向图还是比较特殊的
vector<int> child[505]; 
int A,B;

int ans[505];		//从i到B的推理路径数量，最后输出的是ans[A] 
int flag = 1;		//默认认为是可以逻辑自洽
int dfs(int now)
{
	if(now!=B && child[now].size()==0)    //走到了无法再走，但却并不是B，说明不是逻辑自洽
	{
		flag = 0;
		return 0;
	}
	
	if(ans[now]!=0)            //如果已经被计算过了，就可以直接调用！
	{
		return ans[now];
	}
	
	int result = 0;		//从now到B的路径数量
	
	for(int i=0;i<child[now].size();i++)
	{
		result += dfs(child[now][i]);
	}
	
	ans[now] = result;
	return ans[now];
}

int main()
{
	int N,M;
	cin>>N>>M;
	
	int s1,s2;
	for(int i=1;i<=M;i++)	
	{
		cin>>s1>>s2;
		child[s1].push_back(s2);
	}
	
	cin>>A>>B;
	
	ans[B] = 1;
	dfs(A);
	
	cout<<ans[A];
	if(flag==1)	cout<<" Yes\n";
	else		cout<<" No\n";
	
	return 0;
}