洛谷 P1262 间谍网络（强连通分量，缩点）

间谍网络

题目描述

由于外国间谍的大量渗入，国家安全正处于高度的危机之中。如果 A 间谍手中掌握着关于 B 间谍的犯罪证据，则称 A 可以揭发 B。有些间谍收受贿赂，只要给他们一定数量的美元，他们就愿意交出手中掌握的全部情报。所以，如果我们能够收买一些间谍的话，我们就可能控制间谍网中的每一分子。因为一旦我们逮捕了一个间谍，他手中掌握的情报都将归我们所有，这样就有可能逮捕新的间谍，掌握新的情报。

我们的反间谍机关提供了一份资料，包括所有已知的受贿的间谍，以及他们愿意收受的具体数额。同时我们还知道哪些间谍手中具体掌握了哪些间谍的资料。假设总共有 $n$ 个间谍（$n$ 不超过 $3000$），每个间谍分别用 $1$ 到 $3000$ 的整数来标识。

请根据这份资料，判断我们是否有可能控制全部的间谍，如果可以，求出我们所需要支付的最少资金。否则，输出不能被控制的一个间谍。

输入格式

第一行只有一个整数 $n$。

第二行是整数 $p$。表示愿意被收买的人数，$1\le p\le n$。

接下来的 $p$ 行，每行有两个整数，第一个数是一个愿意被收买的间谍的编号，第二个数表示他将会被收买的数额。这个数额不超过 $20000$。

紧跟着一行只有一个整数 $r$，$1\le r\le 8000$。然后 $r$ 行，每行两个正整数，表示数对 $(A,B)$，$A$ 间谍掌握 $B$ 间谍的证据。

输出格式

如果可以控制所有间谍，第一行输出 YES，并在第二行输出所需要支付的贿金最小值。否则输出 NO，并在第二行输出不能控制的间谍中，编号最小的间谍编号。

样例 #1

样例输入 #1

3
2
1 10
2 100
2
1 3
2 3

样例输出 #1

YES
110

样例 #2

样例输入 #2

4
2
1 100
4 200
2
1 2
3 4

样例输出 #2

NO
3

1、先用tarjan算法求出图的连通块，做缩点处理。
2、统计入度为0的连通块
3、如果存在入度为0的并且没有可以收买的间谍的连通块，那么输出NO，
	用堆来存储这些连通块，所有间谍的编号，选出编号最小的。
4、如果每一个入度为0的连通块，都有可以收买的间谍，就从这个连通块选一个
	赎金最小的间谍来收买。 当然，也是用堆来存储 <赎金，编号>。 
	用cost 来累加所有的花费。 答案输出 "YES", cost

#include <bits/stdc++.h>
using namespace std;
const int MaxN = 1e5 + 10;
int money[MaxN];
vector<int> e[MaxN];
int dfn[MaxN], low[MaxN], tot;
int stk[MaxN], instk[MaxN], top;
int scc[MaxN], siz[MaxN], cnt;
// scc[x] 表示x节点属于的强联通编号
// siz[i] 表示第i个强连通分量的大小
int n, p, r;
int degree_in[MaxN];
// degree_in[i] 表示第i个 强连通块的入度

priority_queue<pair<int, int> > pq[MaxN];	//记录赎金最小的间谍
priority_queue<int> pq1[MaxN];	//记录间谍的最小编号


void tarjan(int x)
{
	// 入x时，盖戳，入栈
	dfn[x] = low[x] = ++tot;
	stk[++top] = x, instk[x] = 1;
	for(int y : e[x])
	{
		if(!dfn[y])//若y尚未访问
		{
			tarjan(y);
			low[x] = min(low[x], low[y]);
		}else if(instk[y]){	// 若y已经访问且在栈中
			low[x] = min(low[x], dfn[y]);
		}
	}

	//离x时， 记录scc
	if(dfn[x] == low[x]) // 若x是scc的根
	{
		int y; ++cnt;
		do{
			y = stk[top--]; instk[y] = 0;
			if(money[y])
				pq[cnt].push(make_pair(-money[y], y));
			pq1[cnt].push(-y);
			scc[y] = cnt; // scc编号
			++siz[cnt];
		}while(y != x);
	}
}

int main()
{
	scanf("%d", &n);
	scanf("%d", &p);
	int id, value, x, y;
	for(int i = 1; i <= p; ++i)
	{
		scanf("%d%d", &id, &value);
		money[id] = value;
	}
	scanf("%d", &r);
	for(int i = 1; i <= r; ++i)
	{
		scanf("%d%d", &x, &y);
		e[x].push_back(y);
	}
	for(int i = 1; i <= n; ++i)
	{
		if(!dfn[i])
		{
			tarjan(i);
		}
	}
	// 暴力遍历每一条边
	for(int i = 1; i <= n; ++i)
	{
		for(int j: e[i])
		{
			if(scc[i] != scc[j])
			{
				degree_in[scc[j]]++;
			}
		}
	}
	//统计入度为0的缩点个数
	int min_index = n + 10;
	long long cost = 0;
	bool flag = true;
	for(int i = 1; i <= cnt; ++i)
	{
		if(degree_in[i] == 0)
		{
			if(pq[i].empty()) // 连通块 i 能收买的间谍是 0 个
			{
				min_index = min(min_index, -pq1[i].top());	
				flag = false;
			}else{
				cost += pq[i].top().first * (-1);	
			}
		}
	}
	if(false == flag)
	{
		printf("NO\n");
		printf("%d\n", min_index);
	}else{
		printf("YES\n");
		printf("%lld\n", cost);
	}
	return 0;
}