POJ - 3169 Layout（多源点建图）

POJ - 3169 Layout（多源点建图）

因为求的是最大距离，所以用最短路，要找出形如x[i]<=x[j]+c的不等式
①序号大的牛不可能在序号小的牛的前面 -> x[i]<=x[i+1]; i+1->i , 0
②好感度 -> x[b]-x[a]<=c -> x[b]<=x[a]+c; a->b , c
③反感度 -> x[b]-x[a]>=c -> x[a]<=x[b]-c. b->a , c

还需要找一个超级源点（记超级源点为x[0]），发现这里求的距离都是牛之间的相对距离，所以联想到令x[0]=0，
又因为求的是形如x[i]<=x[j]+c的不等式，所以将所有点放在0点的左侧就可以得出 x[i]<=x[0]+0的不等式,这样从源点出发可以遍历所有的点，
那么就可以遍历到所有的边。
所以就需要从源点向所有点连一条长度是0的边，（PS：这里可以通过在建立队列的时候将所有的点推入队列来代替）

当存在负环时，不存在方案输出-1，
因为求的是一个相对距离，所以记第i头牛到第1头牛的距离为dist[i]，那么第1头牛到第N头牛之间的距离就是dist[N],当dist[N]=INF时输出-2.

#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
const int N = 1010, M = 20010;
int n,ml,md;
int h[N],w[M],e[M],ne[M],idx;
int dist[N],cnt[N];
bool st[N];
void add(int a,int b,int c)
{
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
bool Spfa(int s)
{
    queue<int> q;
    memset(dist,0x3f,sizeof dist);
    memset(st,false,sizeof st);
    memset(cnt,0,sizeof cnt);
    
    q.push(s);
    dist[s]=0;
    st[s]=true;
    
    while(!q.empty())
    {
        int t=q.front();q.pop();
		st[t]=false;
		for(int i=h[t];~i;i=ne[i])
		{
			int j=e[i];
			if(dist[j]>dist[t]+w[i])
			{
				dist[j]=dist[t]+w[i];
				cnt[j]=cnt[t]+1;//这条路上点的个数
				if(cnt[j]>=n) return true;
				if (!st[j])     // 如果队列中已存在j，则不需要将j重复插入
				{
					q.push(j);
					st[j]=true;
				}
			}
		}
    }
    return false;
}
int main()
{
	memset(h,-1,sizeof h);
	cin>>n>>ml>>md;
	for(int i=1;i<=ml;i++)
	{
		int a,b,c;cin>>a>>b>>c;
		if(a>b) swap(a,b);
		add(a,b,c);
	}
	for(int i=1;i<=md;i++)
	{
		int a,b,c;cin>>a>>b>>c;
		if(a>b) swap(a,b);
		add(b,a,-c);
	}
	for(int i=1;i<n;i++) add(i+1,i,0);
	
	if(Spfa(n)) cout<<"-1"<<endl;
	else
	{
		Spfa(1);
		cout<<(dist[n]==0x3f3f3f3f?-2:dist[n])<<endl;
	}
	return 0;
}