DMOJ IOI '17 P3 - Toy Train【拓扑排序】

传送：https://dmoj.ca/problem/ioi17p3
参考：https://blog.csdn.net/qq_27327327/article/details/80711824
妙啊……首先题意就是走到一个包含充电点的环里就能赢
因为出度至少是1，所以如果所有点都能到充电点那么全部是先手必胜；否则，不能到充点电的点以及一定能到这些点的点就一定是先手必败（能到的不是先手必胜，因为可能到了之后再出去进入别的环）
能到充点电的点是A支配并且能到至少一个充点电的点或B支配只能到充电点的点，每次求出这个集合，判断如果是全集就退出，否则用补集找到先手必败点删去，然后再对剩下不确定的点做上述操作即可

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
using namespace std;
const int N=5005;
int n,m;
vector<int>e[N],e2[N];
vector<int>wk(int fl,vector<int>a,vector<int>r,vector<int>b)
{
    vector<int>ans(n),d(n);
    queue<int>q;
    for(int i=0;i<n;i++)
        if(r[i]&&b[i])
            q.push(i),ans[i]=1;
    for(int i=0;i<n;i++)
        for(int j=0;j<e[i].size();j++)
            if(b[e[i][j]])
            {
                if(a[i]^fl)
                    d[i]++;
                else
                    d[i]=1;
            }
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=0;i<e2[u].size();i++)
            if(!ans[e2[u][i]]&&b[e2[u][i]])
                if(!(--d[e2[u][i]]))
                {
                    ans[e2[u][i]]=1;
                    q.push(e2[u][i]);
                }
    }
    return ans;
}
vector<int> who_wins(vector<int>a,vector<int>r,vector<int>u,vector<int>v)
{
    n=a.size(),m=u.size();
    for(int i=0;i<m;i++)
        e[u[i]].push_back(v[i]),e2[v[i]].push_back(u[i]);
    vector<int>ans(n);
    for(int i=0;i<n;i++)
        ans[i]=1;
    while(1)
    {
        int fl=1;
        vector<int>b1=wk(1,a,r,ans);
        for(int i=0;i<n;i++)
            if(ans[i]&&!b1[i])
                fl=0;
        if(fl)
            return ans;
        for(int i=0;i<n;i++)
            b1[i]^=1;
        vector<int>b2=wk(0,a,b1,ans);
        for(int i=0;i<n;i++)
            if(b2[i])
                ans[i]=0;
    }
    return ans;
}