差分约束——P3275 [SCOI2011]糖果

题目

P3275 [SCOI2011]糖果

算法分析

设 $A,B$ 两个小朋友分到的糖果数分别为 $a,b$

1. 当 $x=1$ 时，$a=b$ ——> $a-b>=0$ && $a-b<=0$
2. 当 $x=2$ 时，$a<b$ ——> $b>=a+1$
3. 当 $x=3$ 时，$a>=b$ ——> $a>=b$
4. 当 $x=4$ 时，$a>b$ ——> $a>=b+1$
5. 当 $x=5$ 时，$a<=b$ ——> $b>=a$

于是根据差分约束系统，建立边 $a->b$ 或 $b->a$ ，边权分别为 $0$或$1$，使用图论相关算法，计算最长路即可。
注意要判断图中是否有环，遍历结点是统计个数，若个数大于等于总结点数，则有环。

具体代码如下：

C++ Code

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#define rg register
using namespace std;
typedef long long ll;
inline int sread()
{
	int x=0,f=1;char c=getchar();
	while(c>'9'||c<'0') {if(c=='-') f=-1;c=getchar();}
	while(c>='0'&&c<='9') {x=x*10+c-'0';c=getchar();}
	return f*x; 
}
const int maxn=2e5+5;
struct node{
	int next;
	int to;
	int val;
}edg[maxn];
queue <int> q;
int n,k,op,u,v;
int h[maxn],cnt;
int dis[maxn],vis[maxn],ia[maxn];
void add(int u,int v,int val)
{
	cnt++;
	edg[cnt].next=h[u];
	edg[cnt].to=v;
	edg[cnt].val=val;
	h[u]=cnt;
}//建图
int spfa()
{
	while(!q.empty())
	{
		int u=q.front();	q.pop();
		vis[u]=0; ia[u]=0;
		for(rg int i=h[u];i;i=edg[i].next)
		{
			int v=edg[i].to;
			if(dis[v]<dis[u]+edg[i].val)
			{
				dis[v]=dis[u]+edg[i].val;
				ia[i]++;
				if(ia[i]>=n) return 0;//结点个数大于等于n,有环，输出-1；
				if(!vis[v]){vis[v]=1;q.push(v);} 
			}
		}
	}
	return 1;
}//判断是否有环 
int main()
{
	n=sread(); k=sread();
	for(rg int i=1;i<=k;++i)
	{
		op=sread(); u=sread(); v=sread();
		if(op==1)
		{
			add(u,v,0);
			add(v,u,0);
		} 
		if(op==2)	add(u,v,1);
		if(op==3)	add(v,u,0);
		if(op==4)	add(v,u,1);
		if(op==5)	add(u,v,0);
		if((op==2||op==4)&&u==v) 
		{
			printf("-1\n");return 0;
		}
	//如果A,B两个小朋友分的糖果数不能相同，但数据相同，则不满足题意输出-1；
	}
	for(rg int i=1;i<=n;++i)
	{
		vis[i]=1;
		ia[i]=1;
		dis[i]=1;
		q.push(i);
	}
	if(!spfa())
	{
		printf("-1\n");
		return 0;
	}
	ll ans=0;//注意要开 long long
	for(rg int i=1;i<=n;++i)
	{
		ans+=dis[i];
	}
	printf("%lld\n",ans);
	return 0;
} 

反思与总结

1. 要将约束条件转化成相同形式，选择采用最短路算法或是最长路算法。
   若是 $v>=u+val$ 的形式， 则找最长路，输出最小值。
   若是 $v<=u+val$ 的形式， 则找最短路，输出最大值。
2. 可根据特殊情况进行特判，以节省时间和空间。