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

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

超时代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 505;
int cnt1=0,cnt2=0;
vector<int> v[maxn];
int p(int a,int b)
{
    if(a==b||v[a].size()==0){
        cnt2++;
        if(a==b)
            cnt1++;
        return 0;
    }
    for(int i=0;i<v[a].size();i++)
        p(v[a][i],b);
    return 0;
}
int main()
{
    int n,m;
    cin>>n>>m;
    int x,y;
    for(int i=0;i<m;i++)
    {
        cin>>x>>y;
        v[x].push_back(y);
    }
    int a,b;
    cin>>a>>b;
    p(a,b);
    cout<<cnt1;
    if(cnt1==cnt2)
        cout<<" Yes";
    else
        cout<<" No";
    return 0;
}

正确代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 505;
int vis[maxn], path[maxn];//标记要经过的节点   该节点若能到达b标记路径数
int f=1,cnt;
vector<int> v[maxn];//连接各节点
int p(int a)
{
    vis[a]=1;//标记要经过的节点
    if(path[a])
        return path[a];
    for(int i=0;i<v[a].size();i++)
        path[a]+=p(v[a][i]);
    return path[a];
}
int main()
{
    int n,m;
    cin>>n>>m;
    int x,y;
    for(int i=0;i<m;i++)
    {
        cin>>x>>y;
        v[x].push_back(y);
    }
    int a,b;
    cin>>a>>b;
    path[b]=1;
    cnt=p(a);//返回总的路径数
    cout<<cnt;
    for(int i=1;i<=n;i++)
    {
        if(vis[i]&&!path[i])//若经过该编号节点但不能到达b节点
        {
            f=0;
            break;
        }
    }
    if(f)
        cout<<" Yes";
    else
        cout<<" No";
    return 0;
}