牛客寒假算法基础集训营4, F题Applese 的QQ群(拓扑排序 与二分查找)

原题链接

https://ac.nowcoder.com/acm/contest/330/F

Applese 有一个QQ群。在这个群中，大家互相请教问题。如 b 向 a 请教过问题，就把 a 叫做是 b 的"老板"。这样一个群中就会有很多老板。

同时规定：如果 a 是 b 的老板，b 是 c 的老板，那么 a 也是 c 的老板。

为了不破坏群里面和谐交流的氛围，Applese 定了一个群规：不允许出现 a 既是 b 的老板， b 又是 a 的老板。

你需要帮助 Applese 判断大家是否遵守了群规。

输入描述:

第一行两个整数 n, m，表示群里的人数以及请教问题的数量。
接下来 m 行，每行两个整数 a, b，表示 a 是 b 的"老板"，即 b 向 a 请教了一个问题。
注：无论是否违反了群规，a 都会成为 b 的老板。

输出描述:

对于每次提问，输出一行"Yes"表示大家都遵守了群规，反之输出"No"。

示例1

输入

复制

4 4
1 2
2 3
3 1
1 4

输出

复制

Yes
Yes
No
No

备注:

1≤n≤1051≤n≤105
1≤m≤2⋅1051≤m≤2⋅105
1≤a,b≤n1≤a,b≤n

 

解题思路

开始脑袋一抽， 见这种题目就并查集， 然而， 并查集只能绑定一个老板， 而一个人可有多个老板， 这样就错了， 题目简单看来就是找有没有形成环， 这个用拓扑排序解决， 而题意可知， 答案为前一串Yes和后面一串No组成， 找到最后一个Yes就可以了， 这个用二分查找解决。

#include <iostream>
#include <cstdio>
#include<string.h>
#include<vector>
#include<queue>
using namespace std;

const int maxn = 1e5 + 10;
int n, m;
vector<pair<int, int>> e;
vector<int> v[maxn];
int in[maxn];

bool check(int mid)//判断mid之前是否出现环， 拓扑排序
{
	for(int i = 0; i <= n; i++)
		v[i].clear(), in[i] = 0;

	for(int i = 0; i < mid; i++)// 这里用<的原因为e中保存的边从0开始
	{
		v[e[i].first].push_back(e[i].second);
		in[e[i].second]++;     //in数组记录入度
	}
	queue<int> q;

	for(int i = 1; i <= n; i++)
		if(!in[i]) q.push(i);   //入度为零的点用q保存
	
	int tot = 0;
	while(!q.empty())
	{
		int x = q.front();
		q.pop();
		tot++;
		for(int i = 0; i < v[x].size(); i++)
		{
			int y = v[x][i];
			if(--in[y] == 0) q.push(y);  //除去入度为零的点x， 并由x至与其相连的点的入度减一
		}

	}

	return tot == n;
}

int main()
{
	scanf("%d %d", &n, &m);
	int u, v;
	for(int i = 0; i < m; i++)
	{
		scanf("%d %d", &u, &v);
		e.push_back(make_pair(u, v));//e来保存有向边
	}
	int l = 1, r = m, mid, ans;
	while(l <= r)  //二分查找最后一个yes的位置， 用ans记录
	{
		mid = (l + r) >> 1;
		if(check(mid)) l = mid + 1, ans = mid;
		else
			r = mid - 1;
	}
	for(int i = 1; i <= ans; i++)
		puts("Yes");
	for(int i = ans; i < m; i++)
		puts("No");
}