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
>=inf) //n和1 无约束 可以无限大
{
cout<<-2<<endl;
return;
}
cout<<d
<<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;
}