poj 3169 BellmanFord—差分约束

点击打开链接

#include <iostream>
#include <algorithm>
#include <vector>
#include <cstring>
using namespace std;
const int M = 1100;
const int inf=1000000000; 
int n,ml,md; 
int d[M];
struct edge{
	int to,cost;
	edge(int a,int b):to(a),cost(b)
	{
		
	}
};
vector <edge> g[21000];


bool check()
{
	bool flag=false;
	
		for(int i=1;i<=n;i++)
		{
			for(int j=0;j<g[i].size();j++)
			{
				edge p=g[i][j];
				if(d[p.to]>d[i]+p.cost)
				{
					d[p.to]=d[i]+p.cost; 
					flag=true;
					return true;
				}
			}
		}
		return flag;
}


void solve(int s)
{
	fill(d+1,d+n+1,inf);
	d[s]=0;
	// d(k)[u] 源点s到u最多经过k条边的最短路
	// d(1)[u]=edge[s][u]
	//d(k)[u]=min(d(k-1)[u],min(d(k-1)[j]+edge[j][u])) 
	
	for(int k=1;k<=n;k++) // 无负圈时 最短路最多只含 n-1条边 
	{
		if(k==n) //如果k==n更新说明有负圈 
		{
			if(check())
			{
				cout<<-1<<endl; // 有负圈即 i+1能排到i前面 不符合题意 
				return;
			}
			else
			break;
		} 
		for(int i=1;i<=n;i++)
		{
			for(int j=0;j<g[i].size();j++)
			{
				edge p=g[i][j];
				if(d[p.to]>d[i]+p.cost)
				{
					d[p.to]=d[i]+p.cost; 
				
				}
			}
		}
	}
	if(d[n]>=inf) //n和1 无约束 可以无限大 
	{
		cout<<-2<<endl;
		return;
	}
	cout<<d[n]<<endl;
}



int main()
{
	cin>>n>>ml>>md;
	while(ml--) 
	{	
		int x,y,w;
		cin>>x>>y>>w;
		g[x].push_back(edge(y,w)); //  d[x]+w>=d[y] 
		
	}
	while(md--)
	{
		int  x,y,w;
		cin>>x>>y>>w;
		g[y].push_back(edge(x,-w)); // d[x]+w<=d[y]  -> d[y]+(-w)>=d[x]
	}
	for(int i=1;i<n;i++) 
	{
		g[i+1].push_back(edge(i,0));         // d[i]<=d[i+1] ->      d[i+1]+0>=d[i]
	}
	// u->v有多条边 取最小边 即求最短路 
	solve(1);
	return 0;
}