1. 程式人生 > >P1993 小K的農場 - 差分約束

P1993 小K的農場 - 差分約束

訪問 const tro def 一次 {} 通過 mat ring

看出不等式之後,通過移項套模型
大概不等式模型是這樣的:\(x_v <= x_u + w_{(u, v)}\)
\(x_v - x_u <= w_{(u, v)}\)
從減數到被減數連邊
長得和最短路的三角形不等式似的
所以求解也圍繞著不等式
可以看出\(x_v\)的最大值就是一條最短路的形式
若最短路不存在(負環),\(x_v\)的值也不存在
用dfs_spfa判最短路,註意退出遞歸層的時候讓vis[x] = 0
另外註意求最短路時初始化d數組為INF,雖然說判負環初始化為0貌似沒啥問題(也可能是這題數據水),但是初始化對於復雜度影響不大。。。

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
using namespace std;
#define debug(x) cerr << #x << "=" << x << endl;
const int MAXN = 10000 + 10;
int n,m,vis[MAXN],cnt[MAXN],last[MAXN],edge_tot,d[MAXN];
struct Edge{
    int u, v, w, to;
    Edge(){}
    Edge(int u, int v, int w, int to) : u(u), v(v), w(w), to(to) {}
}e[MAXN * 2];
inline void add(int u, int v, int w) {
    e[++edge_tot] = Edge(u, v, w, last[u]);
    last[u] = edge_tot; 
}
bool flg;
queue<int> q;

void spfa(int x) {
    vis[x] = 1;
    if(flg) return;
    for(int i=last[x]; i; i=e[i].to) {
        int v = e[i].v, w = e[i].w;
        if(d[v] > d[x] + w) {
            if(vis[v]) {
                flg = true;//在遞歸層裏又被訪問一次,並且路徑更短,那麽是不是可以多走幾次這樣的路?
                return;
            }
            d[v] = d[x] + w;
            spfa(v);
        }
    }
    vis[x] = 0;
}

int main() {
    scanf("%d%d", &n, &m);
    for(int i=1; i<=m; i++) {
        int cmd, a, b, c;
        scanf("%d", &cmd);
        if(cmd == 1) {
            scanf("%d%d%d", &a, &b, &c);
            add(a, b, -c);
        } else if(cmd == 2) {
            scanf("%d%d%d", &a, &b, &c);
            add(b, a, c);
        } else {
            scanf("%d%d", &a, &b);
            add(a ,b, 0);
            add(b, a, 0);
        }
    }
    memset(d, 0x3f, sizeof(d));
    for(int i=1; i<=n; i++) {   
        d[i] = 0;
        spfa(i);
        if(flg) break;
    }
    if(flg) printf("No");
    else printf("Yes");
    return 0;
}

另外,bfs判負環(復雜度上界為n*m):

bool SPFA(int s) {
    dist[s] = 0;
    q.push(s);
    vis[s] = 1;
    cnt[s] = 0;
    while(!q.empty()) {
        int x = q.front();
        q.pop();
        vis[x] = 0;
        for(int i=last[x]; i; i=e[i].to) {
            int v = e[i].v;
            int w = e[i].w;
            if(dist[v] > dist[x] + w) {
                dist[v] = dist[x] + w;
                cnt[v] = cnt[x] + 1;
                if(cnt[v] >= n) {
                    return 1;  
                }
                if(!vis[v]) q.push(v), vis[v] = 1;
            }
        }
    }
    return 0;
}

P1993 小K的農場 - 差分約束