codeforces 1019C Sergey's problem

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/yzyyylx/article/details/81706250

题面

题意

给出一张有向图，要求在其中选取一个点集，要求：
1.点集中的任意两个点之间都没有边。
2.任何在点集之外的点与点集内的点的最短距离不大于2。

做法

首先正着遍历所有点，如果这个点还没有标记则标记为1，将它能一步到达的且标记为0的点全部标记为-1，表示这些点不能选择，然后倒着遍历一遍，将标记为1的计入答案，并将其能一步到达的点的标记全部改为-1。
可以发现正着遍历之后可能会导致存在a->b，且a，b同时被标记为1，且b

代码

#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
#include<queue>
#define P pair<int,int>
#define mp make_pair
#define fi first
#define se second
#define N 1001000
using namespace std;

int n,m,first[N],ds[N],bb,bj[N];
bool ok[N];
struct Bn
{
    int to,next;
} bn[N<<1];
vector<int>ans;

inline void add(int u,int v)
{
    bb++;
    bn[bb].to=v;
    bn[bb].next=first[u];
    first[u]=bb;
}

int main()
{
    memset(first,-1,sizeof(first));
    int i,j,o,p,q,t;
    cin>>n>>m;
    for(i=1; i<=m; i++)
    {
        scanf("%d%d",&p,&q);
        add(p,q);
    }
    for(i=1; i<=n; i++)
    {
        if(bj[i]) continue;
        bj[i]=1;
        for(p=first[i];p!=-1;p=bn[p].next)
        {
            if(!bj[bn[p].to])
            {
                bj[bn[p].to]=-1;
            }
        }
    }
    for(i=n;i>=1;i--)
    {
        if(bj[i]!=1 || ok[i]) continue;
        ans.push_back(i);
        for(p=first[i];p!=-1;p=bn[p].next)
        {
            bj[bn[p].to]=-1;
        }
    }
    cout<<ans.size()<<endl;
    for(i=0;i<ans.size();i++) printf("%d ",ans[i]);
}