刷题记录:牛客NC20284[SCOI2011]糖果 tarjan+差分约束+拓扑排序

yingjiayu12

于 2023-01-31 22:01:43 发布

阅读量515

点赞数

分类专栏： c++算法文章标签：算法图论 c++

本文链接：https://blog.csdn.net/yingjiayu12/article/details/128823872

版权

c++算法专栏收录该内容

428 篇文章 2 订阅

订阅专栏

传送门:牛客

题目描述:

幼儿园里有N个小朋友，lxhgww老师现在想要给这些小朋友们分配糖果，要求每个小朋友都要分到糖果。但是
小朋友们也有嫉妒心，总是会提出一些要求，比如小明不希望小红分到的糖果比他的多，于是在分配糖果的时
候，lxhgww需要满足小朋友们的K个要求。幼儿园的糖果总是有限的，lxhgww想知道他至少需要准备多少个
糖果，才能使得每个小朋友都能够分到糖果，并且满足小朋友们所有的要求。
如果X=1,表示第A个小朋友分到的糖果必须和第B个小朋友分到的糖果一样多；
如果X=2,表示第A个小朋友分到的糖果必须少于第B个小朋友分到的糖果；
如果X=3,表示第A个小朋友分到的糖果必须不少于第B个小朋友分到的糖果；
如果X=4,表示第A个小朋友分到的糖果必须多于第B个小朋友分到的糖果；
如果X=5,表示第A个小朋友分到的糖果必须不多于第B个小朋友分到的糖果；
输入:
5 7
1 1 2
2 3 2
4 4 1
3 4 5
5 4 5
2 3 5
4 5 1
输出:
11

一道 $t a r jan$ +差分约束+拓扑排序的题目.这种题目比较经典可以仔细理解

首先读完我们的题面,我们会发现这肯定是一道差分约束的题目.题目中几个 $x$ 的限制显然就是告诉我们几个关于 $A, B$ 的不等式.并且对于最终是求所有答案的最小值.根据差分约束来说,我们跑的肯定是最长路.然后我们会发现跑最长路需要 $Sp f a$ ,但是此时我们的 $N, M$ 较大,这就导致直接跑 $Sp f a$ 会被卡.所以我们需要找一个效率更高的算法.

首先是跑最长路的算法,所以我们是不能使用 $D ijk s t r a$ 的,我们继续观察一下我们的题面,我们会发现这道题有一个很强的性质(一般导致正常算法被卡的肯定会出现几个性质来转化方法的).此题也不例外,此题我们化简我们的不等式之后会发现所有边权只有0或者1.也就是全部都是正边的图.试想一下,对于差分约束来说,跑最长路存在正环将会导致无解.对于本题来说,只要我们存在一个环,如果这个环里的所有边权不都为0的话,就意味存在正环(此时直接输出-1).那么对于我们找正环来说,我们可以使用 $t a r jan$ 来进行缩点,缩点结束后判断一下每一个强连通分量里的边是否存在正边即可.并且因为都是正边权的问题,当我们缩点之后得强连通分量肯定都是边权为0的,那么我们从外部经过这个强连通分量之后可以不用顾及这个强连通分量内部的情况,内部的点肯定都是等价的(这个很重要,需要仔细体会,这是本题正确性的保证).

此时我们缩点结束后就是一个简单的DAG了,DAG有很多很强的性质,在本题中我们就可以直接使用拓扑排序来进行跑最长路了.此时我们的复杂度就从 $Sp f a$ 的 $O (nm)$ 降为了 $O (n)$ .并且对于强连通分量中的所有点的最长路距离都是相等的

需要注意的是题目中讲了所有人分到的糖果数大于等于1,这有两种方法可以解决:一种可以在拓扑排序时将初始入度为0的点的dis值赋值为1.另外一种可以借用差分约束,所有值大于等于1,就存在了一个不等式,所以我们可以设置一个0点,向所有的点连一个值为1的边(注意,此步骤需要放在tarjan前面)

注意使用longlong

下面是具体的代码部分:(代码量较大)

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
const double eps=1e-8;
#define	int_INF 0x3f3f3f3f
#define ll_INF 0x3f3f3f3f3f3f3f3f
int n,k;
struct Node{
	int v,w;
};
vector<Node>edge[maxn];
int dfn[maxn],dfncnt,low[maxn],vis[maxn],color[maxn],num;
stack<int>s;
void tarjan(int u) {
	dfn[u]=low[u]=++dfncnt;
	s.push(u);vis[u]=1;
	for(int i=0;i<edge[u].size();i++) {
		int v=edge[u][i].v;
		if(!dfn[v]) {
			tarjan(v);
			low[u]=min(low[u],low[v]);
		}else{
			if(vis[v]) {
				low[u]=min(low[u],low[v]);
			}
		}
	}
	if(dfn[u]==low[u]) {
		num++;
		while(s.top()!=u) {
			color[s.top()]=num;vis[s.top()]=0;
			s.pop();
		}
		color[u]=num;vis[u]=0;
		s.pop();
	}
}
/*====================================================================*/
vector<Node>edge2[maxn];int in[maxn];int dis[maxn];
void topo() {
	queue<int>q;
	for(int i=1;i<=num;i++) {
		if(in[i]==0) {
			q.push(i);
		}
	}
	while(!q.empty()) {
		int u=q.front();q.pop();
		for(int i=0;i<edge2[u].size();i++) {
			int v=edge2[u][i].v;
			dis[v]=max(dis[v],dis[u]+edge2[u][i].w);
			in[v]--;
			if(in[v]==0) q.push(v);
		}
	}
}
int main() {
	n=read();k=read();
	for(int i=1;i<=k;i++) {
		int x=read(),a=read(),b=read();
		if(x==1) {
			edge[a].push_back({b,0});
			edge[b].push_back({a,0});
		}
		else if(x==2) edge[a].push_back({b,1});
		else if(x==3) edge[b].push_back({a,0});
		else if(x==4) edge[b].push_back({a,1});
		else if(x==5) edge[a].push_back({b,0});
	}
	for(int i=1;i<=n;i++) {
		edge[0].push_back({i,1});
	}	
	for(int i=0;i<=n;i++) {
		if(!dfn[i]) tarjan(i);
	}
	for(int i=0;i<=n;i++) {//建缩点完之后的边
		for(int j=0;j<edge[i].size();j++) {
			int v=edge[i][j].v;
			if(color[i]==color[v]) {
				if(edge[i][j].w) {
					puts("-1");
					return 0;
				}
			}else {
				edge2[color[i]].push_back({color[v],edge[i][j].w});
				in[color[v]]++;
			}
		}
	}
	topo();
	ll ans=0;
	for(int i=1;i<=n;i++) {
		ans+=(1ll*dis[color[i]]);
	}
	cout<<ans<<endl;
	return 0;
}