简单的差分约束,涨姿势。
题意:有n头牛。有ml个关系:每个关系A,B,C表示B距离A不能超过C。有md个关系:每个关系A,B,C表示B距离A至少为C。为牛1和牛n的距离是最大是多少,可能是任意的,也可能是无解的。
思路:对于每个关系B-A<=C,建一条边A->B,权值为C;对于每个关系B-A>=C,建一条边B->A,权值为-C。用spfa求最短路,若最后求得dis[n]=INF,则距离是任意的,若图中有负权回路,则无解。
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <queue>
using namespace std;
#define maxn 11000
int vis[11000],n,ml,md,num;
int b[maxn<<2],in[maxn],next[maxn<<2],val[maxn<<2],p[maxn],dis[maxn];
void add(int x,int y,int v)
{
num++;
b[num]=y;
val[num]=v;
next[num]=p[x];
p[x]=num;
}
int spfa()
{
memset(vis,0,sizeof(vis));
int cur,tot;
queue<int>q;
for(int i=1;i<=n;i++)
{
dis[i]=0x3f3f3f3f;
}
dis[1]=0;
vis[1]=1;
q.push(1);
in[1]=1;
while(!q.empty())
{
cur=q.front();
q.pop();
vis[cur]=0;
for(int e=p[cur];e;e=next[e])
{
if(dis[b[e]]>dis[cur]+val[e])
{
dis[b[e]]=dis[cur]+val[e];
if(!vis[b[e]])
{
vis[b[e]]=1;
q.push(b[e]);
in[b[e]]++;
if(in[b[e]]>n) return -1;
}
}
}
}
if(dis[n]==0x3f3f3f3f) return -2;
return dis[n];
}
void init()
{
for(int i=1;i<=n;i++)
{
p[i]=0;
}
num=0;
}
int main()
{
int a,b,c;
while(scanf("%d%d%d",&n,&ml,&md)!=EOF)
{
init();
while(ml--)
{
scanf("%d%d%d",&a,&b,&c);
add(a,b,c);
}
while(md--)
{
scanf("%d%d%d",&a,&b,&c);
add(b,a,-c);
}
printf("%d\n",spfa());
}
return 0;
}