P3275 [SCOI2011]糖果「差分约束」

P3275 SCOI2011糖果

题目描述：

n个小朋友，m个条件(X,A,B)，条件有五种

• 如果 X=1， 表示第 A个小朋友分到的糖果必须和第 B 个小朋友分到的糖果一样多；
• 如果 X=2 表示第 A 个小朋友分到的糖果必须少于第 B个小朋友分到的糖果；
• 如果 X=3， 表示第 A个小朋友分到的糖果必须不少于第 B个小朋友分到的糖果；
• 如果 X=4， 表示第 A 个小朋友分到的糖果必须多于第 B个小朋友分到的糖果；
• 如果 X=5， 表示第 A 个小朋友分到的糖果必须不多于第 B个小朋友分到的糖果;

问老师至少要准备多少的糖果才能满足所有的小朋友的要求，如果不难满足输出-1

思路：

差分约束，问最小值，即用最长路，则转换成>=

1. a[i] = a[j],转换成 $a[i]>=a[j]+0和a[j]>=a[i]+0$，转换成从i到j建立一条权值为0的边，且从j到i建立一条权值为0的边
2. a[i] < a[j],转换成 $a[j]>=a[i]+1$ ，从i 到j建立一条权值为1的边
3. a[i] >= a[j]，$a[i]>=a[j]+0$, 从j到i建立一条权值为0的边
4. a[i] > a[j]，$a[i]>=a[j]+1$，从j到i建立一条权值为1的边
5. a[i] <= a[j], $a[j]>=a[i]+0$,从i到j建立一条权值为0的边

注意，2和4的情况必须满足i!=j

再建立一个超级源点0，然后跑最长路就可以，计算答案的时候计算 ${\sum }_{i=1}^{n}a[i]$即可

#include <bits/stdc++.h>
using namespace std;

#define endl '\n'
#define inf 0x3f3f3f3f
#define mod 1000000007
#define m_p(a,b) make_pair(a, b)
#define mem(a,b) memset((a),(b),sizeof(a))
#define io ios::sync_with_stdio(false); cin.tie(0); cout.tie(0)

typedef long long ll;
typedef pair <int,int> pii;

#define MAX 300000 + 50
int n, m;
int a, b, op;
int tot;
int head[MAX];
struct ran{
    int to, nex, val;
}tr[MAX];
inline void add(int u, int v, int c){
    tr[++tot].to = v;
    tr[tot].val = c;
    tr[tot].nex = head[u];
    head[u] = tot;
}

bool vis[MAX];
ll dis[MAX];
int cnt[MAX];

void SPFA(){
    queue<int>q;
    for(int i = 1; i <= n; ++i){
        q.push(i);
        vis[i] = 1;
        dis[i] = 1;
    }
    while (!q.empty()) {
        int u = q.front();q.pop();
        vis[u] = 0;
        cnt[u] = 0;
        for(int i = head[u]; i; i = tr[i].nex){
            int v = tr[i].to;
            if(dis[u] + tr[i].val > dis[v]){
                dis[v] = dis[u] + tr[i].val;
                ++cnt[i];
                if(cnt[i] >= n + 1){
                    cout << -1 << endl;
                    return;
                }
                if(!vis[v]){
                    vis[v] = 1;
                    q.push(v);
                }
            }
        }
    }
    ll ans = 0;
    for(int i = 1; i <= n; ++i)ans += dis[i];
    cout << ans << endl;
}

void work(){
    cin >> n >> m;
    for(int i = 1; i <= m; ++i){
        cin >> op >> a >> b;
        if(op == 1){
            add(a, b, 0);add(b, a, 0);
        }
        else if(op == 2)add(a, b, 1);
        else if(op == 3)add(b, a, 0);
        else if (op == 4)add(b, a, 1);
        else add(a, b, 0);
        if(op % 2 == 0 && a == b){
            cout << -1 << endl;
            return;
        }
    }
    SPFA();
}


int main(){
    io;
    work();
    return 0;
}