题目
一个差分约束系统是这样的:
给出 n 个变量和 m 个约束条件,形如xi-xj<=ck,你需要求出一组解,使得所有约束条件均被满足。
怎么样解这个差分约束系统呢?我们将上面的不等式变形一下:
xi<=xj+ck
容易发现这个形式和最短路中的三角形不等式dis v<=dis u+w非常相似。
因此我们就将这个问题转化为一个求最短路的问题:比如对于上面这个不等式,我们从j向i连一条权值为ck 的边
接下来,我们再新建一个 0 号点,从 0 号点向其他所有点连一条权值为 0 的边。
这个操作相当于新增了一个变量x0和n个约束条件:xi<=x0,从而将所有变量都和x0这一个变量联系起来。
然后以0号为起点,用spfa跑最短路。如果有负环,差分约束系统无解。否则设从0号点到i号点的最短路为dis i,则xi=dis i即为差分约束系统的一组可行解。
扩展
很多时候差分约束的条件并不是简单的小于等于号,这时候我们需要稍微做点变形。
如果有xi-xj>=ck,则可以两边同时乘-1,将不等号反转过来
如果有xi-xj=ck,则可以把这个不等式拆分为xi-xj<=ck和xi-xj>=ck两个约束条件
代码如下:
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
const int inf=0x3f3f3f3f;
typedef long long ll;
int n,m;
struct Edge
{
int from,to,dist;
};
vector<Edge>edges;
vector<int>G[maxn];
void add(int from,int to,int dist)
{
edges.push_back({from,to,dist});
int m=edges.size();
G[from].push_back(m-1);
}
int dis[maxn],cnt[maxn];
bool vis[maxn];
bool spfa()
{
memset(dis,inf,sizeof(dis));
dis[0]=0;
vis[0]=true;
queue<int>q;
q.push(0);
while(!q.empty())
{
int u=q.front();
q.pop();
vis[u]=false;
for(int i=0;i<G[u].size();i++)
{
Edge e=edges[G[u][i]];
int v=e.to;
if(dis[v]>dis[u]+e.dist)
{
dis[v]=dis[u]+e.dist;
if(!vis[v])
{
vis[v]=true;
cnt[v]++;
if(cnt[v]>=n+1)//注意是n+1.,算上超级源点一共有n+1个点
return false;
q.push(v);
}
}
}
}
return true;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cin>>n>>m;
for(int i=1;i<=n;i++)
add(0,i,0);
for(int i=0;i<m;i++)
{
int u,v,w;
cin>>u>>v>>w;
add(v,u,w);//注意连边是从j向i连边
}
if(spfa()==false)
cout<<"NO";
else
{
for(int i=1;i<=n;i++)
cout<<dis[i]<<" ";
}
}