POJ 3169 Layout(差分约束)

【题目链接】
http://poj.org/problem?id=3169

题目意思

一群编号1~n牛排队。按编号排(多头牛可以站在同一位置)。由于不同牛关系不同,有ML对恋爱牛要求排的队伍距离不超过w,有MD对仇牛要求距离超过w。现在问你队伍最大长为多少?如果无法找到满足要求的安排输出-1,如果最长无法求输出-2。

解题思路

第一题的差分约束,差分利用最短路的缩点时候d[v] > d[u]+w[u,v],转换后d[v]-d[u] > w[u,v];就满足题目的不等式了,那么只要把不等式转换成边跑最短就能求出想要结果。当时不等式相当于有向边(谁减谁不能乱),这时候就有可能无法到达点,所以按题目的两个条件 A-B>=w和D-C>=w建图还不够,还有个隐藏条件牛排队按序列的也就是 i - (i-1) >=0,也就是含有一条有向边贯穿整张图。

代码部分


#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <string.h>
#include <queue>
#include <string>
#include <map>
using namespace std;
#define LL long long
#define inf 0x3f3f3f3
const int N = 1005;
int dis[N];   ///存储最短路 
int vis[N];  ///记录是否在队列 
int cnt[N];  ///记录进入队列次数 
int k,n,m;
struct node
{
    int v,w; 
};
vector <node> M[N]; 
void add(int u,int v,int w) ///添加边 
{
    node t;
    t.v = v;
    t.w = w;
    M[u].push_back(t);
}
void init()  ///初始化 
{
    for (int i = 1;i <= n; i++)
    {
        dis[i] = inf;
        vis[i] = 0; 
        cnt[i] = 0;
        add (i,i-1,0);  ///s[i] - s[i-1] >= 0 
    }
}
int spfa()
{
    queue<int>q;
    q.push(1);
    vis[1] = 1;
    cnt[1] = 1; 
    dis[1] = 0;
    while (!q.empty())
    { 
        int t = q.front();
        q.pop();
        for (int i = 0; i < M[t].size();i++)
        {
            node p = M[t][i];
            if (dis[p.v] > dis[t]+p.w)  ///松弛 
            {
                dis[p.v] = dis[t]+p.w;
                if (!vis[p.v])   ///判断是否在队列 
                {
                    q.push(p.v);
                    cnt[p.v]++;
                    if (cnt[p.v] > n)  ///存在负环了 
                        return -1;
                    vis[p.v] = 1;
                }
            }
        }
        vis[t] = 0;
    }
    if (dis[n] == inf)
        return -2;
    else return dis[n];
}
int main()
{
    while (~scanf("%d %d %d",&n,&m,&k))
    {
        init();
        for (int i = 0; i< m; i++)
        {
            int u,v,w;
            scanf("%d %d %d",&u,&v,&w);  ///s[u] - s[v] >= w 
            add(u,v,w);
        }
        for (int i = 0; i < k; i++)
        {
            int u,v,w;
            scanf("%d %d %d",&u,&v,&w);
            add(v,u,-w);    ///s[v] - s[u] >= -w
        }
        printf("%d\n",spfa()); 
    }
    return 0;
} 

猜你喜欢

转载自blog.csdn.net/pashuihou/article/details/80082208