程序设计思维与实践 Week8 作业 A 区间选点 II

差分约束系统解决区间选点问题。

题目描述：

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

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

input：

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

output：

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

思路描述：

构造不等式组：dis[i]表示数轴上 [0, i] 之间选点的个数

对于第 i 个区间 [xi,yi] 需要满足 dis[yi]-dis[xi-1]>=ci

同时，应保证dis是有意义的：0≤dis[i]-dis[i-1]≤1

求该差分约束系统的最小解，转化为 ≥ 不等式组用spfa跑最长路，答案为dis[max{yi}]

注意：因为存在不等式0<=sum[0]-sum[-1]<=1,所以将区间整体右移一个单位.

#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
int n,m,tot,dis[50010],sum[50010],head[50010],big_b;
bool flag[50010*5];
queue<int> q;
struct node
{
	int v;
	int w;
	int next;
}e[50010*5];
void add_edge(int u,int v,int w)
{
	e[++tot].v=v;
	e[tot].w=w;
	e[tot].next=head[u];
	head[u]=tot;
}
void spfa()
{
	memset(dis,128,sizeof(dis));
	dis[0]=0;
	q.push(0);
	flag[0]=1;
	while(!q.empty())
	{
		int u=q.front();
		flag[u]=0;
		q.pop();
		for(int i=head[u];i;i=e[i].next)
		{
			int v=e[i].v;
			if(dis[v]<dis[u]+e[i].w)
			{
				dis[v]=dis[u]+e[i].w;
				if(!flag[v])
				{
					q.push(v);
					flag[v]=1;
				}
			}
		}
	}
}
int main()
{
	int x,y,z;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>x>>y>>z;
		add_edge(x,y+1,z);
		big_b=max(big_b,y+1);
	}
	for(int i=1;i<=big_b;i++)
	{
		add_edge(i-1,i,0);
		add_edge(i,i-1,-1);
	}
	spfa();
	cout<<dis[big_b];
	return 0;
}