【差分约束】 P3275 [SCOI2011]糖果

这道题目的差分约束+SLF优化

还是有一个点会TLE，所以我们要进行特判

op=1或op=3时u=v要输出-1
 

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
const int inf=0x3f;
int n,m,cnt;
int head[maxn],dis[maxn],update[maxn],vis[maxn];
struct edge
{
	int to,nxt,v;
}G[maxn<<3];
deque <int> q;
void add(int x,int y,int z)
{
	G[++cnt].to=y; G[cnt].v=z; G[cnt].nxt=head[x]; head[x]=cnt;
}
bool spfa()
{
	int s=0;
	for(int i=1;i<=n;i++) dis[i]=-inf;
	dis[s]=0; vis[s]=1; update[s]=1;
	q.push_front(s);
	while(!q.empty())
	{
		int u=q.front();
		q.pop_front(); vis[u]=0;
		for(int i=head[u];i;i=G[i].nxt)
		{
			int to=G[i].to;
			int v=G[i].v;
			if(dis[to]<dis[u]+v)
			{
				dis[to]=dis[u]+v;
				update[to]++;
				if(update[to]>=n+1) return true;
				if(!vis[to])
				{
					if(q.size() && dis[q.front()]<dis[to]) q.push_back(to);
					else q.push_front(to);
					vis[to]=1;
				}
			}
		}
	}
	return false;
}
int main()
{
	freopen("a.in","r",stdin);	
	scanf("%d%d",&n,&m);
	int op,u,v;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&op,&u,&v);
		if(op==1) {add(u,v,0); add(v,u,0); continue;}
		if(op==2) {if(u==v) {printf("-1"); return 0;}add(u,v,1); continue;}
		if(op==3) {add(v,u,0); continue;}
		if(op==4) {add(v,u,1);if(u==v) {printf("-1"); return 0;}continue;}
		if(op==5) {add(u,v,0); continue;}
	}
	for(int i=1;i<=n;i++) add(0,i,1);
	if(spfa()) 
	{
		printf("-1"); return 0;
	}
	long long ans=0;
	for(int i=1;i<=n;i++)
		ans+=(1ll*dis[i]);
	printf("%lld",ans);
	return 0;
}