排队布局
题目描述
核心思路
这题是需要用到差分约束。对于差分约束,我们最主要的是根据题目描述,找出题目中的所有不等式关系,我们设 x i x_i xi表示第 i i i头牛的位置:
- 根据“一些奶牛相互间存有好感,它们希望两者之间的距离不超过一个给定的数 L”,可以得到奶牛 a a a和奶牛 b b b之间至多相隔距离 L L L,即有 x b − x a ≤ L x_b-x_a\leq L xb−xa≤L
- 根据“一些奶牛相互间非常反感,它们希望两者间的距离不小于一个给定的数 D”,可以得到奶牛 a a a和奶牛 b b b之间至少相隔距离 D D D,即有 x b − x a ≥ D x_b-x_a\geq D xb−xa≥D
- 根据“奶牛排在队伍中的顺序和它们的编号是相同的,且同一位置上可以站多头牛”,可以得到 x i + 1 ≥ x i x_{i+1}\geq x_i xi+1≥xi
那么,我们再来观察题目要求的是 1 1 1号奶牛和 N N N号奶牛之间的最大距离,即要求变量 x n x_n xn的最大值。那么,由差分约束可知,需要跑单源最短路,求出最短距离。
但是,我们要考虑,是否满足 从源点出发,可以到达所有边这个条件。由于从上面的三个不等式中,并不能看出,从某个点出发,一定能走完所有边。因此,我们就需要添加一个虚拟源点了,所以我们添加超级源点 x 0 x_0 x0,然后从它出发,就可以到达所有边了,那么就可以使用差分约束的方法了。
需要找一个超级源点(记超级源点为 x 0 x_0 x0),发现这里求的距离都是牛之间的相对距离,所以联想到令 x 0 = 0 x_0=0 x0=0,又因为求的是形如 x i ≤ x j + c x_i\leq x_j+c xi≤xj+c的不等式,所以将所有点放在超级源点 0 0 0号点的左侧就可以得出 x i ≤ x 0 + 0 x_i\leq x_0+0 xi≤x0+0的不等式,这样从源点出发可以遍历所有的点,那么就可以遍历到所有的边。
因此,我们重新整理一下,上面的三个式子,统一写成 ≤ \leq ≤的形式(因为这是差分约束中求变量最大值的写法):
- x b ≤ x a + L x_b\leq x_a+L xb≤xa+L,则从节点 x a x_a xa向节点 x b x_b xb连一条长度为 L L L的边
- x a ≤ x b − D x_a\leq x_b-D xa≤xb−D,则从节点 x b x_b xb向节点 x a x_a xa连一条长度为 − D -D −D的边
- x i ≤ x i + 1 + 0 x_{i}\leq x_{i+1}+0 xi≤xi+1+0,则从节点 x i + 1 x_{i+1} xi+1向节点 x i x_i xi连一条长度为0的边
对于建立虚拟源点问题,有两种方法:
- 真正开辟一个超级源点0号点,那么此时建立好图后,就有 n + 1 n+1 n+1个节点
- 并不需要真正开辟一个虚拟源点,我们只需要刚开始的时候就把所有节点都放入队列中就行了,这种情况就等效于建立一个超级源点了(注意此时仍是
n
n
n个节点,因为并没有把超级源点0号节点建立出来)。如何理解呢?
- 我们这么想:对于spfa算法来说,假设我们真正设立了一个超级源点,这个源点连向其余的 n n n个节点。那么第一次扩展时,就会把超级源点的所有邻接点都加入队列q中,然后更新这 n n n个节点到超级源点的距离为0。也就是说,如果我们真的建立了一个超级源点,那么第一次扩展时,就会把它的所有邻接点( n n n个顶点)都更新最短距离为0。那么这不就等效于 《我们不用真的开辟一个超级源点,而是一开始就把这 n n n个顶点都放入spfa算法的队列q中,然后初始化这 n n n个点的距离为 0 0 0》嘛?只不过这里并不需要第一次扩展而已啦,想法都是一样的,只不过实现方法有所不同而已
下面使用两种方法来分别实现以上的两种想法
然后这一题,有三个问题:
- 对于第一个问题,我们可以跑一遍spfa算法求负环,如果存在负环,那么在说明不存在满足要求的方案,输出-1即可
- 对于第二个问题,想要求的是“1号牛和n号牛的最小距离是多少?即求 x n − x 1 x_n-x_1 xn−x1”,因此我们可以让1号节点称为起点,即设 x 1 = 0 x_1=0 x1=0,然后用spfa算法求出1号节点到其他各点的最短距离,最终求出 d i s t [ n ] dist[n] dist[n],如果 d i s t [ n ] ≥ I N F dist[n]\geq INF dist[n]≥INF则说明1号奶牛到 n n n号奶牛之间不存在约束关系,距离可以无限远,输出-2。
- 对于第三个问题,如果求出的 d i s t [ n ] < I N F dist[n]<INF dist[n]<INF,那么就直接输出此时的 d i s t [ n ] dist[n] dist[n]即可
代码
算法一:真正开辟一个超级源点0号节点
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=1010,M=3e4+10,INF=0x3f3f3f3f;
int n,m1,m2;
int h[N],e[M],ne[M],w[M],idx;
//记录每个点到起点的最短距离
int dist[N];
//spfa算法的循环队列
int q[N];
//记录每个点到起点一共经过了多少条边,用来判断负环
int cnt[N];
//用来判断某个节点是否已经放进q队列中
bool st[N];
//建图
void add(int a,int b,int c)
{
e[idx]=b;
w[idx]=c;
ne[idx]=h[a];
h[a]=idx++;
}
//spfa算法进行 判断负环和求出最短距离
bool spfa(int s)
{
memset(dist,0x3f,sizeof dist);
memset(cnt,0,sizeof cnt);
memset(st,0,sizeof st);
int hh=0,tt=1;
dist[s]=0;
q[0]=s;
st[s]=true;
while(hh!=tt)
{
//取出队头元素
int t=q[hh++];
//队满
if(hh==N)
hh=0;
st[t]=false;//队头元素出队
//遍历t的所有邻接点
for(int i=h[t];~i;i=ne[i])
{
int j=e[i];//取出t的邻接点j
//松弛
if(dist[j]>dist[t]+w[i])
{
dist[j]=dist[t]+w[i];
//记录j到起点所走过的边数
cnt[j]=cnt[t]+1;
//有n个点,如果此时节点j到起点所经历的边数>=n,则说明有n+1个顶点,则有负环
if(cnt[j]>=n)
return true;
//如果j还没有放入队列q中
if(!st[j])
{
q[tt++]=j; //把j放入队列q中
//如果走到队尾
if(tt==N)
tt=0;
st[j]=true;
}
}
}
}
//说明不存在负环
return false;
}
int main()
{
//初始化表头
memset(h,-1,sizeof h);
scanf("%d%d%d",&n,&m1,&m2);
//建立了一个真正的超级源点,因此需要从超级源点0号点到其他n个节点连一条长度为0的边
for(int i=1;i<=n;i++)
add(0,i,0);
//第一个约束条件,节点i+1向节点i连一条长度为0的边
for(int i=1;i<n;i++)
add(i+1,i,0);
while(m1--)
{
int a,b,L;
scanf("%d%d%d",&a,&b,&L);
//保证b要>=a
if(a>b)
swap(a,b);
//第二个约束条件,节点a向节点b连一条长度为L的边
add(a,b,L);
}
while(m2--)
{
int a,b,D;
scanf("%d%d%d",&a,&b,&D);
//保证b要>=a
if(a>b)
swap(a,b);
//第三个约束条件,节点b向节点a连一条长度为-D的边
add(b,a,-D);
}
//存在负环
if(spfa(0))
puts("-1");
else
{
//把1号节点当作起点
spfa(1);
//说明1号奶牛到N号奶牛的距离可以任意大
if(dist[n]==INF)
puts("-2");
//否则就输出1号奶牛到N号奶牛的最短距离
else
printf("%d\n",dist[n]);
}
return 0;
}
算法二:不需要开辟一个超级源点0号节点
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=1010,M=3e4+10,INF=0x3f3f3f3f;
int n,m1,m2;
int h[N],e[M],ne[M],w[M],idx;
//记录每个点到起点的最短距离
int dist[N];
//spfa算法的循环队列
int q[N];
//记录每个点到起点一共经过了多少条边,用来判断负环
int cnt[N];
//用来判断某个节点是否已经放进q队列中
bool st[N];
//建图
void add(int a,int b,int c)
{
e[idx]=b;
w[idx]=c;
ne[idx]=h[a];
h[a]=idx++;
}
//spfa算法进行 判断负环和求出最短距离
bool spfa(int num)
{
memset(dist,0x3f,sizeof dist);
memset(cnt,0,sizeof cnt);
memset(st,0,sizeof st);
int hh=0,tt=1;
//将这num个节点都加入队列中,并且每个节点的最短距离初始化为0,标记每个节点都放入q队列中了
//那么就等效于开辟了一个超级源点
for(int i=1;i<=num;i++)
{
q[tt++]=i;
dist[i]=0;
st[i]=true;
}
while(hh!=tt)
{
//取出队头元素
int t=q[hh++];
//队满
if(hh==N)
hh=0;
st[t]=false;//队头元素出队
//遍历t的所有邻接点
for(int i=h[t];~i;i=ne[i])
{
int j=e[i];//取出t的邻接点j
//松弛
if(dist[j]>dist[t]+w[i])
{
dist[j]=dist[t]+w[i];
//记录j到起点所走过的边数
cnt[j]=cnt[t]+1;
//有n个点,如果此时节点j到起点所经历的边数>=n,则说明有n+1个顶点,则有负环
if(cnt[j]>=n)
return true;
//如果j还没有放入队列q中
if(!st[j])
{
q[tt++]=j; //把j放入队列q中
//如果走到队尾
if(tt==N)
tt=0;
st[j]=true;
}
}
}
}
//说明不存在负环
return false;
}
int main()
{
//初始化表头
memset(h,-1,sizeof h);
scanf("%d%d%d",&n,&m1,&m2);
//第一个约束条件,节点i+1向节点i连一条长度为0的边
for(int i=1;i<n;i++)
add(i+1,i,0);
while(m1--)
{
int a,b,L;
scanf("%d%d%d",&a,&b,&L);
//保证b要>=a
if(a>b)
swap(a,b);
//第二个约束条件,节点a向节点b连一条长度为L的边
add(a,b,L);
}
while(m2--)
{
int a,b,D;
scanf("%d%d%d",&a,&b,&D);
//保证b要>=a
if(a>b)
swap(a,b);
//第三个约束条件,节点b向节点a连一条长度为-D的边
add(b,a,-D);
}
//存在负环
if(spfa(n))
puts("-1");
else
{
//把1号节点当作起点
spfa(1);
//说明1号奶牛到N号奶牛的距离可以任意大
if(dist[n]==INF)
puts("-2");
//否则就输出1号奶牛到N号奶牛的最短距离
else
printf("%d\n",dist[n]);
}
return 0;
}