洛谷 P3275 [SCOI2011]糖果

洛谷 P3275 [SCOI2011]糖果

Description

• 幼儿园里有 N 个小朋友，lxhgww老师现在想要给这些小朋友们分配糖果，要求每个小朋友都要分到糖果。但是小朋友们也有嫉妒心，总是会提出一些要求，比如小明不希望小红分到的糖果比他的多，于是在分配糖果的时候，lxhgww 需要满足小朋友们的 K 个要求。幼儿园的糖果总是有限的，lxhgww想知道他至少需要准备多少个糖果，才能使得每个小朋友都能够分到糖果，并且满足小朋友们所有的要求。

Input

• 输入的第一行是两个整数 N，K。接下来 K行，表示这些点需要满足的关系，每行 3 个数字，X，A，B。
  • 如果 X=1， 表示第 A 个小朋友分到的糖果必须和第 B 个小朋友分到的糖果一样多；
  • 如果 X=2， 表示第 A 个小朋友分到的糖果必须少于第 B个小朋友分到的糖果；
  • 如果 X=3， 表示第 A个小朋友分到的糖果必须不少于第 B个小朋友分到的糖果；
  • 如果 X=4， 表示第 A个小朋友分到的糖果必须多于第 B 个小朋友分到的糖果；
  • 如果 X=5， 表示第 A个小朋友分到的糖果必须不多于第 B 个小朋友分到的糖果；

Output

• 输出一行，表示 lxhgww老师至少需要准备的糖果数，如果不能满足小朋友们的所有要求，就输出 −1。

Sample Input

5 7
1 1 2
2 3 2
4 4 1
3 4 5
5 4 5
2 3 5
4 5 1

Sample Output

11

Data Size

• 对于 100% 的数据，保证 N≤100000

• 对于所有的数据，保证 K≤100000,1≤X≤5,1≤A,B≤N

题解：

• 差分约束。
• 首先如果遇见a < b这种式子，可以转换成a <= b - 1。
• 第一步把题中的要求抽象成不等式然后建图。
• 第二步建立一个超级源点与每个点连上一条边权为1的边。因为“每个小朋友都要分到糖果”
• 第三步最短路
• 坑点：
  1. 答案是所有dis之和。因为dis[i]表示i这个小孩至少需要多少个糖
  2. 答案要用longlong存
  3. 超级源点与各点连边时，要倒序。不然超时（玄学）
  4. 多条件语句时，用else if，不要用if + continue

#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>
#define N 1000005
using namespace std;

struct E {int next, to, dis;} e[N];
int n, m, num;
int h[N], cnt[N], dis[N];
bool vis[N];

int read()
{
    int x = 0; char c = getchar();
    while(c < '0' || c > '9') c = getchar();
    while(c >= '0' && c <= '9') {x = x * 10 + c - '0'; c = getchar();}
    return x;
}

void add(int u, int v, int w)
{
    e[++num].next = h[u];
    e[num].to = v;
    e[num].dis = w;
    h[u] = num;
}

int spfa()
{
    int flag = 0, ans = 0;
    memset(dis, -0x3f, sizeof(dis));
    queue<int> que;
    dis[n + 1] = 0, vis[n + 1] = 1, que.push(n + 1);
    while(!que.empty())
    {
        int now = que.front();
        cnt[now]++;
        if(cnt[now] >= n) {flag = 1; break;}
        que.pop();  vis[now] = 0;
        for(int i = h[now]; i != 0; i = e[i].next)
            if(dis[now] + e[i].dis > dis[e[i].to])
            {
                dis[e[i].to] = dis[now] + e[i].dis;
                if(!vis[e[i].to])
                {
                    vis[e[i].to] = 1;
                    que.push(e[i].to);
                }
            }
    }
    if(flag) return -1;
    for(int i = 1; i <= n; i++) ans += dis[i];
    return ans;
}

int main()
{
    cin >> n >> m;
    for(int i = 1; i <= m; i++)
    {
        int op = read(), a = read(), b = read();
        if(op == 1) add(a, b, 0), add(b, a, 0);
        else if(op == 2)
        {
            if(a == b) {cout << -1; return 0;}
            add(a, b, 1);
        }
        else if(op == 3) add(b, a, 0);
        else if(op == 4)
        {
            if(a == b) {cout << -1; return 0;}
            add(b, a, 1);
        }
        else if(op == 5) add(a, b, 0);
    }
    for(int i = n; i >= 1; i--) add(n + 1, i, 1); //玄学：倒序建边 
    cout << spfa();
    return 0;
}

转载于:https://www.cnblogs.com/BigYellowDog/p/11230738.html