SDU_week8_A - 区间选点 II（差分约束系统+SPFA求最长路）

题目描述

给定一个数轴上的 n 个区间，要求在数轴上选取最少的点使得第 i 个区间 [ai, bi] 里至少有 ci 个点

请使用差分约束系统的解法解决这道题。

Input
输入第一行一个整数 n 表示区间的个数，接下来的 n 行，每一行两个用空格隔开的整数 a，b 表示区间的左右端点。1 <= n <= 50000， 0 <= ai <= bi <= 50000 并且 1 <= ci <= bi - ai+1。

Output
输出一个整数表示最少选取的点的个数

Sample Input
5
3 7 3
8 10 3
6 8 1
1 3 1
10 11 1

Sample Output
6

思路

这道题是给定若干区间及每个区间内要求存在的点数，求出最少要多少个点。之前使用贪心做过（按照右端点排序），这次要求有了提升，每个区间段内的点不再是一个，而是由输入决定。
解题思路是使用差分约束系统。解体由以下四个步骤：

1. 找到差值不等式。通常使用前缀和构造（可以参考CSP再卖菜）。本题中有dis[yi]-dis[xi-1]>=ci、0≤dis[i]-dis[i-1]≤1三个不等式约束（注意结构体edge要开够）。
2. 利用链式前向星存图。
3. 找到s，即dis[i]，本题中dis[i]表示数轴上 [0, i] 之间选点的个数。
4. 用SPFA跑最短路/<=/求最大 OR 最长路/>=/求最小，注意使用SPFA是一方面因为相邻约束时可能出现负值，另一方面因为最长路和最短路在SPFA中只需改变两处，一个是松弛条件，一个是dis数组初始化值（最短路为正无穷，最长路为负无穷）。

注意

1. 本题中区间段是从0开始，我们需要将0空出来保证dis[yi]-dis[xi-1]>=ci有效，因为整体右移一格,但是SPFA依然从0开始跑。
2. 求最长路构造>=号，同时一方面要理解点会多次入队，另一方面记得dis[]初始化为负无穷。

代码

#define _CRT_SECURE_NO_WARNINGS
#define _ ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#include <iostream>
#include<cmath>
#include<algorithm>
#include<queue>
#include<string>
#include<cstring>
using namespace std;

struct edge
{
    int u, v, w, next;
}e[50009*4];
int tot;
int head[50009];
void addEdge(int u,int v,int w)
{
    e[++tot].u = u;
    e[tot].v = v;
    e[tot].w = w;
    e[tot].next = head[u];
    head[u] = tot;
}

int dis[50009];//数轴上[0,i]之间选点的个数
bool inq[50009];
queue<int> q;

void spfa()
{
    memset(inq, 0, sizeof(inq));
    dis[0] = 0;
    q.push(0);
    while (!q.empty())
    {
        int u = q.front(); q.pop();
        inq[u] = 0;
        for (int i = head[u]; i; i = e[i].next)
        {
            int v = e[i].v;
            int w = e[i].w;
            if (dis[v] < dis[u] + w)
            {
                dis[v] = dis[u] + w;
                if (!inq[v])
                {
                    q.push(v);
                    inq[v] = 1;
                }
            }
        }
    }
}

int main()
{
    memset(e, 0, sizeof(e));
    memset(head, 0, sizeof(head));
    memset(dis, 0xcf, sizeof(dis));//最长路初始化为负无穷
    tot = 0;
    dis[0] = 0;
    int n, a, b, c;//组数，a左边界，b右边界，c权值（区间点数）
    int maxB = 0;//最大右端点
    cin >> n;
    //因为区间从0开始，所以把0空出来，区间传入时右移1位
    while (n--)
    {
        cin >> a >> b >> c;
        addEdge(a, b + 1, c);
        maxB = max(b + 1, maxB);
    }
    for (int i = 1; i <= maxB; i++)
    {
        addEdge(i - 1, i, 0);
        addEdge(i, i - 1, -1);
    }
    spfa();
    //cout << "maxB" << maxB << endl;
    cout << dis[maxB];
    return 0;
}