bzoj 2330: [SCOI2011]糖果 洛谷P3275

https://www.lydsy.com/JudgeOnline/problem.php?id=2330

https://www.luogu.com.cn/problem/P3275

這題直接把每一個沒有訪問過的點初值設爲1，保證題目說的所有小朋友都分到，然後跑最長路的差分約束就可以了。

找了一晚上錯誤，結果是輸入沒有說a!=b，所以再輸入的時候要判斷一下，不然因爲dis[u]+e[i].l>dis[u]，會更新一次，但是不會push進隊列。

#include<bits/stdc++.h>
using namespace std;

const int maxl=1e5+10;

int n,k,cnt,ans,lim;
long long sum;
int ehead[maxl],num[maxl];
long long dis[maxl];
struct ed
{
	int to,nxt,l;
}e[maxl<<2];
bool in[maxl];
queue <int> q;

inline void add(int u,int v,int l)
{
	e[++cnt].to=v;e[cnt].l=l;
	e[cnt].nxt=ehead[u];ehead[u]=cnt;
}

inline void prework()
{
	ans=1;
	scanf("%d%d",&n,&k);
	int op,a,b;
	for(int i=1;i<=k;i++)
	{
		scanf("%d%d%d",&op,&a,&b);
		if(op==1)
			add(b,a,0),add(a,b,0);
		if(op==2)
			add(a,b,1);
		if(op==3)
			add(b,a,0);
		if(op==4)
			add(b,a,1);
		if(op==5)
			add(a,b,0);
		if(op%2==0 && a==b)
			ans=0; 
	}
}

inline bool spfa(int st)
{
	int u,v;
	q.push(st);dis[st]=1;in[st]=true;
	while(!q.empty())
	{
		u=q.front();q.pop();
		for(int i=ehead[u];i;i=e[i].nxt)	
		{
			v=e[i].to;
			if(dis[u]+e[i].l>dis[v])
			{
				dis[v]=dis[u]+e[i].l;
				if(!in[v])
				{
					q.push(v);num[v]++;
					in[v]=true;
					if(num[v]>=n) return false;
				}
			} 
		}	
		in[u]=false;
	}
	return true;
}

inline void mainwork()
{
	if(ans==0)
		return;
	for(int i=1;i<=n;i++)
	if(dis[i]==0)
	{
		if(!spfa(i))
		{
			ans=0;
			return;
		}
	}
	for(int i=1;i<=n;i++)
		sum+=dis[i];
		
}

inline void print()
{
	if(!ans)
		puts("-1");
	else
		printf("%lld",sum);
}

int main()
{
	//freopen("in.in","r",stdin);
	prework();
	mainwork();
	print();
	return 0;
}