POJ 3660 Cow Contest

最新推荐文章于 2024-07-23 17:36:37 发布

x1uc

最新推荐文章于 2024-07-23 17:36:37 发布

阅读量146

点赞数

文章标签：深度优先算法图论 c++ 青少年编程

本文链接：https://blog.csdn.net/weixin_62645351/article/details/127836503

版权

题面：翻译来自 acwing平台

#include <iostream>
#include <vector>
#include <cstring>
using namespace std;
int n, m;
int ind[105], outd[105];
vector<int> vec_1[105]; // vec_1[x][y] 代表y是x的儿子
vector<int> vec_2[105]; // vec_2[x][y] 代表y是x的父亲
int vis[105][105];      // vis[x][y]==1表示 y是x的儿子
int vis2[105][105];     // vis1[x][y]==1表示 y是x的父亲  vis[x][1...n]+vis[y][1...n]==n-1  表示可以确定x的位置
int vist[105];          // vist[x] 表示x已经被搜索过了
void dfs(int x, int p)
{
    if (vist[p]) //如果vist被搜索过了，因为p的子节点一定是x的子节点，所以直接赋值即可
    {
        for (int i = 1; i <= 100; i++)
        {
            vis[x][i] |= vis[p][i];
        }
        return;
    }
    for (int j = 0; j < vec_1[p].size(); j++)
    {
        vis[x][vec_1[p][j]] = 1;       // x 的儿子标记上
        vis[p][vec_1[p][j]] = 1;       // p的儿子标记上
        dfs(vec_1[p][j], vec_1[p][j]); //先把p深搜了 ，然后x就可以直接使用了
        vist[vec_1[p][j]] = 1;         //搜索完成之后把 搜索过的点标记下
        dfs(x, vec_1[p][j]);           // x直接使用
    }
}
void dfs2(int x, int p)  //同理
{
    if (vist[p])
    {
        for (int i = 1; i <= 100; i++)
        {
            vis2[x][i] |= vis2[p][i];
        }
        return;
    }
    for (int j = 0; j < vec_2[p].size(); j++)
    {
        vis2[x][vec_2[p][j]] = 1;
        vis[p][vec_2[p][j]] = 1;
        dfs2(vec_2[p][j], vec_2[p][j]);
        vist[vec_2[p][j]] = 1;
        dfs2(x, vec_2[p][j]);
    }
}
int main()
{
    cin >> n >> m;
    for (int i = 1; i <= m; i++)
    {
        int x, y;
        cin >> x >> y;
        vec_1[x].push_back(y);
        vec_2[y].push_back(x);
    }
    for (int i = 1; i <= n; i++)
    {
        dfs(i, i);
        vist[i] = 1; //搜索完成之后把 搜索过的点标记下
    }
    memset(vist, 0, sizeof(vist)); //刷新
    for (int i = 1; i <= n; i++)
    {
        dfs2(i, i);
        vist[i] = 1; //搜索完成之后把 搜索过的点标记下
    }
    int res = 0;
    for (int i = 1; i <= n; i++)
    {
        int ans = 0;
        for (int j = 1; j <= n; j++)
        {
            if (vis[i][j] || vis2[i][j])
                ans++;
        }
        if (ans == n - 1)
        {
            res++;
        }
    }
    cout << res << '\n';
    return 0;
}