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

题目描述

给定一个数轴上的 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

解题思路

之前曾经用过贪心的策略解决过类似的题目，只不过那次一个区间中只能选取一个点，而这次可以选择多个点。上次用贪心解决，这次用差分约束系统解决此题。当然，在没有听讲解之前是绝不会想到用差分约束系统去解决的…
为了构造差分约束系统，我们用 sum[i] 来表示数轴上[0 , i]之间选取得点的个数。那么对于题目中得每个区间输入的a,b,c，必须满足sum[b]-sum[a-1]>=c。当然，考虑到a可能输入0，这样会导致a-1小于0，为避免此情况，将区间整体右移一位，变成sum[b+1]-sum[a]>=c。由于选取得全部都是整数，则必须满足0<=sum[i]-sum[i-1]<=1。可以看出，这种不等于形式和求最短路时边的松弛条件很相似。那么便可将其化为求单源最短路问题，其中起点为0，但这样求得的解是最大解，题目中要求找出最少的点，那么就需要将<转化为>，并求最长路径，但是此时会出现负权边，这就决定了需要使用SPFA算法来解决，只不过此时默认了不会出现负环的情况，因此简化了许多。

代码

#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
const int maxm=5e4+10,maxn=1e5+10,inf=-5e8;

int head[maxm],dis[maxm],inq[maxm];//inq记录该点当前是否在队列中 
int n,a,b,c;
int tot=0;
int mend=0;//记录区间端点最大值 
int sum=0;//记录最后结果 
struct Edge
{
	int u,v,w,nxt;//出发点，到达点，权值，下一个点 
}e[maxn];
void addedge(int u,int v,int w)
{
    e[tot].u=u;
    e[tot].v=v;
    e[tot].w=w;
    e[tot].nxt=head[u];
    head[u]=tot;
    tot++;
} 

void init(int N)
{
	tot=0;
	for(int i=0;i<N;i++)
	{
		inq[i]=0;
		dis[i]=inf;
	}
}
queue<int> q;//队列 
void spfa()
{
	while(q.size()) q.pop();
	q.push(0);
	dis[0]=0;//初始点距离 
	inq[0]=1;//起点标记在队列中 
	while(!q.empty())
	{
		int t=q.front(); q.pop();
		inq[t]=0;//t已经不在队列 
		for(int i=head[t];i!=-1;i=e[i].nxt)
		{
			int y=e[i].v,w=e[i].w;
			
			if(dis[y]<dis[t]+w)
			{
				dis[y]=dis[t]+w;
                if(!inq[y])
                {
                    q.push(y);
                    inq[y]=1;
			    }
			}	
	    } 
	}
}

int main()
{
	std::ios::sync_with_stdio(false);
	cin>>n;
	
	memset(head,-1,sizeof(head));
	for(int i=0;i<n;i++)
	{
		cin>>a>>b>>c;
		
		if(b+1>mend) mend=b+1;//记录区间最大端点 
        addedge(a,b+1,c);//把区间整体右移 
	}
	for(int i=1;i<=mend;i++)//输入初始约束条件 
	{
		addedge(i,i-1,-1);//长度为1的区间只能取0或1 
		addedge(i-1,i,0);
	}
    init(mend);
    spfa();
    for(int i=1;i<=mend;i++)
        sum+=dis[i]-dis[i-1];
    cout<<sum<<endl;
	return 0;
}