区间选点--差分约束系统不等式找最长路

题目要求

给定一个数轴上的 n 个区间，要求在数轴上选取最少的点使得第 i 个区间 [ai, bi] 里至少有 ci 个点
Input
输入第一行一个整数 n 表示区间的个数，接下来的 n 行，每一行两个用空格隔开的整数 a，b 表示区间的左右端点。1 <= n <= 50000， 0 <= ai <= bi <= 50000 并且 1 <= ci <= bi - ai+1。c是点的个数。
Output
输出一个整数表示最少选取的点的个数
Sample Input
5
3 7 3
8 10 3
6 8 1
1 3 1
10 11 1
Sample Output
6

求解思路

使用差分约束的方法做。

• 构造不等式组
  记sum[i]表示数轴上[0,i]之间选点的个数
  对于第i个区间[ai,bi]需要满足sum[bi]-sum[ai-1]≥ci
• 需要保证sum是有意义的，每个数选或不选，只能是0或1
  0≤sum[i]-sum[i-1]≤1
• 求该差分约束系统的最小解，转化为≥不等式跑最长路，计算sum[max{bi}]
• 根据图论找最长路的方法，将sum数组作为距离0的距离，ci表示边权，找到最右侧端点的距离即为最少端点数
• 因为是从零开始的，所以将所有区间往右移一次，即左右端点都加一，权重c不变，这样区间就是从1开始，可以从0进行spfa，找最长边。

代码

#include<iostream>
#include<queue>
#include<stdio.h>
#define inf -1e8
using namespace std;

struct edge
{
	long long u, v, w, nxt;
	edge()
	{
		u = 0; v = 0; nxt = 0;
	}
};

//链式前向星
long long head[50054];
edge Edge[5000020];
long long tot = 0;
void addedge(long long u, long long v, long long w)
{
	Edge[tot].u = u;
	Edge[tot].v = v;
	Edge[tot].w = w;
	Edge[tot].nxt = head[u];
	head[u] = tot;
	tot++;
}
long long n;

long long dis[50054];
long long pre[50054];
long long cnt[50054];
long long arrive[50054];

void SPFA(long long s)
{
	for (long long i = 0; i <= n+5; i++)//初始化
	{
		arrive[i] = 0;
		dis[i] = inf;
		//dis[i] = 0;
		pre[i] = 0;
		cnt[i] = 0;
	}
	dis[s] = 0;
	arrive[s] = 1;
	queue<long long> q;
	q.push(s);
	while (!q.empty())
	{
		long long temp = q.front();
		q.pop();
		arrive[temp] = 0;
		for (long long i = head[temp]; i != -1; i = Edge[i].nxt)
		{
			if (dis[Edge[i].v] < dis[Edge[i].u] + Edge[i].w)
			{

				cnt[Edge[i].v] = cnt[Edge[i].u] + 1;
				if (cnt[Edge[i].v] >= n)
				{
					//找到负环
				}
				dis[Edge[i].v] = dis[Edge[i].u] + Edge[i].w;
				//	cout << "dis[v]=" << dis[Edge[i].v]<<"v="<< Edge[i].v;
				//	pre[Edge[i].v] = Edge[i].u;
				if (!arrive[Edge[i].v])
				{
					q.push(Edge[i].v);
					arrive[Edge[i].v] = 1;
				}
			}
		}
	}
}

int main()
{
	scanf("%lld", &n);
	tot = 0;
	for (long long i = 0; i <= 50052; i++)//初始化
	{
		head[i] = -1;
	}
	long long maxb = 0;
	for (long long i = 50052; i > -5; i--)
	{
		addedge(i, i - 1, -1);
		addedge(i - 1, i, 0);
	}
	for (long long i = 0; i < n; i++)
	{
		long long a, b, c;
		scanf("%lld%lld%lld", &a, &b, &c);
		//cin >> a >> b >> c;
		if (maxb < b+1)
			maxb = b+1;	
		addedge(a, b+1, c);
	}

	SPFA(0);
	printf("%lld", dis[maxb]);
	//cout << maxb;


}