codeforces 645 D. Robot Rapping Results Report 二分+拓扑排序

题目链接

我们可以发现， 这是一个很明显的二分+拓扑排序....

如何判断根据当前的点， 是否能构造出来一个唯一的拓扑序列呢。

如果有的点没有出现， 那么一定不满足。 如果在加进队列的时候， 同时加了两个点， 也就是队列的size > 1， 那么也不满足。

如果队列空了之后， 还有的点没有操作过， 那么同样不满足。

然后就没有了

#include <iostream>
#include <vector>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <map>
#include <set>
#include <string>
#include <queue>
#include <stack>
#include <bitset>
using namespace std;
#define pb(x) push_back(x)
#define ll long long
#define mk(x, y) make_pair(x, y)
#define lson l, m, rt<<1
#define mem(a) memset(a, 0, sizeof(a))
#define rson m+1, r, rt<<1|1
#define mem1(a) memset(a, -1, sizeof(a))
#define mem2(a) memset(a, 0x3f, sizeof(a))
#define rep(i, n, a) for(int i = a; i<n; i++)
#define fi first
#define se second
typedef pair<int, int> pll;
const double PI = acos(-1.0);
const double eps = 1e-8;
const int mod = 1e9+7;
const int inf = 1061109567;
const int dir[][2] = { {-1, 0}, {1, 0}, {0, -1}, {0, 1} };
const int maxn = 1e5+5;
int n, m;
int a[maxn], b[maxn], de[maxn], head[maxn], vis[maxn], num, cnt;
struct node
{
    int to, nextt;
}e[maxn*2];
void add(int u, int v) {
    e[num].to = v, e[num].nextt = head[u], head[u] = num++;
}
int check(int mid) {
    mem(de);
    mem1(head);
    mem(vis);
    num = cnt = 0;
    for(int i = 1; i<=mid; i++) {
        de[a[i]]++;
        add(b[i], a[i]);
        if(!vis[a[i]])
            cnt++;
        if(!vis[b[i]])
            cnt++;
        vis[a[i]] = vis[b[i]] = 1;
    }
    if(cnt != n) {
        return 0;
    }
    mem(vis);
    cnt = 0;
    queue <int> q;
    for(int i = 1; i<=n; i++) {
        if(de[i] == 0) {
            q.push(i);
            cnt++;
        }
    }
    if(cnt!=1) {
        return 0;
    }
    num = 0;
    while(!q.empty()) {
        int tmp = q.front(); q.pop();
        num++;
        cnt = 0;
        for(int i = head[tmp]; ~i; i = e[i].nextt) {
            int v = e[i].to;
            de[v]--;
            if(de[v] == 0) {
                q.push(v);
                cnt++;
            }
            if(vis[v])
                return 0;
        }
        if(cnt>1) {
            return 0;
        }
    }
    return num == n;
}
int main()
{
    cin>>n>>m;
    for(int i = 1; i<=m; i++) {
        scanf("%d%d", &a[i], &b[i]);
    }
    int l = 1, r = m, ans = -1;
    while(l<=r) {
        int mid = l+r>>1;
        if(check(mid)) {
            r = mid-1;
            ans = mid;
        } else
            l = mid+1;
    }
    cout<<ans<<endl;
}