hihocoder #1343 : Stable Members
解释:
一个学习小组,一共有N个学员,一个主管。每个学员都有自己的导师(一个或者多个),导师可以是其他学员也可以是主管。
每周学员都要把自己的学习报告和收到的报告提交给自己的导师,这个团队设计很合理,没有回环(投递出去的报告不会回到自己手中),并且所有的报告最终都会投递到主管那里。
但这个团队中有的学员会因为其他某个学员不工作而导致报告无法提交到主管手中,我们称这种学员为不可靠的。而不受某个学员不工作而影响到报告提交的就是可靠学员。
问题就是找出可靠学员的数量。
输入:
第一行数字是N,学员总数。接下来每行对应1到N学员的导师数量和编号,例如第二行输入(1 0),代表学员1的导师有1个,并且就是主管(0代表主管);第四行输入(2 1 2),代表学员3的导师有两个,分别是学员1和2。
输出:
可靠学员的数量
题意:一个有向无环图,最上游的点只有一个。若删掉一个点,则某些后续点无法与最上游的点连通,则这些后续点为unstable的。要找出所有unstable的点,然后输出剩下的stable点的数量。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
using namespace std;
const int maxn = 1e5 + 10;
vector<int> s[maxn]; //son结点
vector<int> p[maxn]; //parent结点
int color[maxn]; //染成的颜色
int is_unstable[maxn]; //统计unstable的学生的数量
int N;
bool judge(int son, int col) { //判断父节点是否是同一种颜色
for(int i=0;i<p[son].size();i++){
if(color[p[son][i]]!=col){
return false;
}
}
return true;
}
void solve(int rt) { //拓扑排序寻找unstable学生的数量
if (color[rt]) return;
int col = rt;
color[rt] = rt;
queue<int> q;
q.push(rt);
while (!q.empty()) {
int now = q.front();
q.pop();
for (int i = 0; i < s[now].size(); i++) {
int son = s[now][i];
if (judge(son, col)) { //如果son结点所有的parent结点都被染色,而且是一样的颜色时,说明这son结点是unstable
color[son] = col;
q.push(son);
is_unstable[son] = 1;
}
}
}
}
int main() {
scanf("%d", &N);
for (int i = 1; i <= N; i++) {
int n;
scanf("%d", &n);
for (int j = 1; j <= n; j++) {
int x;
scanf("%d", &x);
s[x].push_back(i);
p[i].push_back(x);
}
}
for (int i = 1; i <= N; i++) {
solve(i);
}
int ans = 0;
//cout<<"color"<<endl;
//for(int i=1;i<=N;i++) printf("%d ",color[i]);
//printf("\n");
//cout<<"vis"<<endl;
//for(int i=1;i<=N;i++) printf("%d ",is_unstable[i]);
//printf("\n");
for (int i = 1; i <= N; i++) {
ans+=is_unstable[i];
}
ans = N - ans ;
printf("%d", ans);
//system("pause");
}